用少量标注点和局部偏好提升电镜图像分割的自适应能力
Prefer-DAS: Learning from Local Preferences and Sparse Prompts for Domain Adaptive Segmentation of Electron Microscopy
- 通过稀疏提示与局部偏好对齐实现弱监督训练
- 在四个任务上优于同类方法,接近甚至超过有监督模型
- 支持全/部分/无提示交互式分割,适用场景更灵活
领域自适应分割(DAS)是一种在无需每个领域大量标注数据的情况下,从大规模电子显微镜(EM)图像中分割细胞内结构的有前景范式。然而,现有无监督域适应(UDA)方法常表现有限且存在偏差,限制了实际应用。本研究探索在目标域中使用稀疏点和局部人类偏好作为弱标签,提出更贴近现实且标注成本低的新设置。我们开发了Prefer-DAS,首次实现稀疏可提示学习与局部偏好对齐。该模型为可提示多任务架构,融合自训练与提示引导对比学习。不同于SAM类方法仅支持完整提示,Prefer-DAS在训练和推理阶段均可使用完整、部分甚至无提示点,实现交互式分割。我们引入局部直接偏好优化(LPO),以空间变化的人类反馈进行插件式对齐;针对可能缺失反馈的情况,还设计了无监督偏好优化(UPO),利用自学习偏好。结果表明,Prefer-DAS可根据提示与偏好可用性,有效执行弱监督与无监督DAS。在四个挑战性DAS任务上的全面实验显示,其在自动与交互分割模式下均优于SAM类方法及现有无监督/弱监督DAS方法,展现强大泛化性与灵活性。性能接近甚至超越有监督模型。
原文摘要 · Abstract (English)
Domain adaptive segmentation (DAS) is a promising paradigm for delineating intracellular structures from various large-scale electron microscopy (EM) without incurring extensive annotated data in each domain. However, the prevalent unsupervised domain adaptation (UDA) strategies often demonstrate limited and biased performance, which hinders their practical applications. In this study, we explore sparse points and local human preferences as weak labels in the target domain, thereby presenting a more realistic yet annotation-efficient setting. Specifically, we develop Prefer-DAS, which pioneers sparse promptable learning and local preference alignment. The Prefer-DAS is a promptable multitask model that integrates self-training and prompt-guided contrastive learning. Unlike SAM-like methods, the Prefer-DAS allows for the use of full, partial, and even no point prompts during both training and inference stages and thus enables interactive segmentation. Instead of using image-level human preference alignment for segmentation, we introduce Local direct Preference Optimization (LPO), plug-and-play solutions for alignment with spatially varying human feedback. To address potential missing feedback, we also introduce Unsupervised Preference Optimization (UPO), which leverages self-learned preferences. As a result, the Prefer-DAS model can effectively perform both weakly-supervised and unsupervised DAS, depending on the availability of points and human preferences. Comprehensive experiments on four challenging DAS tasks demonstrate that our model outperforms SAM-like methods as well as unsupervised and weakly-supervised DAS methods in both automatic and interactive segmentation modes, highlighting strong generalizability and flexibility. Additionally, the performance of our model is very close to or even exceeds that of supervised models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。