arXiv reaDer
医療画像における教師なし異常検出とセグメンテーションのための自己教師あり擬似マルチクラス事前トレーニング
Self-supervised Pseudo Multi-class Pre-training for Unsupervised Anomaly Detection and Segmentation in Medical Images
教師なし異常検出 (UAD) 手法は、正常 (または健康な) 画像のみを使用してトレーニングされますが、テスト中に、正常な画像と異常 (または病気の) 画像を分類できます。 UAD は、疾患のスクリーニング問題に適用される重要な医用画像解析 (MIA) 手法です。これは、これらの問題で利用できるトレーニング セットには通常、正常な画像しか含まれていないためです。ただし、通常の画像のみに依存すると、さまざまなサイズ、外観、形状の目に見えない異常な病変を検出およびセグメント化するのに十分な感度が得られない、非効率な低次元画像表現の学習が行われる可能性があります。コンピューター ビジョン技術に基づいた自己教師あり学習による UAD メソッドの事前トレーニングは、この課題を軽減できますが、口実タスクを設計するためのドメイン知識を探索せず、対照的な学習損失が発生しないため、最適とは言えません。正常なトレーニング画像をクラスタ化すると、正常な画像がまばらに分布し、異常検出には効果がなくなる可能性があります。この論文では、MIA UAD アプリケーションのための新しい自己教師あり事前トレーニング方法、Pseudo Multi-class Strong Augmentation via Contrastive Learning (PMSACL) を提案します。 PMSACL は、正常な画像クラスと、合成された異常画像の複数の疑似クラスを対比する新しい最適化手法で構成されており、各クラスは特徴空間内で密なクラスターを形成するように強制されます。実験では、PMSACL 事前トレーニングにより、結腸内視鏡検査、眼底スクリーニング、および Covid-19 胸部 X 線データセットを使用した多くの MIA ベンチマークにおける SOTA UAD メソッドの精度が向上することが示されました。コードは https://github.com/tianyu0207/PMSACL 経由で公開されています。
Unsupervised anomaly detection (UAD) methods are trained with normal (or healthy) images only, but during testing, they are able to classify normal and abnormal (or disease) images. UAD is an important medical image analysis (MIA) method to be applied in disease screening problems because the training sets available for those problems usually contain only normal images. However, the exclusive reliance on normal images may result in the learning of ineffective low-dimensional image representations that are not sensitive enough to detect and segment unseen abnormal lesions of varying size, appearance, and shape. Pre-training UAD methods with self-supervised learning, based on computer vision techniques, can mitigate this challenge, but they are sub-optimal because they do not explore domain knowledge for designing the pretext tasks, and their contrastive learning losses do not try to cluster the normal training images, which may result in a sparse distribution of normal images that is ineffective for anomaly detection. In this paper, we propose a new self-supervised pre-training method for MIA UAD applications, named Pseudo Multi-class Strong Augmentation via Contrastive Learning (PMSACL). PMSACL consists of a novel optimisation method that contrasts a normal image class from multiple pseudo classes of synthesised abnormal images, with each class enforced to form a dense cluster in the feature space. In the experiments, we show that our PMSACL pre-training improves the accuracy of SOTA UAD methods on many MIA benchmarks using colonoscopy, fundus screening and Covid-19 Chest X-ray datasets. The code is made publicly available via https://github.com/tianyu0207/PMSACL.
updated: Mon Aug 14 2023 21:32:45 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アソシエイト