Confidence-Aware Teacher-Student Distillation for 3D Medical Segmentation

📅 2026-09-20
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
本文针对3D医学图像分割中注释稀疏的问题,提出了一种基于点提示的学生-教师框架,并通过置信度感知优化策略训练3D学生网络以提高分割精度。
📝 Abstract
Medical image segmentation models typically rely on large amounts of densely annotated volumetric data, limiting their scalability across tasks and imaging modalities. This work addresses the challenge of predicting entire 3D anatomical structures from extreme annotation sparsity. An annotation-efficient student-teacher framework is proposed for automatic 3D medical segmentation that requires only a set of point prompts on a single 2D slice per volume, as input. A foundation model serves as an offline teacher, utilizing the provided point prompts from the selected slice to full-volume pseudo-annotations alongside their corresponding spatial confidence scores prior to student training. To mitigate the error propagation of noisy pseudo-annotations, a task-specific 3D student network is trained using a confidence-aware optimization strategy. By leveraging the teacher's pre-computed confidence scores, this strategy explicitly excludes statically uncertain regions of the pseudo-annotations from the loss calculation, while simultaneously emphasizing regions with higher confidence. Evaluated on 3D cardiac MRI datasets, our framework outperforms state-of-the-art semi-supervised methods, improving segmentation performance by up to 43.6%. Furthermore, it drastically reduces the manual annotation burden to just a few positive point prompts per volume, while improving surface boundary precision by up to 14.7% over the teacher and successfully recovering up to 34.1% of the performance gap toward the fully supervised upper bound.
Problem

Research questions and friction points this paper is trying to address.

3D Medical Segmentation
Annotation Sparsity
Confidence-Aware
Innovation

Methods, ideas, or system contributions that make the work stand out.

Confidence-Aware Distillation
Point Prompts
Annotation-Efficient Framework
Pseudo-Annotations
3D Medical Segmentation
🔎 Similar Papers
No similar papers found.
💼 Related Jobs
No related jobs found.
G
Georgios Triantafyllou
Department of Computer Science and Biomedical Informatics, University of Thessaly, 2-4 Papasiopoulou st. 35131, Lamia, Greece
D
Dimitris K. Iakovidis
Department of Computer Science and Biomedical Informatics, University of Thessaly, 2-4 Papasiopoulou st. 35131, Lamia, Greece