arXiv reaDer
医療画像セグメンテーションのための対照的かつ選択的な隠し埋め込み
Contrastive and Selective Hidden Embeddings for Medical Image Segmentation
医療画像のセグメンテーションは、臨床診断、分析、および治療計画の中心的な手順として広く認識されています。ただし、面倒で費用のかかる注釈プロセスは、さらなる進歩の速度を遅らせます。対照的な学習ベースの重みの事前トレーニングは、ラベルのないデータを活用して適切な表現を学習することにより、代替手段を提供します。この論文では、対照的な学習が一般的な監督された医療セグメンテーションタスクにどのように役立つかを調査します。この目的のために、パッチドラッグソー対照正則化(PDCR)が提案され、継続的な親和性スコアによって制御される範囲でパッチレベルの引っ張りと反発を実行します。また、不確実性を意識した特徴選択ブロック(UAFS)と呼ばれる新しい構造は、特徴選択プロセスを実行するように設計されています。これにより、不確実性の高い少数の特徴によって引き起こされる学習ターゲットのシフトを処理できます。提案された2つのモジュールを既存のセグメンテーションアーキテクチャにプラグインすることにより、6つのドメインからの8つのパブリックデータセットにわたって最先端の結果を達成します。新しく設計されたモジュールは、トレーニングデータの量をさらに4分の1に減らし、同等のパフォーマンスを達成します。この観点から、ラベルに含まれる情報をさらに掘り下げることにより、元の自己/教師なし対照学習とは反対の方向に進みます。
Medical image segmentation has been widely recognized as a pivot procedure for clinical diagnosis, analysis, and treatment planning. However, the laborious and expensive annotation process lags down the speed of further advances. Contrastive learning-based weight pre-training provides an alternative by leveraging unlabeled data to learn a good representation. In this paper, we investigate how contrastive learning benefits the general supervised medical segmentation tasks. To this end, patch-dragsaw contrastive regularization (PDCR) is proposed to perform patch-level tugging and repulsing with the extent controlled by a continuous affinity score. And a new structure dubbed uncertainty-aware feature selection block (UAFS) is designed to perform the feature selection process, which can handle the learning target shift caused by minority features with high uncertainty. By plugging the proposed 2 modules into the existing segmentation architecture, we achieve state-of-the-art results across 8 public datasets from 6 domains. Newly designed modules further decrease the amount of training data to a quarter while achieving comparable, if not better, performances. From this perspective, we take the opposite direction of the original self/un-supervised contrastive learning by further excavating information contained within the label.
updated: Fri Apr 29 2022 14:00:59 GMT+0000 (UTC)
published: Fri Jan 21 2022 16:52:19 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アソシエイト