Multi-Level Residual Feature Fusion Network for Thoracic Disease Classification in Chest X-Ray Images
Open Access
- 21 April 2023
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Access
- Vol. 11, 40988-41002
Abstract
Motivation: Automated identification of thoracic diseases from chest X-ray images (CXR) is a significant area in computer-aided diagnosis. However, most existing methods have limited ability to extract multi-scale features and accurately capture the spatial location of lesions when dealing with thoracic diseases that exhibit concurrency and large variations in lesion size. Method: Based on the above problems, we propose a multi-level residual feature fusion network (MLRFNet) for classifying thoracic diseases. Our approach can quickly capture receptive field information across different lesion sizes and enhance disease-specific features within the spatial domain on feature maps. The MLRFNet comprises two main components: a feature extractor that learns multi-scale semantic information from chest X-ray images and a multi-level residual feature classifier (MRFC) that refines disease-specific pathological features at spatial locations to reduce interference from irrelevant regions. Additionally, the ECA attention modules connect both components to enable flexible channel-wise focus on critical pathological information. Results: We evaluated the performance of MLRFNet through a series of experiments on two datasets: ChestX-Ray14 and CheXpert. Our results show that MLRFNet achieves an average AUC of 0.853 on the ChestX-Ray14 dataset and 0.904 on the CheXpert dataset. Conclusion: The results of experiments demonstrate that our proposed method works better than the current state-of-the-art baselines. Future work will focus on investigating the interdependencies among labels for thoracic diseases and techniques for model compression.Keywords
Funding Information
- National Natural Science Foundation of China (61471263, 62272337)
- Natural Science Foundation of Tianjin, China (16JCZDJC31100)
- Tianjin University Innovation Foundation (2021XZC-0024)
This publication has 40 references indexed in Scilit:
- Focal Loss for Dense Object Detection2017
- Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization2017
- Densely Connected Convolutional Networks2017
- ChestX-Ray8: Hospital-Scale Chest X-Ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseases2017
- ImageNet classification with deep convolutional neural networksCommunications of the ACM, 2017
- Automatic Skin Lesion Segmentation Using Deep Fully Convolutional Networks With Jaccard DistanceIEEE Transactions on Medical Imaging, 2017
- Deep Residual Learning for Image Recognition2016
- Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer LearningIEEE Transactions on Medical Imaging, 2016
- Going deeper with convolutions2015
- ImageNet: A large-scale hierarchical image database2009