COVID19 Diagnosis using AutoML from 3D CT scans

被引:8
作者
Anwar, Talha
机构
来源
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW 2021) | 2021年
关键词
D O I
10.1109/ICCVW54120.2021.00061
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Coronavirus is a pandemic that affects the respiratory system causing cough, shortness of breath, and death in severe cases. Polymerase chain reaction (PCR) tests are used to diagnose coronavirus. The false-negative rate of these tests is high, so there needs a supporting method for an accurate diagnosis. CT scan provides a detailed examination of the chest to diagnose COVID, but a single CT scan comprises hundreds of slices. Expert and experienced radiologists and pulmonologists can diagnose COVID from these hundreds of slices, but this is very time-consuming. So an automatic artificial intelligence (AI) based method is required to diagnose coronavirus with high accuracy. Developing this AI-based technique requires a lot of resources and time, but once it is developed, it can significantly help the clinicians. This paper used an Automated machine learning (AutoML) technique that requires fewer resources (optimal architecture trials) and time to develop, resulting in the best diagnosis. The AutoML models are trained on 2D slices instead of 3D CT scans, and the predictions on unknown data (slices of CT scan) are aggregated to form a prediction of 3D CT scan. The aggregation process picked the most occurred case, whether COVID or non-COVID from all CT scan slices and labeled the 3D CT scan accordingly. Different thresholds are also used to label COVID or non-COVID 3D CT scans from 2D slices. The approach resulted in accuracy and F1-score of 89% and 88%, respectively. Implementation is available at github.com/talhaanwarch/mia-covid19
引用
收藏
页码:503 / 507
页数:5
相关论文
共 28 条
[1]  
Anwar T., 2020, 2020 IEEE 23 INT MUL, P1, DOI DOI 10.1109/INMIC50486.2020.9318212
[2]   AI-Empowered Computational Examination of Chest Imaging for COVID-19 Treatment: A Review [J].
Deng, Hanqiu ;
Li, Xingyu .
FRONTIERS IN ARTIFICIAL INTELLIGENCE, 2021, 4
[3]  
Gao X, 2021, ARXIV PREPRINT ARXIV
[4]   CT in relation to RT-PCR in diagnosing COVID-19 in The Netherlands: A prospective study [J].
Gietema, Hester A. ;
Zelis, Noortje ;
Nobel, J. Martijn ;
Lambriks, Lars J. G. ;
van Alphen, Lieke B. ;
Lashof, Astrid M. L. Oude ;
Wildberger, Joachim E. ;
Nelissen, Irene C. ;
Stassen, Patricia M. .
PLOS ONE, 2020, 15 (07)
[5]  
Guo J, 2020, J MACH LEARN RES, V21
[6]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[7]  
Howard AG., 2017, ARXIV, V2017
[8]   Searching for MobileNetV3 [J].
Howard, Andrew ;
Sandler, Mark ;
Chu, Grace ;
Chen, Liang-Chieh ;
Chen, Bo ;
Tan, Mingxing ;
Wang, Weijun ;
Zhu, Yukun ;
Pang, Ruoming ;
Vasudevan, Vijay ;
Le, Quoc V. ;
Adam, Hartwig .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :1314-1324
[9]  
Hsu C.-Y., 2021, ARXIV PREPRINT ARXIV
[10]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/CVPR.2018.00745, 10.1109/TPAMI.2019.2913372]