arXiv reaDer
胎児超音波のための解剖学的認識対比表現学習
Anatomy-Aware Contrastive Representation Learning for Fetal Ultrasound
自己教師あり対照的表現学習は、転移学習のために、ラベル付けされていない医療データセットから意味のある視覚的表現を学習するという利点を提供します。ただし、ドメイン固有の解剖学的特徴を考慮せずに現在の対照的な学習アプローチを医療データに適用すると、外観とセマンティクスに一貫性のない視覚的表現が生じる可能性があります。この論文では、解剖学的情報を組み込んで対照的な学習方法で正/負のペアサンプリングを強化する解剖学的認識対比学習(AWCL)を介して、医用画像の視覚的表現を改善することを提案します。提案されたアプローチは、自動化された胎児超音波イメージング タスクに対して実証され、解剖学的に類似している同じまたは異なる超音波スキャンからのポジティブ ペアをまとめて、表現学習を改善できるようにします。対照的な学習のために、粗粒度と細粒度の解剖学情報を含めることの効果を経験的に調査し、クラス内の違いを保持する細粒度の解剖学情報を使用した学習が、対応するものよりも効果的であることを発見しました。また、AWCL フレームワークに対する解剖学的比率の影響を分析し、より明確ではあるが解剖学的に類似したサンプルを使用して正のペアを構成すると、より質の高い表現が得られることがわかりました。大規模な胎児超音波データセットでの実験は、私たちのアプローチが 3 つの臨床下流タスクにうまく移行する表現の学習に効果的であり、ImageNet 教師ありおよび現在の最先端の対照学習法と比較して優れたパフォーマンスを達成することを示しています。特に、AWCL は、クロスドメイン セグメンテーション タスクで、ImageNet 教師ありメソッドよりも 13.8% 優れ、最先端のコントラスト ベースのメソッドよりも 7.1% 優れています。
Self-supervised contrastive representation learning offers the advantage of learning meaningful visual representations from unlabeled medical datasets for transfer learning. However, applying current contrastive learning approaches to medical data without considering its domain-specific anatomical characteristics may lead to visual representations that are inconsistent in appearance and semantics. In this paper, we propose to improve visual representations of medical images via anatomy-aware contrastive learning (AWCL), which incorporates anatomy information to augment the positive/negative pair sampling in a contrastive learning manner. The proposed approach is demonstrated for automated fetal ultrasound imaging tasks, enabling the positive pairs from the same or different ultrasound scans that are anatomically similar to be pulled together and thus improving the representation learning. We empirically investigate the effect of inclusion of anatomy information with coarse- and fine-grained granularity, for contrastive learning and find that learning with fine-grained anatomy information which preserves intra-class difference is more effective than its counterpart. We also analyze the impact of anatomy ratio on our AWCL framework and find that using more distinct but anatomically similar samples to compose positive pairs results in better quality representations. Experiments on a large-scale fetal ultrasound dataset demonstrate that our approach is effective for learning representations that transfer well to three clinical downstream tasks, and achieves superior performance compared to ImageNet supervised and the current state-of-the-art contrastive learning methods. In particular, AWCL outperforms ImageNet supervised method by 13.8% and state-of-the-art contrastive-based method by 7.1% on a cross-domain segmentation task.
updated: Mon Aug 22 2022 22:49:26 GMT+0000 (UTC)
published: Mon Aug 22 2022 22:49:26 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アソシエイト