论文标题
对比度不足的MRI细分的学习策略
A Learning Strategy for Contrast-agnostic MRI Segmentation
论文作者
论文摘要
我们提出了一种深度学习策略,该策略首次可以对完全未经处理的大脑MRI扫描进行对比鲜明的语义分割,而无需对新方式进行额外的培训或微调。经典的贝叶斯方法通过无监督的强度模型解决了这种细分问题,但需要大量的计算资源。相反,基于学习的方法可以在测试时很快,但对培训中可用的数据敏感。我们提出的学习方法Synthseg利用一组训练分割(无需强度图像)来生成训练过程中随时形成鲜明对比的合成样本图像。这些样品是使用经典贝叶斯分割框架的生成模型产生的,并随机采样参数,用于外观,变形,噪声和偏置场。由于每个迷你批次都有不同的合成对比度,因此最终网络不会偏向任何MRI对比度。我们在四个数据集上全面评估了我们的方法,其中包括1,000多名受试者和四种类型的MR对比。结果表明,我们的方法成功地将数据中的每个对比度都截然不同,其性能比古典贝叶斯分割略好,并快三个数量级。此外,即使在相同类型的MRI对比度中,与使用真实图像进行训练相比,我们的策略在整个数据集中的推广也明显更好。最后,我们发现,即使不现实的对比,综合的综合范围也会增加神经网络的概括。我们的代码和模型是https://github.com/bbillot/synthseg的开源。
We present a deep learning strategy that enables, for the first time, contrast-agnostic semantic segmentation of completely unpreprocessed brain MRI scans, without requiring additional training or fine-tuning for new modalities. Classical Bayesian methods address this segmentation problem with unsupervised intensity models, but require significant computational resources. In contrast, learning-based methods can be fast at test time, but are sensitive to the data available at training. Our proposed learning method, SynthSeg, leverages a set of training segmentations (no intensity images required) to generate synthetic sample images of widely varying contrasts on the fly during training. These samples are produced using the generative model of the classical Bayesian segmentation framework, with randomly sampled parameters for appearance, deformation, noise, and bias field. Because each mini-batch has a different synthetic contrast, the final network is not biased towards any MRI contrast. We comprehensively evaluate our approach on four datasets comprising over 1,000 subjects and four types of MR contrast. The results show that our approach successfully segments every contrast in the data, performing slightly better than classical Bayesian segmentation, and three orders of magnitude faster. Moreover, even within the same type of MRI contrast, our strategy generalizes significantly better across datasets, compared to training using real images. Finally, we find that synthesizing a broad range of contrasts, even if unrealistic, increases the generalization of the neural network. Our code and model are open source at https://github.com/BBillot/SynthSeg.