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

通过迭代人在环更新的医学图像分割快速模型转换:从CT中多器官分割的标记公共数据集到未标记临床数据集

Rapid model transfer for medical image segmentation via iterative human-in-the-loop update: from labelled public to unlabelled clinical datasets for multi-organ segmentation in CT

The Chinese University of Hong Kong(香港中文大学);The First Hospital of Jilin University(吉林大学第一医院)

Wenao Ma, Shuang Zheng, Lei Zhang, Huimao Zhang, Qi Dou

摘要:

尽管深度学习在医学图像分析方面取得了显着的成功,但如何将AI模型从一个数据集快速转移到另一个数据集以用于临床应用仍在探索中。本文提出了一种新的和通用的人在循环计划,有效地将分割模型从一个小规模的标记数据集转移到一个大规模的未标记的数据集在CT多器官分割。为了实现这一点,我们建议使用一个点火器网络,它可以从一个小规模的标记数据集学习,并生成粗略的注释来启动人机交互的过程。然后,我们为我们的大规模数据集使用一个sustainer网络,并在新的注释数据上迭代更新它。此外,我们提出了一个灵活的标签策略的注释器,以减少初始的注释工作量。报告和分析了在我们的私人数据集上评估的每个主题的模型性能和注释的时间成本。实验结果表明,该方法不仅能在Dice上提高性能19.7%,而且在模型传输过程中,每个CT体积的人工标记时间从13.87 min缩短到1.51 min,具有良好的临床应用前景。

英文摘要:

Despite the remarkable success on medical image analysis with deep learning, it is still under exploration regarding how to rapidly transfer AI models from one dataset to another for clinical applications. This paper presents a novel and generic human-in-the-loop scheme for efficiently transferring a segmentation model from a small-scale labelled dataset to a larger-scale unlabelled dataset for multi-organ segmentation in CT. To achieve this, we propose to use an igniter network which can learn from a small-scale labelled dataset and generate coarse annotations to start the process of human-machine interaction. Then, we use a sustainer network for our larger-scale dataset, and iteratively updated it on the new annotated data. Moreover, we propose a flexible labelling strategy for the annotator to reduce the initial annotation workload. The model performance and the time cost of annotation in each subject evaluated on our private dataset are reported and analysed. The results show that our scheme can not only improve the performance by 19.7% on Dice, but also expedite the cost time of manual labelling from 13.87 min to 1.51 min per CT volume during the model transfer, demonstrating the clinical usefulness with promising potentials.

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

↑