用87个数据集测试全身体积CT预训练模型的迁移能力
SegBook: A Simple Baseline and Cookbook for Volumetric Medical Image Segmentation
- 构建87个跨模态、跨目标的数据集评估模型迁移性
- 小样本和大样本微调效果优于中等规模数据集
- 全身体积CT预训练模型可有效迁移到MRI和病灶分割任务
计算机断层扫描(CT)是医学影像中最常用的模态之一,已形成覆盖全身解剖结构的最大公开体积分割数据集。大量全身体积CT图像为预训练强大模型(如监督式训练的STU-Net)提供了可能,以分割多种解剖结构。然而,这些预训练模型在不同下游任务中的迁移条件仍不明确,尤其是跨模态与多样目标的情况。为此,我们收集了87个涵盖不同模态、目标和样本量的公共数据集,构建大规模基准评估全身体积CT预训练模型的迁移能力。采用代表性模型STU-Net及其多尺度版本,在跨模态和跨目标上进行迁移学习。实验表明:(1) 微调时存在数据集规模瓶颈,小规模与大规模数据集均有显著提升,中等规模提升有限;(2) 全身体积CT预训练模型具备良好模态迁移能力,能有效适配如MRI等其他模态;(3) 预训练不仅提升解剖结构检测性能,也支持病灶检测任务,展现出对不同目标任务的适应性。本研究旨在推动体积医学图像分割领域迁移学习的未来发展。
原文摘要 · Abstract (English)
Computed Tomography (CT) is one of the most popular modalities for medical imaging. By far, CT images have contributed to the largest publicly available datasets for volumetric medical segmentation tasks, covering full-body anatomical structures. Large amounts of full-body CT images provide the opportunity to pre-train powerful models, e.g., STU-Net pre-trained in a supervised fashion, to segment numerous anatomical structures. However, it remains unclear in which conditions these pre-trained models can be transferred to various downstream medical segmentation tasks, particularly segmenting the other modalities and diverse targets. To address this problem, a large-scale benchmark for comprehensive evaluation is crucial for finding these conditions. Thus, we collected 87 public datasets varying in modality, target, and sample size to evaluate the transfer ability of full-body CT pre-trained models. We then employed a representative model, STU-Net with multiple model scales, to conduct transfer learning across modalities and targets. Our experimental results show that (1) there may be a bottleneck effect concerning the dataset size in fine-tuning, with more improvement on both small- and large-scale datasets than medium-size ones. (2) Models pre-trained on full-body CT demonstrate effective modality transfer, adapting well to other modalities such as MRI. (3) Pre-training on the full-body CT not only supports strong performance in structure detection but also shows efficacy in lesion detection, showcasing adaptability across target tasks. We hope that this large-scale open evaluation of transfer learning can direct future research in volumetric medical image segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。