Mar, 2024

多类型肿瘤的分割基础模型

TL;DR通过使用 Resblock-backbone 和 Transformer-bottleneck 构建 16 亿参数的大规模肿瘤分割基础模型 (TSFM),我们利用医学影像中肿瘤和器官之间的空间相关性,创新地融合了 7 个肿瘤数据集和 3 个多器官数据集,构建了一个三维医学数据集池,包括 2779 个病例和 30 万张医学图像,其规模目前超过许多其他公开数据集。TSFM 是用于医学影像分割的预训练模型,也可以用于多个下游任务的微调学习。与 nnU-Net 相比,我们的预训练模型的平均性能高出 2%,在迁移学习任务中,TSFM 只需 nnU-Net 的 5% 训练轮次即可达到相似的性能,并且可以在 10% 的训练轮次上平均超过 nnU-Net 2%。预训练的 TSFM 模型及其代码即将发布。