arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2506.22101cs.CVcs.LGstat.ML

用于少样本医学图像分割的绑定原型模型

Tied Prototype Model for Few-Shot Medical Image Segmentation

Department of Physics and Technology, UiT - The Arctic University of Norway(物理与技术系,UiT - 北极大学);Norwegian Centre for Clinical Artificial Intelligence (SPKI)(挪威临床人工智能中心);University Hospital of North-Norway(北挪威大学医院);Norwegian Computing Center(挪威计算中心)

Hyeongji Kim, Stine Hansen, Michael Kampffmeyer

AI总结:

针对ADNet在少样本医学图像分割中的原型单一、二分类局限及阈值固定问题,提出绑定原型模型(TPM),通过绑定原型位置实现多原型与多类分割,并利用类先验优化自适应阈值,有效提升了分割精度。

摘要:

常见的基于原型的医学图像少样本分割(FSS)方法使用特定类的原型对前景类和背景类进行建模。然而,鉴于背景的高变异性,一个更有前景的方向是仅关注前景建模,将背景视为异常——这是ADNet引入的一种方法。但是,ADNet面临三个关键局限:每个类依赖单一原型、侧重于二分类,以及固定阈值无法适应患者和器官的变异性。为了解决这些不足,我们提出了绑定原型模型(TPM),这是对ADNet的原则性重构,为前景和背景分布设定了绑定的原型位置。基于其概率基础,TPM自然地扩展到多原型和多类分割,同时有效分离非典型背景特征。值得注意的是,这两种扩展都带来了分割精度的提升。最后,我们利用自然存在的类先验来定义自适应阈值的理想目标,从而提升分割性能。综上所述,TPM为医学图像分割中基于原型的FSS提供了新的视角。代码可在https://github.com/hjk92g/TPM-FSS获取。

英文摘要:

Common prototype-based medical image few-shot segmentation (FSS) methods model foreground and background classes using class-specific prototypes. However, given the high variability of the background, a more promising direction is to focus solely on foreground modeling, treating the background as an anomaly -- an approach introduced by ADNet. Yet, ADNet faces three key limitations: dependence on a single prototype per class, a focus on binary classification, and fixed thresholds that fail to adapt to patient and organ variability. To address these shortcomings, we propose the Tied Prototype Model (TPM), a principled reformulation of ADNet with tied prototype locations for foreground and background distributions. Building on its probabilistic foundation, TPM naturally extends to multiple prototypes and multi-class segmentation while effectively separating non-typical background features. Notably, both extensions lead to improved segmentation accuracy. Finally, we leverage naturally occurring class priors to define an ideal target for adaptive thresholds, boosting segmentation performance. Taken together, TPM provides a fresh perspective on prototype-based FSS for medical image segmentation. The code can be found at https://github.com/hjk92g/TPM-FSS.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑