Encoder-decoder convolutional neural network for simple CT segmentation of COVID-19 infected lungs

被引:0
作者
Newson, Kiri S. [1 ]
Benoit, David M. [2 ]
Beavis, Andrew W. [3 ,4 ,5 ]
机构
[1] Department of Physics and Mathematics, University of Hull, Hull
[2] E. A. Milne Centre for Astrophysics, Department of Physics and Mathematics, University of Hull, Hull
[3] Medical Physics Department, Queen’s Centre for Oncology, Hull University Teaching Hospitals NHS Trust, Hull, Cottingham
[4] Medical Physics and Biomedical Engineering, University College London, University of London, London
[5] Hull York Medical School, University of Hull, Hull
关键词
Autoencoder; Automated segmentation; CNN; COVID-19; Encoder-decoder; Lung CT; Lung segmentation; Machine learning; Simple segmentation;
D O I
10.7717/PEERJ-CS.2178
中图分类号
学科分类号
摘要
This work presents the application of an Encoder-Decoder convolutional neural network (ED-CNN) model to automatically segment COVID-19 computerised tomography (CT) data. By doing so we are producing an alternative model to current literature, which is easy to follow and reproduce, making it more accessible for real-world applications as little training would be required to use this. Our simple approach achieves results comparable to those of previously published studies, which use more complex deep-learning networks. We demonstrate a high-quality automated segmentation prediction of thoracic CT scans that correctly delineates the infected regions of the lungs. This segmentation automation can be used as a tool to speed up the contouring process, either to check manual contouring in place of a peer checking, when not possible or to give a rapid indication of infection to be referred for further treatment, thus saving time and resources. In contrast, manual contouring is a time-consuming process in which a professional would contour each patient one by one to be later checked by another professional. The proposed model uses approximately 49 k parameters while others average over 1,000 times more parameters. As our approach relies on a very compact model, shorter training times are observed, which make it possible to easily retrain the model using other data and potentially afford “personalised medicine” workflows. The model achieves similarity scores of Specificity (Sp) = 0.996 ± 0.001, Accuracy (Acc) = 0.994 ± 0.002 and Mean absolute error (MAE) = 0.0075 ± 0.0005. © 2024 Newson et al.
引用
收藏
相关论文
共 27 条
[1]  
Abraham B, Nair MS., Computer-aided classification of prostate cancer grade groups from mri images using texture features and stacked sparse autoencoder, Computerized Medical Imaging and Graphics, 69, 10071, pp. 60-68, (2018)
[2]  
Adil MT, Rahman R, Whitelaw D, Jain V, Al-Taan O, Rashid F, Munasinghe A, Jambulingam P., Sars-cov-2 and the pandemic of covid-19, Postgraduate Medical Journal, 97, 1144, pp. 110-116, (2021)
[3]  
Ahishali M, Degerli A, Yamac M, Kiranyaz S, Chowdhury ME, Hameed K, Hamid T, Mazhar R, Gabbouj M., Advance warning methodologies for covid-19 using chest x-ray images, IEEE Access, 9, pp. 41052-41065, (2021)
[4]  
Ai T, Yang Z, Hou H, Zhan C, Chen C, Lv W, Tao Q, Sun Z, Xia L., Correlation of chest ct and rt-pcr testing for coronavirus disease 2019 (covid-19) in China: a report of 1014 cases, Radiology, 296, 2, pp. E32-E40, (2020)
[5]  
Cheng MM, Fan DP., Structure-measure: a new way to evaluate foreground maps, International Journal of Computer Vision, 129, 9, pp. 2622-2638, (2021)
[6]  
Ciotti M, Ciccozzi M, Terrinoni A, Jiang WC, Wang CB, Bernardini S., The covid-19 pandemic, Critical Reviews in Clinical Laboratory Sciences, 57, 6, pp. 365-388, (2020)
[7]  
Das N, Das S., Epoch and accuracy based empirical study for cardiac MRI segmentation using deep learning technique, PeerJ, 11, 6, (2023)
[8]  
Dean RB, Dixon WJ., Simplified statistics for small numbers of observations, Analytical Chemistry, 23, 4, pp. 636-638, (1951)
[9]  
Elharrouss O, Subramanian N, Al-Maadeed S., An encoder–decoder-based method for segmentation of COVID-19 lung infection in CT images, SN Computer Science, 3, 1, (2021)
[10]  
Fan D-P, Cheng M-M, Liu Y, Li T, Borji A., Structure-measure: a new way to evaluate foreground maps, Proceedings of the IEEE International Conference on Computer Vision (ICCV), (2017)