arXiv reaDer
対照学習によるマンモグラフィ画像解析のための領域汎化
Domain Generalization for Mammographic Image Analysis via Contrastive Learning
マンモグラフィの画像解析は、深層学習の進歩により近年目覚ましい進歩を遂げているコンピュータ支援診断スキームの根幹をなす問題です。ただし、ディープ ラーニング モデルの構築には、画像のスタイルと品質の点で、大規模で十分に多様なトレーニング データが必要です。特に、画像スタイルの多様性は、主にベンダー要因に起因する可能性があります。ただし、できるだけ多くのベンダーからマンモグラムを収集すると、非常に費用がかかり、ラボ規模の研究には非現実的な場合があります。したがって、リソースが限られているさまざまなベンダーに対して深層学習モデルの一般化機能をさらに強化するために、新しい対照学習スキームが開発されました。具体的には、バックボーン ネットワークはまず、さまざまなベンダー スタイルに不変の機能を埋め込むためのマルチスタイルおよびマルチビューの教師なし自己学習スキームでトレーニングされます。その後、バックボーン ネットワークは、質量検出、マルチビュー マス マッチング、BI-RADS 分類、および特定の教師あり学習による乳房密度分類の下流のタスクに合わせて再調整されます。提案された方法は、4 つのベンダーと 2 つの目に見えない公開データセットからのマンモグラムで評価されます。実験結果は、私たちのアプローチが目に見える領域と見えない領域の両方で分析パフォーマンスを効果的に改善し、多くの最先端の (SOTA) 一般化方法よりも優れていることを示唆しています。
Mammographic image analysis is a fundamental problem in the computer-aided diagnosis scheme, which has recently made remarkable progress with the advance of deep learning. However, the construction of a deep learning model requires training data that 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, mammogram collection 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 models 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 tasks of mass detection, multi-view mass matching, BI-RADS classification and breast density classification with specific supervised learning. The proposed method is evaluated with mammograms from four vendors and two unseen public datasets. The experimental results suggest that our approach can effectively improve analysis performance on both seen and unseen domains, and outperforms many state-of-the-art (SOTA) generalization methods.
updated: Sun May 07 2023 03:45:26 GMT+0000 (UTC)
published: Thu Apr 20 2023 11:40:21 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アソシエイト