Semantic Segmentation with Attention Dense U-Net for Lung Extraction from X-ray Images

Deep learning and digital image processing are crucial in medical imaging research. Lung segmentation is particularly challenging, demanding accurate differentiation of complex structures, sophisticated algorithms, and deep learning models for reliable results. In our research, We used the Shen-zhen chest X-ray dataset, comprising 566 frontal chest X-rays focused on pulmonary tuberculosis. This dataset, compiled by the Guangdong Medical College and Shenzhen No. 3 People's Hospital, was released by the United States National Library of Medicine. Images, captured using a Philips DR Digital Diagnostic system, were resized from
$3000 \times 3000$
pixels to
$512 \times 512$
pixels for computer-aided diagnosis research. We initially used the Attention Dense U-Net to extract lungs from chest X-rays, leveraging attention mechanisms to focus on relevant features and dense blocks to improve feature reuse. Our goal is to assess its performance for chest X-ray segmentation, comparing it with other deep learning models due to limited research in this area. So, the model is compared with four other variants of U-Net architectures to segment lung pixels from an X-ray image. After implementing the approach using a benchmark dataset and comparing it to other existing architectures, we present that Attention Dense U-Net gives the best accuracy for all given parameters, with a result of accuracy: 97.48%, Dice coefficient: 94.87%, and IoU: 93.87%.