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

用于模糊医学图像分割的多评分者提示

Multi-rater Prompting for Ambiguous Medical Image Segmentation

College of Computer Science and Technology, Zhejiang University(浙江大学计算机科学与技术学院);Institute of Wenzhou, Zhejiang University(浙江大学温州研究院);State Key Laboratory of Transvascular Implantation Devices of The Second Affiliated Hospital, Zhejiang University School of Medicine(浙江大学医学院第二附属医院血管植入装置国家重点实验室);University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);University of Notre Dame(圣母大学);School of Public Health Zhejiang University(浙江大学公共卫生学院)

Jinhong Wang, Yi Cheng, Jintai Chen, Hongxia Xu, Danny Chen, Jian Wu

AI总结:

针对多评分者标注的模糊医学图像分割问题,提出基于多评分者提示的方法,通过引入评分者感知提示进行不确定性估计,并采用混合训练策略整合专家共识与分歧,仅需更新0.3%参数即可高效微调,在两个公开数据集上验证了有效性。

摘要:

当医学图像由多位专家(评分者)独立标注时,通常会出现多评分者标注的情况。在本文中,我们解决了医学图像分割中多评分者标注(称为模糊医学图像分割)出现的两个挑战:(1)当一组评分者产生一组多样但合理的标注时,如何训练深度学习模型;(2)当计算资源不足以在不同数据集领域上重新训练整个模型时,如何高效地微调模型。我们提出了一种基于多评分者提示的方法来解决这两个挑战。具体来说,我们引入了一系列评分者感知提示,可以插入U-Net模型中进行不确定性估计,以处理多标注情况。在基于提示的微调过程中,与训练整个模型相比,仅需更新0.3%的可学习参数。此外,为了整合专家共识和分歧,我们探索了不同的多评分者整合策略,并设计了一种混合训练策略以进行全面洞察学习。大量实验验证了我们的新方法在两个公开数据集上对模糊医学图像分割的有效性,同时减轻了模型重新训练的繁重负担。

英文摘要:

Multi-rater annotations commonly occur when medical images are independently annotated by multiple experts (raters). In this paper, we tackle two challenges arisen in multi-rater annotations for medical image segmentation (called ambiguous medical image segmentation): (1) How to train a deep learning model when a group of raters produces a set of diverse but plausible annotations, and (2) how to fine-tune the model efficiently when computation resources are not available for re-training the entire model on a different dataset domain. We propose a multi-rater prompt-based approach to address these two challenges altogether. Specifically, we introduce a series of rater-aware prompts that can be plugged into the U-Net model for uncertainty estimation to handle multi-annotation cases. During the prompt-based fine-tuning process, only 0.3% of learnable parameters are required to be updated comparing to training the entire model. Further, in order to integrate expert consensus and disagreement, we explore different multi-rater incorporation strategies and design a mix-training strategy for comprehensive insight learning. Extensive experiments verify the effectiveness of our new approach for ambiguous medical image segmentation on two public datasets while alleviating the heavy burden of model re-training.

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

↑