Coronary artery stenosis segmentation using U-Net architecture with customised loss function.
Authors
Affiliations (4)
Affiliations (4)
- Department of Software Engineering, Faculty of Computing and IT, University of Sargodha, Sargodha, Pakistan.
- Department of Information Systems, College of Computer and Information Sciences, Princess Nourah Bint Abdulrahman University, P.O. Box 84428, 11671, Riyadh, Saudi Arabia. [email protected].
- Department of Computer Science and IT, University of RASUL, Mandi Bahauddin, Pakistan.
- Department of Computer Science, University of Gujrat, Gujrat, Pakistan.
Abstract
The disorders that affect our heart and blood vessels are cardiovascular disorders, and they are the leading cause of death worldwide. A significant disorder among these is coronary artery stenosis. Stenosis detection is a time-consuming process that requires an expert cardiologist and is also prone to human error. A fully automated system can handle all these challenges. Therefore, we presented a deep learning-based framework for binary stenosis segmentation in the coronary arteries using an X-ray angiography dataset. In this research work, three architectures 1st standard U-Net, then the U-Net enhanced with squeeze-and-excitation blocks, and last the U-Net incorporating dense blocks, where the final configuration achieved the best performance in the stenosis segmentation task. A custom loss function is employed to enhance model performance, utilising the publicly available ARCADE dataset. This dataset comprises X-ray angiography images from 1,500 patients, with 1,000 for training, 300 for validation, and 200 for testing. The training dataset was augmented to address limited data availability and enhance model generalizability. The model achieved a precision of 0.5985, a recall of 0.6319, and an F1 Score of 61.47%, whereas previous research in this challenge has achieved only a 53.4% F1 Score. The experimental results show that our method can achieve promising performance, which successfully segments the stenotic regions under the effect of small vessel size and low contrast.