Momentum Contrastive Voxel-wise Representation Learning for Semi-supervised Volumetric Medical Image Segmentation. 2022

Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
Electrical Engineering, Yale University, New Haven, CT USA.

Contrastive learning (CL) aims to learn useful representation without relying on expert annotations in the context of medical image segmentation. Existing approaches mainly contrast a single positive vector (i.e., an augmentation of the same image) against a set of negatives within the entire remainder of the batch by simply mapping all input features into the same constant vector. Despite the impressive empirical performance, those methods have the following shortcomings: (1) it remains a formidable challenge to prevent the collapsing problems to trivial solutions; and (2) we argue that not all voxels within the same image are equally positive since there exist the dissimilar anatomical structures with the same image. In this work, we present a novel Contrastive Voxel-wise Representation Learning (CVRL) method to effectively learn low-level and high-level features by capturing 3D spatial context and rich anatomical information along both the feature and the batch dimensions. Specifically, we first introduce a novel CL strategy to ensure feature diversity promotion among the 3D representation dimensions. We train the framework through bi-level contrastive optimization (i.e., low-level and high-level) on 3D images. Experiments on two benchmark datasets and different labeled settings demonstrate the superiority of our proposed framework. More importantly, we also prove that our method inherits the benefit of hardness-aware property from the standard CL approaches. Codes will be available soon.

UI MeSH Term Description Entries

Related Publications

Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
November 2022, Computers in biology and medicine,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
April 2022, International journal of neural systems,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
July 2023, IEEE transactions on neural networks and learning systems,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
July 2023, Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
October 2023, IEEE journal of biomedical and health informatics,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
April 2024, Computerized medical imaging and graphics : the official journal of the Computerized Medical Imaging Society,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
January 2023, Medical image analysis,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
January 2023, IEEE journal of biomedical and health informatics,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
September 2023, Physics in medicine and biology,
Chenyu You, and Ruihan Zhao, and Lawrence Staib, and James S Duncan
February 2024, IEEE transactions on medical imaging,
Copied contents to your clipboard!