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 条
  • [21] EfficientUNet: Modified encoder-decoder architecture for the lung segmentation in chest x-ray images
    Agrawal, Tarun
    Choudhary, Prakash
    EXPERT SYSTEMS, 2022, 39 (08)
  • [22] An improved edge detection algorithm for X-Ray images based on the statistical range
    Bharodiya, Anil K.
    Gonsai, Atul M.
    HELIYON, 2019, 5 (10)
  • [23] Deep Learning on Chest X-ray Images to Detect and Evaluate Pneumonia Cases at the Era of COVID-19
    Hammoudi, Karim
    Benhabiles, Halim
    Melkemi, Mahmoud
    Dornaika, Fadi
    Arganda-Carreras, Ignacio
    Collard, Dominique
    Scherpereel, Arnaud
    JOURNAL OF MEDICAL SYSTEMS, 2021, 45 (07)
  • [24] Deep Learning on Chest X-ray Images to Detect and Evaluate Pneumonia Cases at the Era of COVID-19
    Karim Hammoudi
    Halim Benhabiles
    Mahmoud Melkemi
    Fadi Dornaika
    Ignacio Arganda-Carreras
    Dominique Collard
    Arnaud Scherpereel
    Journal of Medical Systems, 2021, 45
  • [25] Machine-Learning-Enabled Diagnostics with Improved Visualization of Disease Lesions in Chest X-ray Images
    Rahman, Md Fashiar
    Tseng, Tzu-Liang
    Pokojovy, Michael
    McCaffrey, Peter
    Walser, Eric
    Moen, Scott
    Vo, Alex
    Ho, Johnny C.
    DIAGNOSTICS, 2024, 14 (16)
  • [26] Deep Learning Based Pneumonia Infection Classification in Chest X-ray Images Using Convolutional Neural Network Model
    Nayak, Jyoti
    Sahu, Devbrat
    DISTRIBUTED COMPUTING AND OPTIMIZATION TECHNIQUES, ICDCOT 2021, 2022, 903 : 273 - 283
  • [27] A Robust CNN Framework with Dual Feedback Feature Accumulation for Detecting Pneumonia Opacity from Chest X-ray Images
    Alam, Md Jahin
    Ali, Shams Nafisa
    Hasan, Md Zubair
    PROCEEDINGS OF 2020 11TH INTERNATIONAL CONFERENCE ON ELECTRICAL AND COMPUTER ENGINEERING (ICECE), 2020, : 77 - 80
  • [28] Detection and Classification of Lung Disease Using Deep Learning Architecture from X-ray Images
    Kabiraj, Anwesh
    Meena, Tanushree
    Reddy, Pailla Balakrishna
    Roy, Sudipta
    ADVANCES IN VISUAL COMPUTING, ISVC 2022, PT I, 2022, 13598 : 444 - 455
  • [29] Deep learning, reusable and problem-based architectures for detection of consolidation on chest X-ray images
    Behzadi-khormouji, Hamed
    Rostami, Habib
    Salehi, Sana
    Derakhshande-Rishehri, Touba
    Masoumi, Marzieh
    Salemi, Siavash
    Keshavarz, Ahmad
    Gholamrezanezhad, Ali
    Assadi, Majid
    Batouli, Ali
    COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 185
  • [30] FocusCovid: automated COVID-19 detection using deep learning with chest X-ray images
    Tarun Agrawal
    Prakash Choudhary
    Evolving Systems, 2022, 13 : 519 - 533