Pre-training of U-Net Encoder for Improved Keypoint Detection in Transmitral Doppler Imaging

Abas Abdi, Jevgeni Jevsikov, Eman I Alajrami, Isreal Ufumaka, Patricia Fernandes, Nasim DadashiSerej, Darrel P Francis, Massoud Zolgharni

Keywords: Deep Learning, Doppler Imagine, Echocardiography

Abstract: Self-supervised learning enables models to extract meaningful representations from unlabelled data. These representations can then be effectively transferred to supervised learning tasks, often requiring less labelled data compared to traditional approaches. The BT-Unet method leverages the strengths of U-Net for medical image segmentation tasks and is specifically designed to facilitate Barlow twins pre-training of backbone networks, such as ResNet50. However, accurate keypoint detection in medical images remains a challenge. This study investigates the potential of pre-training these backbones with unlabelled, domain-specific medical imagery using the Barlow Twins method to enhance keypoint detection performance in BT-Unet. We hypothesise that pre-training with domain-specific data will lead to more accurate and robust detection of Doppler peak velocities in Mitral Inflow ultrasound images compared to models trained without pre-training.

Submission Number: 142