Ideal: Improved Dense Local Contrastive Learning For Semi-Supervised Medical Image Segmentation

• Published in: ICASSP 2023 - 2023 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) pp. 1 - 5
• Main Authors: Basak, Hritam, Chattopadhyay, Soumitri, Kundu, Rohit, Nag, Sayan, Mallipeddi, Rammohan
• Format: Conference Proceeding
• Language: English
• Published: IEEE 04-06-2023
• Subjects: Contrastive learning; Head; Image segmentation; Magnetic resonance imaging; Measurement; MRI; Segmentation; Self-supervised learning; Semi-supervised learning; Source coding; Training
• ISSN: 2379-190X
• DOI: 10.1109/ICASSP49357.2023.10094869

Due to the scarcity of labeled data, Contrastive Self-Supervised Learning (SSL) frameworks have lately shown great potential in several medical image analysis tasks. However, the existing contrastive mechanisms are sub-optimal for dense pixel-level segmentation tasks due to their inability to mine local features. To this end, we extend the concept of metric learning to the segmentation task, using a dense (dis)similarity learning for pre-training a deep encoder network, and employing a semi-supervised paradigm to fine-tune for the downstream task. Specifically, we propose a simple convolutional projection head for obtaining dense pixel-level features, and a new contrastive loss to utilize these dense projections thereby improving the local representations. A bidirectional consistency regularization mechanism involving two-stream model training is devised for the downstream task. Upon comparison, our IDEAL method outperforms the SoTA methods by fair margins on cardiac MRI segmentation. Our source codes are publicly accessible at: https://github.com/Rohit-Kundu/IDEAL-ICASSP23.