arXiv reaDer
医用画像に対する教師なし局所差別
Unsupervised Local Discrimination for Medical Images
対照学習は、医用画像分析における高価な注釈の高い需要を軽減するのに効果的であることが証明されています。最近の作品は主にインスタンスごとの識別に基づいており、グローバルな識別機能を学びます。ただし、主に局所的な類似性によって区別される小さな解剖学的構造、病変、および組織を扱うために臨床医を支援することはできません。この作業では、モデルの初期化のために医用画像から局所的な識別機能を学習するための一般的な教師なしフレームワークを提案します。同じ身体領域の画像は同様の解剖学的構造を共有し、同じ構造のピクセルは同様の意味パターンを持つ必要があるという事実に続いて、類似のコンテキストを持つピクセルがクラスター化され、異なるピクセルであるローカルの識別可能な埋め込みスペースを構築するニューラルネットワークを設計します分散しています。このネットワークには、主に2つのブランチが含まれています。ピクセル単位の埋め込みを生成する埋め込みブランチと、同じ構造のピクセルをまとめてセグメンテーションを生成するクラスタリングブランチです。領域識別損失は、相互に有益なパターンでこれら2つのブランチを最適化するために提案され、クラスタリングブランチによってクラスター化されたピクセルが類似の埋め込みベクトルを共有し、トレーニングされたモデルがピクセルごとの類似性を測定できるようにします。ダウンストリームタスクに転送されると、フレームワークに基づく学習された特徴抽出器は、より優れた一般化能力を示します。これは、広範な最先端の方法からのものを上回り、カラー眼底および胸部X線の12のダウンストリームタスクすべてのうち11を獲得します。さらに、ピクセル単位の埋め込みを利用して地域の類似性を測定し、形状に基づくクロスモダリティセグメンテーションフレームワークと中心に敏感なワンショットランドマークローカリゼーションアルゴリズムを提案します。コードはhttps://github.com/HuaiChen-でリリースされています。 1994 / LDLearning。
Contrastive learning has been proven effective to alleviate the high demand of expensive annotations in medical images analysis. Recent works are mainly based on instance-wise discrimination and learn global discriminative features; however, they cannot assist clinicians to deal with tiny anatomical structures, lesions, and tissues which are mainly distinguished by local similarities. In this work, we propose a general unsupervised framework to learn local discriminative features from medical images for models' initializations. Following the fact that images of the same body region should share similar anatomical structures, and pixels of the same structure should have similar semantic patterns, we design a neural network to construct a local discriminative embedding space where pixels with similar contexts are clustered and dissimilar pixels are dispersed. This network mainly contains two branches: an embedding branch to generate pixel-wise embeddings, and a clustering branch to gather pixels of the same structure together and generate segmentations. A region discriminative loss is proposed to optimize these two branches in a mutually beneficial pattern, making pixels clustered together by the clustering branch share similar embedded vectors and the trained model can measure pixel-wise similarity. When transferred to downstream tasks, the learnt feature extractor based on our framework shows better generalization ability, which outperforms those from extensive state-of-the-art methods and wins 11 out of all 12 downstream tasks in color fundus and chest X-ray. Furthermore, we utilize the pixel-wise embeddings to measure regional similarity and propose a shape-guided cross-modality segmentation framework and a center-sensitive one-shot landmark localization algorithm.Codes are released in https://github.com/HuaiChen-1994/LDLearning.
updated: Sun Nov 28 2021 11:47:41 GMT+0000 (UTC)
published: Sat Aug 21 2021 04:53: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アソシエイト