arXiv reaDer
医用画像における教師なし異常検出とセグメンテーションのための自己監視型マルチクラス事前トレーニング
Self-supervised Multi-class Pre-training for Unsupervised Anomaly Detection and Segmentation in Medical Images
通常の(健康な)トレーニング画像のみを必要とする教師なし異常検出(UAD)は、異常(または疾患)の収集と注釈付けが困難な場合が多いため、疾患スクリーニングなどの医療画像分析(MIA)アプリケーションの開発を可能にする重要なツールです。 )MIAの画像。ただし、通常の画像に大きく依存すると、モデルのトレーニングが通常のクラスに適合しなくなる可能性があります。自己管理型の事前トレーニングは、この問題の効果的な解決策です。残念ながら、コンピュータビジョンから採用された現在の自己監視方法は、口実タスクまたはトレーニングプロセスを設計するためのMIAドメイン知識を調査しないため、MIAアプリケーションには最適ではありません。この論文では、MIAアプリケーション用に設計されたUADの新しい自己監視事前トレーニング方法を提案します。これは、対照学習によるマルチクラスの強力な拡張(MSACL)と呼ばれます。 MSACLは、合成された異常画像の正常なクラスと複数のクラスを対比するための新しい最適化に基づいており、各クラスは、ユークリッド距離と余弦の類似性に関して緊密で密なクラスターを形成するように強制されます。異常な画像は、さまざまな数の病変をシミュレートすることによって形成されます。通常の画像ではさまざまなサイズと外観があります。実験では、MSACL事前トレーニングにより、結腸内視鏡検査、眼底スクリーニング、およびCovid-19胸部X線データセットを使用して、多くのMIAベンチマークでSOTAUADメソッドの精度が向上することを示しています。
Unsupervised anomaly detection (UAD) that requires only normal (healthy) training images is an important tool for enabling the development of medical image analysis (MIA) applications, such as disease screening, since it is often difficult to collect and annotate abnormal (or disease) images in MIA. However, heavily relying on the normal images may cause the model training to overfit the normal class. Self-supervised pre-training is an effective solution to this problem. Unfortunately, current self-supervision methods adapted from computer vision are sub-optimal for MIA applications because they do not explore MIA domain knowledge for designing the pretext tasks or the training process. In this paper, we propose a new self-supervised pre-training method for UAD designed for MIA applications, named Multi-class Strong Augmentation via Contrastive Learning (MSACL). MSACL is based on a novel optimisation to contrast normal and multiple classes of synthetised abnormal images, with each class enforced to form a tight and dense cluster in terms of Euclidean distance and cosine similarity, where abnormal images are formed by simulating a varying number of lesions of different sizes and appearance in the normal images. In the experiments, we show that our MSACL pre-training improves the accuracy of SOTA UAD methods on many MIA benchmarks using colonoscopy, fundus screening and Covid-19 Chest X-ray datasets.
updated: Wed Nov 24 2021 04:55:17 GMT+0000 (UTC)
published: Fri Sep 03 2021 04:25:57 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アソシエイト