TY - JOUR
T1 - Achieving Faster and Smarter Chest X-Ray Classification with Optimized CNNs
AU - Louati, Hassen
AU - Louati, Ali
AU - Mansour, Khalid
AU - Kariri, Elham
N1 - Publisher Copyright:
© 2013 IEEE.
PY - 2025
Y1 - 2025
N2 - X-ray imaging is essential in medical diagnostics, particularly for identifying anomalies like respiratory diseases. However, building accurate and efficient deep learning models for X-ray image classification remains challenging, requiring both optimized architectures and low computational complexity. In this paper, we present a three-stage framework to enhance X-ray image classification using Neural Architecture Search (NAS), Transfer Learning, and Model Compression via filter pruning, specifically targeting the ChestX-Ray14 dataset. First, NAS is employed to automatically discover the optimal convolutional neural network (CNN) architecture tailored to the ChestX-Ray14 dataset, reducing the need for extensive manual tuning. Subsequently, we leverage transfer learning by incorporating pre-trained models, which enhances the model's generalizability and reduces dependency on large volumes of labeled X-ray data. Finally, model compression through filter pruning, driven by evolutionary algorithms, trims redundant parameters to improve computational efficiency while preserving model accuracy. Experimental results demonstrate that this approach not only boosts classification accuracy on the ChestX-Ray14 dataset but also significantly reduces model size, making it suitable for deployment in resource-constrained environments, such as mobile and edge devices. This framework provides a practical, scalable solution to improve both the accuracy and efficiency of medical image classification.
AB - X-ray imaging is essential in medical diagnostics, particularly for identifying anomalies like respiratory diseases. However, building accurate and efficient deep learning models for X-ray image classification remains challenging, requiring both optimized architectures and low computational complexity. In this paper, we present a three-stage framework to enhance X-ray image classification using Neural Architecture Search (NAS), Transfer Learning, and Model Compression via filter pruning, specifically targeting the ChestX-Ray14 dataset. First, NAS is employed to automatically discover the optimal convolutional neural network (CNN) architecture tailored to the ChestX-Ray14 dataset, reducing the need for extensive manual tuning. Subsequently, we leverage transfer learning by incorporating pre-trained models, which enhances the model's generalizability and reduces dependency on large volumes of labeled X-ray data. Finally, model compression through filter pruning, driven by evolutionary algorithms, trims redundant parameters to improve computational efficiency while preserving model accuracy. Experimental results demonstrate that this approach not only boosts classification accuracy on the ChestX-Ray14 dataset but also significantly reduces model size, making it suitable for deployment in resource-constrained environments, such as mobile and edge devices. This framework provides a practical, scalable solution to improve both the accuracy and efficiency of medical image classification.
KW - CNN optimization
KW - NAS
KW - medical diagnostics
KW - model compression
KW - transfer learning
UR - http://www.scopus.com/inward/record.url?scp=85215257259&partnerID=8YFLogxK
U2 - 10.1109/ACCESS.2025.3529206
DO - 10.1109/ACCESS.2025.3529206
M3 - Article
AN - SCOPUS:85215257259
SN - 2169-3536
VL - 13
SP - 10070
EP - 10082
JO - IEEE Access
JF - IEEE Access
ER -