Single-Shot Lightweight Model For The Detection of Lesions And The Prediction of COVID-19 From Chest CT Scans
Ter-Sarkisov, A.
Show abstract
We introduce a lightweight model based on Mask R-CNN with ResNet18 and ResNet34 backbone models that segments lesions and predicts COVID-19 from chest CT scans in a single shot. The model requires a small dataset to train: 650 images for the segmentation branch and 3000 for the classification branch, and it is evaluated on 21292 images to achieve a 42.45% average precision (main MS COCO criterion) on the segmentation test split (100 images), 93.00% COVID-19 sensitivity and F1-score of 96.76% on the classification test split (21192 images) across 3 classes: COVID-19, Common Pneumonia and Control/Negative. The full source code, models and pretrained weights are available on https://github.com/AlexTS1980/COVID-Single-Shot-Model.
Matching journals
The top 8 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- A Framework for Falsifiable Explanations of Machine Learning Models with an Application in Computational Pathology 95%
- STAMP: Simultaneous Training and Model Pruning for Low Data Regimes in Medical Image Segmentation 94%
- Spatial Transcriptomics Expression Prediction from Histopathology Based on Cross-Modal Mask Reconstruction and Contrastive Learning 93%
Similar papers in this journal
- Audit to Forget: A Unified Method to Revoke Patients' Private Data in Intelligent Healthcare 92%
- Generative AI Enables Medical Image Segmentation in Ultra Low-Data Regimes 92%
- Segmenting functional tissue units across human organs using community-driven development of generalizable machine learning algorithms 92%
Similar papers in this journal
- Generalized Radiograph Representation Learning via Cross-supervision between Images and Free-text Radiology Reports 97%
- UDCT: Unsupervised data to content transformation with histogram-matching cycle-consistent generative adversarial networks 92%
- COSIME: Cooperative multi-view integration with Scalable and Interpretable Model Explainer 91%
Similar papers in this journal
- Deep learning models for COVID-19 chest x-ray classification: Preventing shortcut learning using feature disentanglement 96%
- Small hand-designed convolutional neural networks outperform transfer learning in automated cell shape detection in confluent tissues 94%
- Detection and measurement of butterfly eyespot and spot patterns using convolutional neural networks 93%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.