The key to preventing the COVID-19 is to diagnose patients quickly and accurately.Studies have shown that using Convolutional Neural Networks(CNN)to analyze chest Computed Tomography(CT)images is helpful for timely CO...The key to preventing the COVID-19 is to diagnose patients quickly and accurately.Studies have shown that using Convolutional Neural Networks(CNN)to analyze chest Computed Tomography(CT)images is helpful for timely COVID-19 diagnosis.However,personal privacy issues,public chest CT data sets are relatively few,which has limited CNN’s application to COVID-19 diagnosis.Also,many CNNs have complex structures and massive parameters.Even if equipped with the dedicated Graphics Processing Unit(GPU)for acceleration,it still takes a long time,which is not conductive to widespread application.To solve above problems,this paper proposes a lightweight CNN classification model based on transfer learning.Use the lightweight CNN MobileNetV2 as the backbone of the model to solve the shortage of hardware resources and computing power.In order to alleviate the problem of model overfitting caused by insufficient data set,transfer learning is used to train the model.The study first exploits the weight parameters trained on the ImageNet database to initialize the MobileNetV2 network,and then retrain the model based on the CT image data set provided by Kaggle.Experimental results on a computer equipped only with the Central Processing Unit(CPU)show that it consumes only 1.06 s on average to diagnose a chest CT image.Compared to other lightweight models,the proposed model has a higher classification accuracy and reliability while having a lightweight architecture and few parameters,which can be easily applied to computers without GPU acceleration.Code:github.com/ZhouJie-520/paper-codes.展开更多
基金This work was supported,in part,by the Natural Science Foundation of Jiangsu Province under Grant Numbers BK20201136,BK20191401in part,by the National Nature Science Foundation of China under Grant Numbers 61502240,61502096,61304205,61773219in part,by the Priority Academic Program Development of Jiangsu Higher Education Institutions(PAPD)fund.
文摘The key to preventing the COVID-19 is to diagnose patients quickly and accurately.Studies have shown that using Convolutional Neural Networks(CNN)to analyze chest Computed Tomography(CT)images is helpful for timely COVID-19 diagnosis.However,personal privacy issues,public chest CT data sets are relatively few,which has limited CNN’s application to COVID-19 diagnosis.Also,many CNNs have complex structures and massive parameters.Even if equipped with the dedicated Graphics Processing Unit(GPU)for acceleration,it still takes a long time,which is not conductive to widespread application.To solve above problems,this paper proposes a lightweight CNN classification model based on transfer learning.Use the lightweight CNN MobileNetV2 as the backbone of the model to solve the shortage of hardware resources and computing power.In order to alleviate the problem of model overfitting caused by insufficient data set,transfer learning is used to train the model.The study first exploits the weight parameters trained on the ImageNet database to initialize the MobileNetV2 network,and then retrain the model based on the CT image data set provided by Kaggle.Experimental results on a computer equipped only with the Central Processing Unit(CPU)show that it consumes only 1.06 s on average to diagnose a chest CT image.Compared to other lightweight models,the proposed model has a higher classification accuracy and reliability while having a lightweight architecture and few parameters,which can be easily applied to computers without GPU acceleration.Code:github.com/ZhouJie-520/paper-codes.
