arXiv reaDer
マルチスタイルおよびマルチビューの対照学習によるマンモグラフィ検出のためのドメインの一般化
Domain Generalization for Mammography Detection via Multi-style and Multi-view Contrastive Learning
病変の検出は、マンモグラフィのコンピュータ支援診断スキームにおける基本的な問題です。トレーニングデータが大きく、画像のスタイルと品質の点で十分に多様であるという条件で、深層学習技術の進歩はこのタスクで目覚ましい進歩を遂げました。特に、画像スタイルの多様性は、主にベンダーの要因に起因する可能性があります。ただし、ベンダーからできるだけ多くのマンモグラムを収集することは非常に費用がかかり、実験室規模の研究には実用的でない場合があります。したがって、限られたリソースを持つさまざまなベンダーへの深層学習モデルの一般化機能をさらに強化するために、新しい対照的な学習スキームが開発されています。具体的には、バックボーンネットワークは、最初に、さまざまなベンダースタイルに不変の機能を埋め込むためのマルチスタイルおよびマルチビューの教師なし自己学習スキームでトレーニングされます。その後、バックボーンネットワークは、特定の教師あり学習を使用して病変検出のダウンストリームタスクに再調整されます。提案された方法は、4つのベンダーからのマンモグラムと1つの目に見えない公開データセットで評価されます。実験結果は、私たちのアプローチが、見えているドメインと見えていないドメインの両方で検出パフォーマンスを効果的に改善でき、多くの最先端(SOTA)の一般化手法よりも優れていることを示唆しています。
Lesion detection is a fundamental problem in the computer-aided diagnosis scheme for mammography. The advance of deep learning techniques have made a remarkable progress for this task, provided that the training data are large and sufficiently diverse in terms of image style and quality. In particular, the diversity of image style may be majorly attributed to the vendor factor. However, the collection of mammograms from vendors as many as possible is very expensive and sometimes impractical for laboratory-scale studies. Accordingly, to further augment the generalization capability of deep learning model to various vendors with limited resources, a new contrastive learning scheme is developed. Specifically, the backbone network is firstly trained with a multi-style and multi-view unsupervised self-learning scheme for the embedding of invariant features to various vendor-styles. Afterward, the backbone network is then recalibrated to the downstream task of lesion detection with the specific supervised learning. The proposed method is evaluated with mammograms from four vendors and one unseen public dataset. The experimental results suggest that our approach can effectively improve detection performance on both seen and unseen domains, and outperforms many state-of-the-art (SOTA) generalization methods.
updated: Sun Nov 21 2021 14:29:50 GMT+0000 (UTC)
published: Sun Nov 21 2021 14:29:50 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アソシエイト