Attention decoupled contrastive learning for semi-supervised segmentation method based on data augmentation

Abstract

Objective. Deep learning algorithms have demonstrated impressive performance by leveraging large labeled data. However, acquiring pixel-level annotations for medical image analysis, especially in segmentation tasks, is both costly and time-consuming, posing challenges for supervised learning techniques. Existing semi-supervised methods tend to underutilize representations of unlabeled data and handle labeled and unlabeled data separately, neglecting their interdependencies. Approach. To address this issue, we introduce the Data-Augmented Attention-Decoupled Contrastive model (DADC). This model incorporates an attention decoupling module and utilizes contrastive learning to effectively distinguish foreground and background, significantly improving segmentation accuracy. Our approach integrates an augmentation technique that merges information from both labeled and unlabeled data, notably boosting network performance, especially in scenarios with limited labeled data. Main results. We conducted comprehensive experiments on the automated breast ultrasound (ABUS) dataset and the results demonstrate that DADC outperforms existing segmentation methods in terms of segmentation performance.