arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.06384cs.CVcs.AI

基于提示组的鲁棒文本引导核分割训练

Prompt Group-Aware Training for Robust Text-Guided Nuclei Segmentation

Fudan University, Shanghai, China(复旦大学,上海,中国)

Yonghuang Wu, Zhenyang Liang, Wenwen Zeng, Xuan Xie, Jinhua Yu

AI总结:

本文提出基于提示组的鲁棒文本引导核分割训练方法,通过组内一致性约束和质量引导正则化提升分割鲁棒性和泛化能力。

摘要:

基础模型如Segment Anything Model 3(SAM3)能够实现灵活的文本引导医学图像分割,但其预测结果仍然对提示语的表述高度敏感。即使语义等价的描述也能产生不一致的掩码,限制了临床和病理工作流程中的可靠性。我们重新将提示敏感性视为一个组内一致性问题。语义相关的提示被组织成共享相同真实掩码的提示组,并引入一种提示组感知的训练框架,以实现鲁棒的文本引导核分割。该方法结合(i)利用分割损失作为隐含排序信号的质量引导组正则化,以及(ii)一种带有停止梯度策略的logit层面一致性约束,以在每组内对齐预测。该方法不需要架构修改,且保持推理不变。在多数据集核分割基准上的广泛实验显示,在文本提示下实现了稳定的提升,并在提示质量水平上显著减少了性能波动。在六个零样本跨数据集任务中,我们的方法平均将Dice值提高了2.16个百分点。这些结果展示了在计算病理学中的视觉-语言分割中改进的鲁棒性和泛化能力。

英文摘要:

Foundation models such as Segment Anything Model 3 (SAM3) enable flexible text-guided medical image segmentation, yet their predictions remain highly sensitive to prompt formulation. Even semantically equivalent descriptions can yield inconsistent masks, limiting reliability in clinical and pathology workflows. We reformulate prompt sensitivity as a group-wise consistency problem. Semantically related prompts are organized into \emph{prompt groups} sharing the same ground-truth mask, and a prompt group-aware training framework is introduced for robust text-guided nuclei segmentation. The approach combines (i) a quality-guided group regularization that leverages segmentation loss as an implicit ranking signal, and (ii) a logit-level consistency constraint with a stop-gradient strategy to align predictions within each group. The method requires no architectural modification and leaves inference unchanged. Extensive experiments on multi-dataset nuclei benchmarks show consistent gains under textual prompting and markedly reduced performance variance across prompt quality levels. On six zero-shot cross-dataset tasks, our method improves Dice by an average of 2.16 points. These results demonstrate improved robustness and generalization for vision-language segmentation in computational pathology.

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

↑