Improving Medical X-Ray Imaging Diagnosis With Attention Mechanisms and Robust Transfer Learning Techniques

X-ray imaging remains a cornerstone in medical diagnostics for conditions such as bone fractures, knee osteoarthritis, and lung diseases. However, variability in image quality and dataset diversity presents significant challenges for automated analysis using deep learning models. This study addresses these issues by proposing an EfficientNet B0 architecture enhanced with a Convolutional Block Attention Module (CBAM) to improve classification accuracy and interpretability across multiple X-ray datasets: FracAtlas, Knee, and Lung X-ray. A robust preprocessing pipeline comprising LAB color space conversion, morphological filtering, gamma correction, Non-Local Means denoising, resizing, and normalization was applied to optimize image quality, with each step’s effectiveness verified through established image quality metrics. Additionally, geometric augmentation techniques were performed to increase dataset variability and improve model generalization. Comparative experiments with transfer learning, transformer-based, and attention-based models identified the attention-based EfficientNet B0 as the best performer. An extensive ablation study on the Knee X-ray dataset refined hyperparameters to maximize performance. The optimized model achieved high test accuracies of 98.09%, 97.12%, and 99.51% on FracAtlas, Knee, and Lung datasets, respectively. Further analyses, including noise robustness testing, k-fold cross-validation, and Grad-CAM visualization, demonstrated the model’s resilience, consistency, and explainability. The findings highlight the critical role of attention mechanisms in enhancing feature representation and generalization across heterogeneous medical imaging tasks. This work lays a foundation for reliable, interpretable AI systems that can support clinical decision-making, with future efforts focusing on expanding dataset diversity and real-world clinical validation to accelerate adoption.

Authors

Institutions

Publication Details

Journal
IEEE Access
Published
2025-01-01
DOI
https://doi.org/10.1109/access.2025.3607639
Citations
10
Primary Topic
Brain Tumor Detection and Classification
Type
article
Field-Weighted Citation Impact
8.42
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Improving Medical X-Ray Imaging Diagnosis With Attention Mechanisms and Robust Transfer Learning Techniques

Shahab Abdulla, Sadika Zaman, Md. Mehedi Hassan, Md. Alif Sheakh et al.
10 citations
IEEE Access
Brain Tumor Detection and Classification
8.42
article

Improving Medical X-Ray Imaging Diagnosis With Attention Mechanisms and Robust Transfer Learning Techniques

Shahab Abdulla, Sadika Zaman, Md. Mehedi Hassan, Md. Alif Sheakh, Mst. Sazia Tahosin, Abhay Shukla, Indrajit Das
article en
10 citations

Abstract

X-ray imaging remains a cornerstone in medical diagnostics for conditions such as bone fractures, knee osteoarthritis, and lung diseases. However, variability in image quality and dataset diversity presents significant challenges for automated analysis using deep learning models. This study addresses these issues by proposing an EfficientNet B0 architecture enhanced with a Convolutional Block Attention Module (CBAM) to improve classification accuracy and interpretability across multiple X-ray datasets: FracAtlas, Knee, and Lung X-ray. A robust preprocessing pipeline comprising LAB color space conversion, morphological filtering, gamma correction, Non-Local Means denoising, resizing, and normalization was applied to optimize image quality, with each step’s effectiveness verified through established image quality metrics. Additionally, geometric augmentation techniques were performed to increase dataset variability and improve model generalization. Comparative experiments with transfer learning, transformer-based, and attention-based models identified the attention-based EfficientNet B0 as the best performer. An extensive ablation study on the Knee X-ray dataset refined hyperparameters to maximize performance. The optimized model achieved high test accuracies of 98.09%, 97.12%, and 99.51% on FracAtlas, Knee, and Lung datasets, respectively. Further analyses, including noise robustness testing, k-fold cross-validation, and Grad-CAM visualization, demonstrated the model’s resilience, consistency, and explainability. The findings highlight the critical role of attention mechanisms in enhancing feature representation and generalization across heterogeneous medical imaging tasks. This work lays a foundation for reliable, interpretable AI systems that can support clinical decision-making, with future efforts focusing on expanding dataset diversity and real-world clinical validation to accelerate adoption.

IEEE AccessVol. 13
Chittagong University of Engineering & Technology (BD), Institute of Management Technology (IN), University of South Australia (AU), University of Southern Queensland (AU), Daffodil International University (BD), Indian Institute of Technology Kanpur (IN)
Openalex Percentile: Top 2%
Brain Tumor Detection and Classification
8.42
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.