arXiv reaDer
3DMRI分類のための連続プロキシメタデータによる対照学習
Contrastive Learning with Continuous Proxy Meta-Data for 3D MRI Classification
ディープニューラルネットワークを使用した従来の教師あり学習では、優れたソリューションに収束するために膨大な量のラベル付きデータが必要です。 3D医用画像の場合、特定の病状に対して大規模で均質な注釈付きデータセットを作成することは、多くの場合非現実的です。自己監視法は、ニューラルネットワークを使用して教師なしの方法で画像の表現を学習する新しい方法を提供します。特に、対照学習は、視覚タスクで完全に監視されたCNNのパフォーマンスを(ほぼ)一致させることにより、大きな期待を示しています。それにもかかわらず、この方法は、事前知識と見なされる参加者の年齢などの利用可能なメタデータを利用しません。ここでは、対照的な学習フレームワークで、y-Aware InfoNCE損失と呼ばれる新しい損失を導入することにより、継続的なプロキシメタデータを活用することを提案します。具体的には、アンカーと同様のプロキシメタデータを持つポジティブな例を追加することで、プレトレーニング中のポジティブサンプリングを改善します。これらは、類似した識別セマンティック機能を共有していると想定しています。この方法では、10 ^ 4マルチで事前トレーニングされた3DCNNモデル-サイトの健康な脳MRIスキャンは、統合失調症、双極性診断、アルツハイマー病の検出という3つの分類タスクに関連する特徴を抽出できます。微調整すると、これらのタスクでゼロからトレーニングされた3D CNNや、最先端の自己監視方式よりも優れたパフォーマンスを発揮します。私たちのコードはここで公開されています。
Traditional supervised learning with deep neural networks requires a tremendous amount of labelled data to converge to a good solution. For 3D medical images, it is often impractical to build a large homogeneous annotated dataset for a specific pathology. Self-supervised methods offer a new way to learn a representation of the images in an unsupervised manner with a neural network. In particular, contrastive learning has shown great promises by (almost) matching the performance of fully-supervised CNN on vision tasks. Nonetheless, this method does not take advantage of available meta-data, such as participant's age, viewed as prior knowledge. Here, we propose to leverage continuous proxy metadata, in the contrastive learning framework, by introducing a new loss called y-Aware InfoNCE loss. Specifically, we improve the positive sampling during pre-training by adding more positive examples with similar proxy meta-data with the anchor, assuming they share similar discriminative semantic features.With our method, a 3D CNN model pre-trained on 10^4 multi-site healthy brain MRI scans can extract relevant features for three classification tasks: schizophrenia, bipolar diagnosis and Alzheimer's detection. When fine-tuned, it also outperforms 3D CNN trained from scratch on these tasks, as well as state-of-the-art self-supervised methods. Our code is made publicly available here.
updated: Wed Jun 16 2021 14:17:04 GMT+0000 (UTC)
published: Wed Jun 16 2021 14:17:04 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アソシエイト