Attention-enhanced architecture for improved pneumonia detection in chest X-ray images

被引:1
|
作者
Li, Dikai [1 ,2 ]
机构
[1] Shenzhen Technol Univ, Ctr Adv Mat Diagnost Technol, Shenzhen Key Lab Ultraintense Laser & Adv Mat Tech, Lantian Rd, Shenzhen 518118, Guangdong, Peoples R China
[2] Shenzhen Technol Univ, Coll Engn Phys, Lantian Rd, Shenzhen 518118, Guangdong, Peoples R China
关键词
Pneumonia detection; Attention-enhanced architecture; Imbalanced training samples; Medical imaging;
D O I
10.1186/s12880-023-01177-1
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
In this paper, we propose an attention-enhanced architecture for improved pneumonia detection in chest X-ray images. A unique attention mechanism is integrated with ResNet to highlight salient features crucial for pneumonia detection. Rigorous evaluation demonstrates that our attention mechanism significantly enhances pneumonia detection accuracy, achieving a satisfactory result of 96% accuracy. To address the issue of imbalanced training samples, we integrate an enhanced focal loss into our architecture. This approach assigns higher weights to minority classes during training, effectively mitigating data imbalance. Our model's performance significantly improves, surpassing that of traditional approaches such as the pretrained ResNet-50 model. Our attention-enhanced architecture thus presents a powerful solution for pneumonia detection in chest X-ray images, achieving an accuracy of 98%. By integrating enhanced focal loss, our approach effectively addresses imbalanced training sample. Comparative analysis underscores the positive impact of our model's spatial and channel attention modules. Overall, our study advances pneumonia detection in medical imaging and underscores the potential of attention-enhanced architectures for improved diagnostic accuracy and patient outcomes. Our findings offer valuable insights into image diagnosis and pneumonia prevention, contributing to future research in medical imaging and machine learning.
引用
收藏
页数:13
相关论文
共 50 条
  • [41] Pneumonia X-ray detection with anchor-free detection framework and data augmentation
    Yan, Nan
    Tao, Ye
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2023, 33 (04) : 1235 - 1246
  • [42] Optimal threshold selection for segmentation of Chest X-Ray images using opposition-based swarm-inspired algorithm for diagnosis of pneumonia
    Khosla, Tejna
    Verma, Om Prakash
    MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (09) : 27089 - 27119
  • [43] FA-Net: A Fuzzy Attention-aided Deep Neural Network for Pneumonia Detection in Chest X-Rays
    Roy, Ayush
    Bhattacharjee, Anurag
    Oliva, Diego
    Ramos-Soto, Oscar
    Alvarez-Padilla, Francisco J.
    Sarkar, Ram
    2024 IEEE 37TH INTERNATIONAL SYMPOSIUM ON COMPUTER-BASED MEDICAL SYSTEMS, CBMS 2024, 2024, : 338 - 343
  • [44] Wavelet Frequency Separation Attention Network for Chest X-ray Image Super-Resolution
    Yu, Yue
    She, Kun
    Liu, Jinhua
    MICROMACHINES, 2021, 12 (11)
  • [45] An Ensemble-based Approach by Fine-Tuning the Deep Transfer Learning Models to Classify Pneumonia from Chest X-Ray Images
    Venu, Sagar Kora
    ICAART: PROCEEDINGS OF THE 13TH INTERNATIONAL CONFERENCE ON AGENTS AND ARTIFICIAL INTELLIGENCE - VOL 2, 2021, : 390 - 401
  • [46] Optimal threshold selection for segmentation of Chest X-Ray images using opposition-based swarm-inspired algorithm for diagnosis of pneumonia
    Tejna Khosla
    Om Prakash Verma
    Multimedia Tools and Applications, 2024, 83 : 27089 - 27119
  • [47] Fast-YOLO Network Model for X-Ray Image Detection of Pneumonia
    Zhao, Bin
    Chang, Lianjun
    Liu, Zhenyu
    ELECTRONICS, 2025, 14 (05):
  • [48] Pneumonia Detection from Pediatric Lung X-Ray Artificial Neural Networks
    Dundar, Ozguer
    Kocer, Sabri
    JOURNAL OF POLYTECHNIC-POLITEKNIK DERGISI, 2024, 27 (05):
  • [49] Evaluation of Deep Convolutional Generative Adversarial Networks for Data Augmentation of Chest X-ray Images
    Kora Venu, Sagar
    Ravula, Sridhar
    FUTURE INTERNET, 2021, 13 (01) : 1 - 13
  • [50] A multichannel EfficientNet deep learning-based stacking ensemble approach for lung disease detection using chest X-ray images
    Ravi, Vinayakumar
    Acharya, Vasundhara
    Alazab, Mamoun
    CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS, 2023, 26 (02): : 1181 - 1203