atlas

Research map / Label-efficient and robust learning / Semi-supervised learning

Combinatorial CNN-Transformer Learning with Manifold Constraints for Semi-supervised Medical Image Segmentation

AAAI 2024 · Label-efficient and robust learning › Semi-supervised learning · approach: Semi-supervised medical segmentation · 12 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes M-CnT, a dual-student CNN-Transformer semi-supervised medical segmentation framework with intra-student class-wise manifold losses and inter-student covariance-based knowledge transfer.

Problem

Semi-supervised medical image segmentation suffers from CNN limits in global context and from neglected class-level distribution structure; CNN and Transformer students learn heterogeneous features that are hard to combine with few labels.

Approach

Combines a U-Net (CNN) and Swin-UNet (Transformer) as two students, applies a prototype-based intra-student class-wise aggregation and separation loss in a manifold space, and aligns the two students' class-wise covariance matrices via Frobenius distance, plus an entropy-based selection of pseudo labels from the more confident student.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add pixel-level consistency loss Lpixel (ablation)M-CnT · ACDC and ISIC, 3% labeledDSC: ACDC: 0.510 -> 0.715; ISIC: 0.677 -> 0.734Helped
Add Laggregate to compact each class cluster (ablation)M-CnT · ACDC and ISIC, 3% labeledDSC: ACDC: +1.1%; ISIC: +1.5%Helped
Add Leliminate to increase distance among classes (ablation)M-CnT · ACDC and ISIC, 3% labeledDSC: ACDC: +2.3%; ISIC: +2.5%Helped
Add both Laggregate and Leliminate (ablation)M-CnT · ACDC and ISIC, 3% labeledDSC: ACDC: +2.8%; ISIC: +3.4%Helped
Add inter-student knowledge transfer loss Linter (ablation)M-CnT · ACDC and ISIC, 3% labeledDSCHelped
Use manifold constraints instead of ED, IFVD, or PLCL class-wise losses (ablation)M-CnT · ACDC and ISIC, 3% labeledDSCHelped
Use Gaussian warming-up λ(t) rather than a fixed λm (ablation)M-CnT · ACDC, 3% labeledDSCHelped
Set λp=1 (ablation)M-CnT · ACDC, 3% labeledDSCHelped
Apply manifold constraints at the final decoder stage (ablation)M-CnT · ACDC and ISIC, 3% labeledDSC: ACDC: +3.8%; ISIC: +4.5%Helped

Limitations

Experimental setup

Datasets: ACDC ISIC Synapse

Benchmarks: ACDC ISIC Synapse

Metrics: DSC HD

Baselines: MT UA-MT EM DCT CCT CPS ICT DAN URPC CTCT SSNet ICT-Med

Builds on: Mean Teacher Swin-UNet CTCT multi-manifold attention

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy