SemSim: Revisiting Weak-to-Strong Consistency from a Semantic Similarity Perspective for Semi-supervised Medical Image Segmentation
Not on the map. Why?
Abstract (arXiv)
Semi-supervised learning (SSL) for medical image segmentation is a challenging yet highly practical task, which reduces reliance on large-scale labeled dataset by leveraging unlabeled samples. Among SSL techniques, the weak-to-strong consistency framework, popularized by FixMatch, has emerged as a state-of-the-art method in classification tasks. Notably, such a simple pipeline has also shown competitive performance in medical image segmentation. However, two key limitations still persist, impeding its efficient adaptation: (1) the neglect of contextual dependencies results in inconsistent predictions for similar semantic features, leading to incomplete object segmentation; (2) the lack of exploitation of semantic similarity between labeled and unlabeled data induces considerable class-distribution discrepancy. To address these limitations, we propose a novel semi-supervised framework based on FixMatch, named SemSim, powered by two appealing designs from semantic similarity perspective: (1) rectifying pixel-wise prediction by reasoning about the intra-image pair-wise affinity map, thus integrating contextual dependencies explicitly into the final prediction; (2) bridging labeled and unlabeled data via a feature querying mechanism for compact class representation learning, which fully considers cross-image anatomical similarities. As the reliable semantic similarity extraction depends on robust features, we further introduce an effective spatial-aware fusion module (SFM) to explore distinctive information from multiple scales. Extensive experiments show that SemSim yields consistent improvements over the state-of-the-art methods across three public segmentation benchmarks.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 32 of its 42 references that are in the map. First posted on arXiv 17 Oct 2024.
Builds on
32 of 42 references.
References · 32
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
- 2016Layer Normalizationcited by 5,579
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,845
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,875
- 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,080
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
20 more
- 2019Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)cited by 657
- 2018ADVENT: Adversarial Entropy Minimization for Domain Adaptation in Semantic Segmentationcited by 623
- 2019UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentationcited by 583
- 2021FlexMatch: Boosting Semi-Supervised Learning with Curriculum Pseudo Labelingcited by 537
- 2021Semi-Supervised Semantic Segmentation with Cross Pseudo Supervisioncited by 462
- 2019Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentationcited by 378
- 2020Semi-Supervised Semantic Segmentation with Cross-Consistency Trainingcited by 375
- 2019Label Propagation for Deep Semi-supervised Learningcited by 287
- 2019Interpolation Consistency Training for Semi-Supervised Learningcited by 248
- 2022FreeMatch: Self-adaptive Thresholding for Semi-supervised Learningcited by 232
- 2018Deep Co-Training for Semi-Supervised Image Recognitioncited by 145
- 2023Bidirectional Copy-Paste for Semi-Supervised Medical Image Segmentationcited by 115
- 2020Efficient Semi-Supervised Gross Target Volume of Nasopharyngeal Carcinoma Segmentation via Uncertainty Rectified Pyramid Consistencycited by 102
- 2021Semi-Supervised Medical Image Segmentation via Cross Teaching between CNN and Transformercited by 99
- 2022Exploring Smoothness and Class-Separation for Semi-supervised Medical Image Segmentationcited by 93
- 2023Self-aware and Cross-sample Prototypical Learning for Semi-supervised Medical Image Segmentationcited by 20
- 2024Cross Prompting Consistency with Segment Anything Model for Semi-supervised Medical Image Segmentationcited by 16
- 2024DiffRect: Latent Diffusion Label Rectification for Semi-supervised Medical Image Segmentationcited by 14
- 2022When CNN Meet with ViT: Towards Semi-Supervised Learning for Multi-Class Medical Image Semantic Segmentationcited by 13
- 2023Fast FixMatch: Faster Semi-Supervised Learning with Curriculum Batch Sizecited by 2
Built on by
Not cited yet.