arXiv reaDer
SynthSeg: Domain Randomisation for Segmentation of Brain MRI Scans of any Contrast and Resolution
データ拡張と転送学習の進歩にもかかわらず、畳み込みニューラルネットワーク(CNN)は、目に見えないターゲットドメインに一般化するのが困難です。脳MRIスキャンのセグメンテーションに適用すると、CNNは解像度とコントラストの変化に非常に敏感です。同じMRモダリティ内であっても、データセット全体でパフォーマンスの低下が観察される可能性があります。 SynthSegを紹介します。これは、あらゆるコントラストと解像度の脳MRIスキャンにとらわれない最初のセグメンテーションCNNです。 SynthSegは、ベイズセグメンテーションに触発された生成モデルからサンプリングされた合成データでトレーニングされています。重要なのは、生成パラメーターを完全にランダム化してトレーニングデータの変動性を最大化するドメインランダム化戦略を採用することです。その結果、SynthSegは、再トレーニングや微調整を行うことなく、任意のターゲットドメインの前処理済みおよび前処理済みの実際のスキャンをセグメント化できます。 SynthSegはセグメンテーションのトレーニングのみを必要とするため(画像なし)、さまざまな母集団の既存のデータセット(萎縮や病変など)から自動的に取得されたラベルマップから学習でき、さまざまな形態学的変動に対する堅牢性を実現します。 SynthSegは、6つのモダリティと10の解像度の5,500スキャンで実証されており、監視対象のCNN、テスト時間の適応、ベイズセグメンテーションと比較して比類のない一般化を示しています。コードとトレーニング済みモデルは、で入手できます。
Despite advances in data augmentation and transfer learning, convolutional neural networks (CNNs) have difficulties generalising to unseen target domains. When applied to segmentation of brain MRI scans, CNNs are highly sensitive to changes in resolution and contrast: even within the same MR modality, decreases in performance can be observed across datasets. We introduce SynthSeg, the first segmentation CNN agnostic to brain MRI scans of any contrast and resolution. SynthSeg is trained with synthetic data sampled from a generative model inspired by Bayesian segmentation. Crucially, we adopt a domain randomisation strategy where we fully randomise the generation parameters to maximise the variability of the training data. Consequently, SynthSeg can segment preprocessed and unpreprocessed real scans of any target domain, without retraining or fine-tuning. Because SynthSeg only requires segmentations to be trained (no images), it can learn from label maps obtained automatically from existing datasets of different populations (e.g., with atrophy and lesions), thus achieving robustness to a wide range of morphological variability. We demonstrate SynthSeg on 5,500 scans of 6 modalities and 10 resolutions, where it exhibits unparalleled generalisation compared to supervised CNNs, test time adaptation, and Bayesian segmentation. The code and trained model are available at
updated: Tue Jul 20 2021 15:22:16 GMT+0000 (UTC)
published: Tue Jul 20 2021 15:22:16 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト