arXiv reaDer
医用画像分類のための弱く監視された生成的敵対的ネットワーク
Weakly-supervised Generative Adversarial Networks for medical image classification
弱教師あり学習は、近年人気のあるテクノロジーになっています。この論文では、弱教師あり生成的敵対的ネットワーク(WSGAN)と呼ばれる新しい医療画像分類アルゴリズムを提案します。これは、ラベルのない少数の実際の画像のみを使用して偽の画像を生成したり、トレーニングのサンプルサイズを拡大するために画像をマスクしたりします。設定。まず、MixMatchと組み合わせて、偽の画像とラベルのない画像の疑似ラベルを生成し、分類を行います。第二に、対照的な学習と自己注意メカニズムが提案された問題に導入され、分類の精度が向上します。第三に、モード崩壊の問題は、周期的な一貫性の喪失によってうまく対処されます。最後に、分類に必要な重要な情報で相互に補完するグローバル分類器とローカル分類器を設計します。 4つの医用画像データセットの実験結果は、WSGANが少数のラベル付きデータとラベルなしデータを使用することで比較的高い学習パフォーマンスを得ることができることを示しています。たとえば、WSGANの分類精度は、OCTデータセットに100個のラベル付き画像と1000個のラベルなし画像がある2番目にランク付けされたMIXMATCHの分類精度よりも11%高くなっています。さらに、アルゴリズムの有効性を検証するためにアブレーション実験も実施しています。
Weakly-supervised learning has become a popular technology in recent years. In this paper, we propose a novel medical image classification algorithm, called Weakly-Supervised Generative Adversarial Networks (WSGAN), which only uses a small number of real images without labels to generate fake images or mask images to enlarge the sample size of the training set. First, we combine with MixMatch to generate pseudo labels for the fake images and unlabeled images to do the classification. Second, contrastive learning and self-attention mechanism are introduced into the proposed problem to enhance the classification accuracy. Third, the problem of mode collapse is well addressed by cyclic consistency loss. Finally, we design global and local classifiers to complement each other with the key information needed for classification. The experimental results on four medical image datasets show that WSGAN can obtain relatively high learning performance by using few labeled and unlabeled data. For example, the classification accuracy of WSGAN is 11% higher than that of the second-ranked MIXMATCH with 100 labeled images and 1000 unlabeled images on the OCT dataset. In addition, we also conduct ablation experiments to verify the effectiveness of our algorithm.
updated: Mon Nov 29 2021 15:38:48 GMT+0000 (UTC)
published: Mon Nov 29 2021 15:38:48 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アソシエイト