A Heatmap Regression Approach for Right Ventricle Anatomical Landmarks Localization on Echocardiography

Hailiang Wan,Jingjun Gu,Zhi Yu,Shuyi Song
DOI: https://doi.org/10.1109/icipca59209.2023.10257931
2023-01-01
Abstract:In the process of quantitative cardiac analysis based on echocardiography, the localization of right ventricle anatomical landmarks is a critical step. However, manual labeling of these landmarks by professional physicians is a costly, time-consuming, and labor-intensive process that is also prone to both inter-observer and intra-observer variations. Therefore, we propose n heatmap regression approach to automatically locate right ventricle anatomical landmarks on echocardiography. Firstly, we propose an improved U-Net model, which has a stronger ability to extract features and exploit relationships between the coordinates of landmarks. Secondly, we analyze the limitations of L2 loss in our task. Building on this analysis, we propose a new loss function, Weighted Smooth L1 loss, which not only balances foreground and background pixels but also makes the model more robust to noise. Experiments show that the proposed improved U-Net and Weighted Smooth L1 loss can effectively enhance the landmark localization effect of the model.
What problem does this paper attempt to address?