Segmentation of COVID-19 CT lesions in CT scans through transfer learning
Since its emergence at the end of 2019, SARS-CoV-2 has infected millions worldwide, challenging healthcare systems globally. This has prompted many researchers to explore how machine learning can assist clinicians in diagnosing infections caused by SARS-CoV-2. Building on previous studies, we propose a novel deep learning framework designed for segmenting lesions evident in Computed Tomography (CT) scans. For this work, we utilized a dataset consisting of 20 CT scans annotated by experts and performed training, validation, and external evaluation of the deep learning models we implemented, using a 5-fold cross-validation scheme. When splitting data by slice, our optimal model achieved noteworthy performance, attaining a Dice Similarity Coefficient (DSC) and Intersection over Union (IoU) score of 0.8644 and 0.7612 respectively, during the validation phase. In the external evaluation phase, the model maintained strong performance with a DSC and an IoU score of 0.7211 and 0.5641, respectively. When splitting data by patient, our optimal model achieved a DSC score of 0.7989 and an IoU score of 0.6686 during the validation phase. During the external evaluation phase, the model maintained strong performance with a DSC and IoU score of 0.7369 and 0.5837, respectively. The results of this research suggest that incorporating transfer learning along with appropriate preprocessing techniques, can contribute to achieving state-of-the-art performance in the segmentation of lesions associated with SARS-CoV-2 infections.
Authors
- Cleo Sgouropoulou (ORCID: https://orcid.org/0000-0001-8173-2622)
- Athanasios Voulodimos (ORCID: https://orcid.org/0000-0002-0632-9769)
- Christos Troussas (ORCID: https://orcid.org/0000-0002-9604-2015)
- Symeon Psaraftis-Souranis
Institutions
- National Technical University of Athens (GR)
- University of West Attica (GR)
Publication Details
- Journal
- Computer Science and Information Systems
- Published
- 2024-11-05
- DOI
- https://doi.org/10.2298/csis240229066p
- Citations
- 1
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 0.40