Publication | Open Access
A deep learning method for classification of chest X-ray images
18
Citations
0
References
2021
Year
Convolutional Neural NetworkMedical Image SegmentationEngineeringMachine LearningDeep Learning ModelDiagnostic ImagingRoc CurveDigital RadiologyImage AnalysisData SciencePattern RecognitionRadiologyHealth SciencesMedical ImagingDeep Learning MethodComputational PathologyMedical Image ComputingDeep LearningRadiographic ImagingDeep Learning MethodsRadiomicsDeep Neural NetworksBiomedical ImagingComputer-aided DiagnosisMedical Image AnalysisLimited Data Learning
Abstract Deep learning techniques have provided new research methods for computer-aided diagnosis, allowing researchers to use deep learning methods to process medical imaging data. Chest X-ray examinations are widely used as a primary screening method for chest diseases. Therefore, it is of great importance to study diagnosis of 14 common pathologies in chest X-ray images using deep learning methods. In this paper, we propose a deep learning model named AM_DenseNet for chest X-ray image classification. The model adopts a dense connection network and adds an attention module after each dense block to optimize the model’s ability to extract features, and finally a Focal Loss function is applied to solve the data imbalance problem. The experiments used chest X-ray images as model input and were trained to output the probabilities of 14 chest pathologies. The Area under the ROC curve (AUC) was used to measure the classification results, and the final average AUC was 0.8537. The experimental results show that the AM_DenseNet model could complete the pathology classification of the chest X-ray images effectively.