PRISM: Past-Regularized Iterative Self-Distillation With Momentum for Polyp Segmentation

PRISM:基于动量过去正则化的迭代自蒸馏算法用于息肉分割

阅读:1

Abstract

Polyps are abnormal tissue growths in the colon that may develop into colorectal cancer if left undetected. Accurate segmentation in medical imaging is essential for early diagnosis and treatment. Although deep learning has greatly improved polyp segmentation, its dependence on large annotated datasets and substantial computational resources hampers generalization across diverse clinical settings. To overcome these challenges, we propose PRISM, a momentum-based self-distillation method that improves segmentation performance without introducing additional inference cost. Instead of storing or reusing past predictions, PRISM constructs a temporally smoothed teacher model by applying an exponential moving average (EMA) to the model's weights throughout training. This momentum-based teacher provides stable and adaptive supervision signals that co-evolve with the student model. We evaluate PRISM on colonoscopy datasets collected from five distinct medical centres and validate its generalization on an unseen independent dataset. PRISM achieves a Dice score of 0.81 and an IoU of 0.75, outperforming baseline and conventional self-distillation methods. Ablation studies confirm the effectiveness of the EMA-based teacher model in improving segmentation accuracy. PRISM offers a computationally efficient and generalizable solution for polyp segmentation tasks. The code is available at: https://github.com/TugberkErol/PRISM.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。