arXiv reaDer
多発性硬化症病変セグメンテーションのための深層学習法のより広い一般化に向けて
Towards broader generalization of deep learning methods for multiple sclerosis lesion segmentation
最近、畳み込みニューラルネットワーク(CNN)に基づくセグメンテーション方法は、自動多発性硬化症(MS)病変セグメンテーションで有望なパフォーマンスを示しました。これらの技術は、管理された評価条件で人間の専門家をも上回っています。ただし、高度に制御されたデータセットで適切に機能するようにトレーニングされた最先端のアプローチでは、目に見えないデータセットからの臨床データを一般化することはできません。セグメンテーション精度の別の改善を提案する代わりに、DeepLesionBrain(DLB)と呼ばれる、ドメインシフトに対して堅牢で、目に見えないデータセットで良好に機能する新しい方法を提案します。この一般化プロパティは、3つの主な貢献から生じます。まず、DLBはコンパクトな3DCNNの大規模なアンサンブルに基づいています。このアンサンブル戦略は、一部の個々のネットワークの一般化の失敗のリスクにもかかわらず、堅牢な予測を保証します。第2に、DLBには、トレーニングデータの特異性(取得プロトコルなど)への依存を減らすための新しい画質データ拡張機能が含まれています。最後に、MS病変のより一般化可能な表現を学習するために、階層的特殊化学習(HSL)を提案します。 HSLは、ローカルに特化したネットワークの初期化としてその重みを使用する前に、脳全体にわたって汎用ネットワークを事前トレーニングすることによって実行されます。この目的により、DLBは、グローバル画像レベルで抽出された一般的な特徴とローカル画像レベルで抽出された特定の特徴の両方を学習します。このペーパーを公開した時点で、DLBは、利用可能なモダリティの半分のみを使用しながら、ISBIチャレンジで公開されたメソッドのトップ3に含まれています。 DLBの一般化は、MSSEG'16、ISBIチャレンジ、および社内データセットでのクロスデータセット実験中に、他の最先端のアプローチとも比較されています。 DLBは、従来の手法よりもセグメンテーションのパフォーマンスと一般化を改善するため、臨床診療により適した堅牢なアプローチを提案します。
Recently, segmentation methods based on Convolutional Neural Networks (CNNs) showed promising performance in automatic Multiple Sclerosis (MS) lesions segmentation. These techniques have even outperformed human experts in controlled evaluation condition. However state-of-the-art approaches trained to perform well on highly-controlled datasets fail to generalize on clinical data from unseen datasets. Instead of proposing another improvement of the segmentation accuracy, we propose a novel method robust to domain shift and performing well on unseen datasets, called DeepLesionBrain (DLB). This generalization property results from three main contributions. First, DLB is based on a large ensemble of compact 3D CNNs. This ensemble strategy ensures a robust prediction despite the risk of generalization failure of some individual networks. Second, DLB includes a new image quality data augmentation to reduce dependency to training data specificity (e.g., acquisition protocol). Finally, to learn a more generalizable representation of MS lesions, we propose a hierarchical specialization learning (HSL). HSL is performed by pre-training a generic network over the whole brain, before using its weights as initialization to locally specialized networks. By this end, DLB learns both generic features extracted at global image level and specific features extracted at local image level. At the time of publishing this paper, DLB is among the Top 3 performing published methods on ISBI Challenge while using only half of the available modalities. DLB generalization has also been compared to other state-of-the-art approaches, during cross-dataset experiments on MSSEG'16, ISBI challenge, and in-house datasets. DLB improves the segmentation performance and generalization over classical techniques, and thus proposes a robust approach better suited for clinical practice.
updated: Mon Dec 14 2020 21:33:53 GMT+0000 (UTC)
published: Mon Dec 14 2020 21:33:53 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト