arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.02416cs.CV
收藏

像医学住院医师一样训练:面向通用医学图像分割的上下文先验学习

Training Like a Medical Resident: Context-Prior Learning Toward Universal Medical Image Segmentation

Rutgers University(罗格斯大学);Shanghai AI Laboratory(上海人工智能实验室)

Yunhe Gao, Zhuowei Li, Di Liu, Mu Zhou, Shaoting Zhang, Dimitris N. Metaxas

AI总结:

本文提出 Hermes 上下文先验学习方法,利用跨模态、部位和任务的数据协同训练通用医学图像分割模型,在11个数据集上达到SOTA并验证了迁移、增量学习与泛化能力。

摘要:

临床影像工作流程的一个主要重点是疾病诊断和管理,这导致医学影像数据集与特定临床目标紧密绑定。这种情况促成了开发任务专用分割模型的普遍做法,而未能从广泛的影像队列中获得洞见。受医学放射科住院医师培训项目的启发,我们提出向通用医学图像分割转变,该范式旨在通过利用不同临床目标、身体部位和成像模态之间的多样性与共性,构建医学图像理解基础模型。为实现这一目标,我们开发了 Hermes,这是一种新颖的上下文先验学习方法,用于应对医学图像分割中数据异质性和标注差异的挑战。在涵盖五种模态(CT、PET、T1、T2 和 cine MRI)及多个身体部位的十一个多样化数据集(2,438 幅三维图像)的大型集合中,我们证明了通用范式相比传统范式在单一模型内处理多个任务方面的优势。通过利用任务间的协同作用,Hermes 在所有测试数据集上取得了最先进性能,并展现出卓越的模型可扩展性。在另外两个数据集上的结果揭示了 Hermes 在迁移学习、增量学习以及向下游任务泛化方面的强劲表现。Hermes 学到的先验展现出一种颇具吸引力的特性,能够反映任务与模态之间的复杂关系,这与放射学中已确立的解剖和成像原则相一致。代码可在 https://github.com/yhygao/universal-medical-image-segmentation 获取。

英文摘要:

A major focus of clinical imaging workflow is disease diagnosis and management, leading to medical imaging datasets strongly tied to specific clinical objectives. This scenario has led to the prevailing practice of developing task-specific segmentation models, without gaining insights from widespread imaging cohorts. Inspired by the training program of medical radiology residents, we propose a shift towards universal medical image segmentation, a paradigm aiming to build medical image understanding foundation models by leveraging the diversity and commonality across clinical targets, body regions, and imaging modalities. Towards this goal, we develop Hermes, a novel context-prior learning approach to address the challenges of data heterogeneity and annotation differences in medical image segmentation. In a large collection of eleven diverse datasets (2,438 3D images) across five modalities (CT, PET, T1, T2 and cine MRI) and multiple body regions, we demonstrate the merit of the universal paradigm over the traditional paradigm on addressing multiple tasks within a single model. By exploiting the synergy across tasks, Hermes achieves state-of-the-art performance on all testing datasets and shows superior model scalability. Results on two additional datasets reveals Hermes' strong performance for transfer learning, incremental learning, and generalization to downstream tasks. Hermes's learned priors demonstrate an appealing trait to reflect the intricate relations among tasks and modalities, which aligns with the established anatomical and imaging principles in radiology. The code is available: https://github.com/yhygao/universal-medical-image-segmentation.

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

↑