COVLIAS 3.0: cloud-based quantized hybrid UNet3+ deep learning for COVID-19 lesion detection in lung computed tomography

被引:1
作者
Agarwal, Sushant [1 ,2 ]
Saxena, Sanjay [3 ]
Carriero, Alessandro [4 ]
Chabert, Gian Luca [5 ]
Ravindran, Gobinath [6 ]
Paul, Sudip [7 ]
Laird, John R. [8 ]
Garg, Deepak [9 ]
Fatemi, Mostafa [10 ]
Mohanty, Lopamudra [11 ,12 ]
Dubey, Arun K. [13 ]
Singh, Rajesh [14 ]
Fouda, Mostafa M. [15 ]
Singh, Narpinder [16 ]
Naidu, Subbaram [17 ]
Viskovic, Klaudija [18 ]
Kukuljan, Melita [19 ]
Kalra, Manudeep K. [20 ]
Saba, Luca [5 ]
Suri, Jasjit S. [15 ,21 ,22 ,23 ]
机构
[1] GBTI, Adv Knowledge Engn Ctr, Roseville, CA 95661 USA
[2] PSIT, Dept CSE, Kanpur, India
[3] IIIT, Dept CSE, Bhubaneswar, India
[4] Univ Piemonte Orientale UPO, Maggiore Carita Hosp, Dept Radiol, Novara, Italy
[5] AOU, Dept Radiol, Cagliari, Italy
[6] SR Univ, Dept Civil Engn, Warangal, Telangana, India
[7] NEHU, Dept Biomed Engn, Shillong, India
[8] Adventist Hlth St Helena, Heart & Vasc Inst, St Helena, CA USA
[9] SR Univ, Sch CS & AI, Warangal, Telangana, India
[10] Mayo Clin, Coll Med & Sci, Dept Physiol & Biomed Engn, Rochester, MN USA
[11] ABES Engn Coll, Dept Comp Sci, Ghaziabad, UP, India
[12] Bennett Univ, Dept Comp Sci, Greater Noida, UP, India
[13] Bharati Vidyapeeths Coll Engn, New Delhi, India
[14] Uttaranchal Univ, Uttaranchal Inst Technol, Div Res & Innovat, Dehra Dun, India
[15] Idaho State Univ, Dept ECE, Pocatello, ID USA
[16] Graph Era Deemed Univ, Dept Food Sci & Technol, Dehra Dun, India
[17] Univ Minnesota, Dept EE, Duluth, MN USA
[18] Univ Hosp Infect Dis, Zagreb, Croatia
[19] Clin Hosp Ctr Rijeka, Dept Intervent & Diagnost Radiol, Rijeka, Croatia
[20] Massachusetts Gen Hosp, Dept Radiol, Boston, MA USA
[21] Graph Era Deemed Univ, Dept Comp Sci & Engn, Dehra Dun, Uttarakhand, India
[22] Symbiosis Int Deemed Univ, Symbiosis Inst Technol, Nagpur Campus, Pune, India
[23] AtheroPoint LLC, Stroke & Monitoring Div, Roseville, CA 95661 USA
来源
FRONTIERS IN ARTIFICIAL INTELLIGENCE | 2024年 / 7卷
基金
英国科研创新办公室;
关键词
COVID-19; computed tomography; COVID lesions; glass ground opacities; segmentation; hybrid deep learning; quantization; TISSUE CHARACTERIZATION; RISK STRATIFICATION; ARTIFICIAL-INTELLIGENCE; ATHEROSCLEROTIC PLAQUE; CHEST CT; ULTRASOUND; ACCURATE; CLASSIFICATION; SEGMENTATION; DIAGNOSIS;
D O I
10.3389/frai.2024.1304483
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Background and novelty: When RT-PCR is ineffective in early diagnosis and understanding of COVID-19 severity, Computed Tomography (CT) scans are needed for COVID diagnosis, especially in patients having high ground-glass opacities, consolidations, and crazy paving. Radiologists find the manual method for lesion detection in CT very challenging and tedious. Previously solo deep learning (SDL) was tried but they had low to moderate-level performance. This study presents two new cloud-based quantized deep learning UNet3+ hybrid (HDL) models, which incorporated full-scale skip connections to enhance and improve the detections. Methodology: Annotations from expert radiologists were used to train one SDL (UNet3+), and two HDL models, namely, VGG-UNet3+ and ResNet-UNet3+. For accuracy, 5-fold cross-validation protocols, training on 3,500 CT scans, and testing on unseen 500 CT scans were adopted in the cloud framework. Two kinds of loss functions were used: Dice Similarity (DS) and binary cross-entropy (BCE). Performance was evaluated using (i) Area error, (ii) DS, (iii) Jaccard Index, (iii) Bland-Altman, and (iv) Correlation plots. Results: Among the two HDL models, ResNet-UNet3+ was superior to UNet3+ by 17 and 10% for Dice and BCE loss. The models were further compressed using quantization showing a percentage size reduction of 66.76, 36.64, and 46.23%, respectively, for UNet3+, VGG-UNet3+, and ResNet-UNet3+. Its stability and reliability were proved by statistical tests such as the Mann-Whitney, Paired t-Test, Wilcoxon test, and Friedman test all of which had a p < 0.001. Conclusion: Full-scale skip connections of UNet3+ with VGG and ResNet in HDL framework proved the hypothesis showing powerful results improving the detection accuracy of COVID-19.
引用
收藏
页数:12
相关论文
共 100 条
  • [1] Symptomatic vs. Asymptomatic Plaque Classification in Carotid Ultrasound
    Acharya, Rajendra U.
    Faust, Oliver
    Alvin, A. P. C.
    Sree, S. Vinitha
    Molinari, Filippo
    Saba, Luca
    Nicolaides, Andrew
    Suri, Jasjit S.
    [J]. JOURNAL OF MEDICAL SYSTEMS, 2012, 36 (03) : 1861 - 1871
  • [2] Diagnosis of Hashimoto's thyroiditis in ultrasound using tissue characterization and pixel classification
    Acharya, U. R.
    Sree, S. Vinitha
    Mookiah, M. R. K.
    Yantri, R.
    Molinari, F.
    Zieleznik, W.
    Malyszek-Tumidajewicz, J.
    Stepien, B.
    Bardales, R. H.
    Witkowska, A.
    Suri, J. S.
    [J]. PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART H-JOURNAL OF ENGINEERING IN MEDICINE, 2013, 227 (07) : 788 - 798
  • [3] Atherosclerotic plaque tissue characterization in 2D ultrasound longitudinal carotid scans for automated classification: a paradigm for stroke risk assessment
    Acharya, U. Rajendra
    Mookiah, Muthu Rama Krishnan
    Sree, S. Vinitha
    Afonso, David
    Sanches, Joao
    Shafique, Shoaib
    Nicolaides, Andrew
    Pedro, L. M.
    Fernandes e Fernandes, J.
    Suri, Jasjit S.
    [J]. MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2013, 51 (05) : 513 - 523
  • [4] Understanding symptomatology of atherosclerotic plaque by image-based tissue characterization
    Acharya, U. Rajendra
    Faust, Oliver
    Sree, Vinitha S.
    Alvin, A. P. C.
    Krishnamurthi, Ganapathy
    Seabra, Jose C. R.
    Sanches, Joao
    Suri, Jasjit S.
    [J]. COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2013, 110 (01) : 66 - 75
  • [5] Acharya UR, 2012, IEEE ENG MED BIO, P4446, DOI 10.1109/EMBC.2012.6346953
  • [6] Acharya UR, 2011, IEEE ENG MED BIO, P4489, DOI 10.1109/IEMBS.2011.6091113
  • [7] Eight pruning deep learning models for low storage and high-speed COVID-19 computed tomography lung segmentation and heatmap-based lesion localization: A multicenter study using COVLIAS 2.0
    Agarwal, Mohit
    Agarwal, Sushant
    Saba, Luca
    Chabert, Gian Luca
    Gupta, Suneet
    Carriero, Alessandro
    Pasche, Alessio
    Danna, Pietro
    Mehmedovic, Armin
    Faa, Gavino
    Shrivastava, Saurabh
    Jain, Kanishka
    Jain, Harsh
    Jujaray, Tanay
    Singh, Inder M.
    Turk, Monika
    Chadha, Paramjit S.
    Johri, Amer M.
    Khanna, Narendra N.
    Mavrogeni, Sophie
    Laird, John R.
    Sobel, David W.
    Miner, Martin
    Balestrieri, Antonella
    Sfikakis, Petros P.
    Tsoulfas, George
    Misra, Durga Prasanna
    Agarwal, Vikas
    Kitas, George D.
    Teji, Jagjit S.
    Al-Maini, Mustafa
    Dhanjil, Surinder K.
    Nicolaides, Andrew
    Sharma, Aditya
    Rathore, Vijay
    Fatemi, Mostafa
    Alizad, Azra
    Krishnan, Pudukode R.
    Yadav, Rajanikant R.
    Nagy, Frence
    Kincses, Zsigmond Tamas
    Ruzsa, Zoltan
    Naidu, Subbaram
    Viskovic, Klaudija
    Kalra, Manudeep K.
    Suri, Jasjit S.
    [J]. COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 146
  • [8] Wilson disease tissue classification and characterization using seven artificial intelligence models embedded with 3D optimization paradigm on a weak training brain magnetic resonance imaging datasets: a supercomputer application
    Agarwal, Mohit
    Saba, Luca
    Gupta, Suneet K.
    Johri, Amer M.
    Khanna, Narendra N.
    Mavrogeni, Sophie
    Laird, John R.
    Pareek, Gyan
    Miner, Martin
    Sfikakis, Petros P.
    Protogerou, Athanasios
    Sharma, Aditya M.
    Viswanathan, Vijay
    Kitas, George D.
    Nicolaides, Andrew
    Suri, Jasjit S.
    [J]. MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2021, 59 (03) : 511 - 533
  • [9] Alqudah A.M., 2020, XRAY IMAGING, V2, P1
  • [10] PCA-based polling strategy in machine learning framework for coronary artery disease risk assessment in intravascular ultrasound: A link between carotid and coronary grayscale plaque morphology
    Araki, Tadashi
    Ikeda, Nobutaka
    Shukla, Devarshi
    Jain, Pankaj K.
    Londhe, Narendra D.
    Shrivastava, Vimal K.
    Banchhor, Sumit K.
    Saba, Luca
    Nicolaides, Andrew
    Shafique, Shoaib
    Laird, John R.
    Suri, Jasjit S.
    [J]. COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2016, 128 : 137 - 158