Reinforcement learning based diagnosis and prediction for COVID-19 from CT images

Siying Chen; Pan Deng; Minghui Liu; Jiali Deng; Xiaomin Wang; Haigang Gong; Ming Liu

doi:10.1117/12.2626556

20 December 2021 Reinforcement learning based diagnosis and prediction for COVID-19 from CT images

Siying Chen, Pan Deng, Minghui Liu, Jiali Deng, Xiaomin Wang, Haigang Gong, Ming Liu

Proceedings Volume 12155, International Conference on Computer Vision, Application, and Design (CVAD 2021); 121550B (2021) https://doi.org/10.1117/12.2626556
Event: International Conference on Computer Vision, Application, and Design (CVAD 2021), 2021, Sanya, China

Abstract

As we all know, COVID-19 is causing more and more human infections and deaths. In order to quickly and efficiently detect COVID-19, this paper has firstly proposed a detection framework based on reinforcement learning for COVID-19 diagnosis. We use the accuracy of the validation set as the reward value, and obtain the initial model for the next epoch by searching the model corresponding to the maximum reward value in each epoch. We also have proposed a prediction framework that integrates multiple detection frameworks using parameter sharing to predict the progression of patients’ disease. We experimented with our own dataset screened by professional physicians and obtained more excellent results. In external validation, we still achieved a high accuracy rate without additional training. Finally, the experimental results show that our classification accuracy can reach 96.81%, and the precision, sensitivity, specificity, and AUC (Area Under Curve) are 95.47%, 98.64%, 95.91%, and 0.9698, respectively. The accuracy of external verification can reach 93.04% and 90.85%. The accuracy of our prediction framework is 91.04%. A large number of experiments have proved that our proposed method is effective and robust for COVID-19 detection and prediction.

Citation Download Citation

Siying Chen, Pan Deng, Minghui Liu, Jiali Deng, Xiaomin Wang, Haigang Gong, and Ming Liu "Reinforcement learning based diagnosis and prediction for COVID-19 from CT images", Proc. SPIE 12155, International Conference on Computer Vision, Application, and Design (CVAD 2021), 121550B (20 December 2021); https://doi.org/10.1117/12.2626556

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available