arXiv reaDer
IDEAL: 半教師あり医用画像セグメンテーションのための高密度局所対照学習の改善
IDEAL: Improved DEnse locAL Contrastive Learning for Semi-Supervised Medical Image Segmentation
ラベル付けされたデータが不足しているため、対照的な自己教師あり学習 (SSL) フレームワークは、最近、いくつかの医療画像分析タスクで大きな可能性を示しています。ただし、既存のコントラスト メカニズムは、局所的な特徴をマイニングできないため、高密度のピクセル レベルのセグメンテーション タスクには最適ではありません。この目的のために、メトリック学習の概念をセグメンテーション タスクに拡張し、ディープ エンコーダー ネットワークを事前トレーニングするための密な (非) 類似性学習を使用し、半教師ありパラダイムを使用してダウンストリーム タスクを微調整します。具体的には、密なピクセルレベルの特徴を取得するための単純な畳み込みプロジェクションヘッドと、これらの密なプロジェクションを利用するための新しいコントラスト損失を提案し、それによってローカル表現を改善します。 2 ストリーム モデル トレーニングを含む双方向の一貫性正則化メカニズムは、ダウンストリーム タスク用に考案されています。比較すると、IDEAL メソッドは、心臓 MRI セグメンテーションでかなりのマージンで SoTA メソッドよりも優れています。利用可能なコード: https://github.com/hritam-98/IDEAL-ICASSP23
Due to the scarcity of labeled data, Contrastive Self-Supervised Learning (SSL) frameworks have lately shown great potential in several medical image analysis tasks. However, the existing contrastive mechanisms are sub-optimal for dense pixel-level segmentation tasks due to their inability to mine local features. To this end, we extend the concept of metric learning to the segmentation task, using a dense (dis)similarity learning for pre-training a deep encoder network, and employing a semi-supervised paradigm to fine-tune for the downstream task. Specifically, we propose a simple convolutional projection head for obtaining dense pixel-level features, and a new contrastive loss to utilize these dense projections thereby improving the local representations. A bidirectional consistency regularization mechanism involving two-stream model training is devised for the downstream task. Upon comparison, our IDEAL method outperforms the SoTA methods by fair margins on cardiac MRI segmentation. Code available: https://github.com/hritam-98/IDEAL-ICASSP23
updated: Thu Mar 02 2023 08:25:37 GMT+0000 (UTC)
published: Wed Oct 26 2022 23:11:02 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アソシエイト