JCM, Vol. 14, Pages 7971: Sustainable Ultralightweight U-Net-Based Architecture for Myocardium Segmentation

Journal of Clinical Medicine doi: 10.3390/jcm14227971

Authors:
Jakub Filarecki
Dorota Mockiewicz
Agata Giełczyk
Tamara Kuźba-Kryszak
Roman Makarewicz
Marek Lewandowski
Zbigniew Serafin

Background: Medical image segmentation is essential for accurate diagnosis and treatment planning. The U-Net architecture is widely regarded as the gold standard, yet its large size and high computational demand pose significant challenges for practical deployment. Methods: Real data (MRI images) from hospital patients were used in this study. We proposed a novel lightweight architecture tailored specifically for myocardium (cardiac muscle) segmentation. Results: We presented results comparable to state-of-the-art methods in terms of IoU and Dice coefficients. Nonetheless, the results achieved are much more favorable from the perspective of AI’s sustainable development. The proposed architecture ensured the following average results: IOU = 0.7889 and Dice = 0.8780 using 263 k parameters and a total of 6.24 G FLOPs. Conclusions: The proposed schema can potentially be used to support radiologists in improving the diagnostic process. The presented approach is efficient and fast. Most promisingly, the reduction in the model’s complexity is significant compared to the state-of-the-art methods.

Source link

Jakub Filarecki www.mdpi.com