arXiv reaDer
監視された対照的な学習のための脳を意識した代替
Brain-Aware Replacements for Supervised Contrastive Learning
脳MRIを使用してアルツハイマー病(AD)検出のための新しいフレームワークを提案します。フレームワークは、Brain-Aware Replaces(BAR)と呼ばれるデータ拡張方法から始まります。これは、標準の脳パーセレーションを利用して、ランダムに選択されたMRIからアンカーMRIの医学的に関連する3D脳領域を置き換え、合成サンプルを作成します。グラウンドトゥルースの「ハード」ラベルも、「ソフト」ラベルを作成するために、置換率に応じて線形に混合されます。 BARは、CutMixなどの他の混合ベースの方法と比較して、局所的な変動性が高い、多種多様なリアルな外観の合成MRIを生成します。 BARに加えて、ソフトラベルを使用して、合成MRIがどのように混合されているかを反映する表現の相対的な類似性を学習することを目的として、ソフトラベル対応の監視付きコントラスト損失を使用することを提案します。このように、ハードラベルのエントロピー容量を完全に使い果たすことはありません。これは、ハードラベルを使用して、BARを介してソフトラベルと合成MRIを作成するためだけに使用するためです。フレームワークを使用して事前トレーニングされたモデルは、合成サンプルの作成に使用されたハードラベルを使用して、クロスエントロピー損失でさらに微調整できることを示します。バイナリAD検出タスクでのフレームワークのパフォーマンスを、ゼロからの教師ありトレーニングと最先端の自己教師ありトレーニングおよび微調整アプローチの両方に対して検証しました。次に、BARをフレームワークに統合することにより、BARの個々のパフォーマンスを別のミックスベースのメソッドCutMixと比較して評価しました。私たちのフレームワークは、AD検出タスクの適合率と再現率の両方で優れた結果をもたらすことを示しています。
We propose a novel framework for Alzheimer's disease (AD) detection using brain MRIs. The framework starts with a data augmentation method called Brain-Aware Replacements (BAR), which leverages a standard brain parcellation to replace medically-relevant 3D brain regions in an anchor MRI from a randomly picked MRI to create synthetic samples. Ground truth "hard" labels are also linearly mixed depending on the replacement ratio in order to create "soft" labels. BAR produces a great variety of realistic-looking synthetic MRIs with higher local variability compared to other mix-based methods, such as CutMix. On top of BAR, we propose using a soft-label-capable supervised contrastive loss, aiming to learn the relative similarity of representations that reflect how mixed are the synthetic MRIs using our soft labels. This way, we do not fully exhaust the entropic capacity of our hard labels, since we only use them to create soft labels and synthetic MRIs through BAR. We show that a model pre-trained using our framework can be further fine-tuned with a cross-entropy loss using the hard labels that were used to create the synthetic samples. We validated the performance of our framework in a binary AD detection task against both from-scratch supervised training and state-of-the-art self-supervised training plus fine-tuning approaches. Then we evaluated BAR's individual performance compared to another mix-based method CutMix by integrating it within our framework. We show that our framework yields superior results in both precision and recall for the AD detection task.
updated: Mon Jul 11 2022 01:17:35 GMT+0000 (UTC)
published: Mon Jul 11 2022 01:17:35 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アソシエイト