医学影像预训练模型Med3D(附网盘下载)
介绍 两天前腾讯开源了他们的预训练模型Med3D的代码和预训练模型文件: Tencent/MedicalNet github.com 模型的训练方法在论文 Med3D: Transfer Learning for 3D Medical Image Analysis 中进行了详细描述,主要包含以下几点贡献: (1)将多个医学比赛中的数据进行了整合,创建了3DSeg-8数据集,数据中包含不同的模态、不同目标器官和不同的病理情况。 (2)设计了一个3D网络(Med3D),共同训练多种训练数据,产生了3D-ResNet系列预训练模型,可作为backbone用于分割、检测和分类任务。 (3)使用训练好的预训练模型,对多个分割任务进行了测试,通过使用该预训练模型进行迁移学习,相比于从头训练,精度提升范围在3%-20%。 网络训练方法 数据选择 8个分割数据集,来源于不同的模态(MRI和CT),不同的扫描区域(脑、心脏、胰腺、脾脏等),3D中不同的分辨率、不同的像素灰度范围。 数据预处理 :空间和灰度分布归一化。 1)空间归一化:spacing归一化。为防止过插值,对面每一个体素的插值,目标spacing选为该体素对应的区域内的所有体素的中值spacing,以保证体素内目标不会应为过插值而丢失其本身的特征。 2)灰度归一化:不同图像模态灰度范围是不相同的。需要对灰度做归一化。同时