Enhancing pulmonary embolism diagnosis: a squeeze-and-attention U-Net for precise detection and segmentation in CT angiography.
Authors
Affiliations (7)
Affiliations (7)
- Department of Biomedical Engineering, Faculty of Engineering, University of Isfahan, Isfahan, Iran. Electronic address: [email protected].
- Department of Biomedical Engineering, Faculty of Engineering, University of Isfahan, Isfahan, Iran. Electronic address: [email protected].
- Department of Epidemiology and Biostatistics, School of Health, Isfahan University of Medical Sciences, Isfahan, Iran. Electronic address: [email protected].
- Medical Imaging Research Center, Shiraz University of Medical Sciences, Shiraz, Iran. Electronic address: [email protected].
- Medical Imaging Research Center, Shiraz University of Medical Sciences, Shiraz, Iran. Electronic address: [email protected].
- Division of Nuclear Medicine & Molecular Imaging, Geneva University Hospital, CH-1211 Geneva, Switzerland. Electronic address: [email protected].
- Division of Nuclear Medicine & Molecular Imaging, Geneva University Hospital, CH-1211 Geneva, Switzerland; Department of Nuclear Medicine and Molecular Imaging, University of Groningen, Groningen, Netherlands; Department of Nuclear Medicine, Odense University Hospital, Odense, Denmark; University Research and Innovation Center, Óbuda University, Budapest, Hungary. Electronic address: [email protected].
Abstract
Pulmonary embolism (PE) is a life-threatening condition requiring rapid and accurate diagnosis. This study proposes a deep learning-based approach for automated PE segmentation, focusing on both pixel-level accuracy and clinical applicability. The objective is to enhance segmentation performance by incorporating Squeeze-and-Attention (SA) modules within the network architecture and to evaluate its generalizability on multi-center datasets. A neural network architecture combining convolutional layers, Long Short-Term Memory (LSTM) units, and SA blocks was developed. The model was trained and evaluated using two publicly available PE datasets and two private datasets. Performance was assessed using five-fold cross-validation, hold-out testing, and ablation studies. Evaluation metrics are computed at pixel, slice, and lesion levels, both per-patient and per-slice. The results demonstrated a Dice similarity coefficient of 69.87% at the pixel level and 98.79% sensitivity at the lesion-level on the public dataset, indicating competitive segmentation accuracy. The incorporation of SA blocks significantly improved performance, increasing the Dice score by 7.58% while reducing both false positive and false negative rates. The model showed good generalizability across different imaging centers and scanner types. The proposed model demonstrates acceptable accuracy in PE segmentation and robustness across diverse datasets. Its design and evaluation framework support its potential utility in clinical settings. However, accurate delineation of small emboli remains challenging, and further investigation is required to improve segmentation performance in such cases. Future work will focus on real-time deployment and integration into diagnostic workflows.