Research on Chest X-ray Image Recognition and Classification Model based on Deep Learning
10.11783/j.issn.1002-3674.2024.03.008
- VernacularTitle:基于深度学习的胸部X射线图像识别及分类模型研究
- Author:
Xiaoxi ZHANG
1
,
2
;
Yuanhan WANG
;
Qingluo YANG
Author Information
1. 山西医科大学公共卫生学院卫生统计学教研室(030000)
2. 重大疾病风险评估山西省重点实验室
- Keywords:
Lung disease;
Model building;
Deep learning;
Image recognition
- From:
Chinese Journal of Health Statistics
2024;41(3):365-369,375
- CountryChina
- Language:Chinese
-
Abstract:
Objectives Building a chest X-ray(CXR)image classification model based on convolutional neural networks in deep learning,providing reliable auxiliary diagnostic techniques for lung diseases.Methods Four kinds of chest X-ray pictures of COVID-19,mild pulmonary infection,viral pneumonia and normal were collected through KAGGLE database,and the data were randomly divided into training set,test set and verification set according to 3∶1∶1 ratio.Building a CXR image classification model based on convolutional neural network architecture,adjusting hyperparameters to strengthen and optimize the model.Subsequently,the model was validated and evaluated using metrics such as confusion matrix,accuracy,sensitivity,and K-fold cross validation results.Results The classification accuracy of this research model for lung imaging images is 0.81,the sensitivity is 0.80,and the loss values of the test and validation sets can be stable at a relatively low level.Compared with models with the same migration algorithm,the accuracy,sensitivity,and F1 score on the test dataset were improved by 1.7%,1.7%,1.3%,and 2.9%,respectively.Conclusion This model has stronger recognition and classification performance for CXR images,and can be more effectively applied to auxiliary analysis and judgment of lung diseases.
