arXiv reaDer
自己教師あり学習とバッチ知識アンサンブルによる自動 COVID-19 検出パフォーマンスの向上
Boosting Automatic COVID-19 Detection Performance with Self-Supervised Learning and Batch Knowledge Ensembling
問題: 胸部 X 線 (CXR) 画像から COVID-19 を検出することは、COVID-19 を検出するための最速かつ最も簡単な方法の 1 つになりました。ただし、既存の方法では通常、自然画像からの教師付き転移学習を事前トレーニング プロセスとして使用します。これらの方法は、COVID-19 の固有の特徴や、COVID-19 と他の肺炎との間の類似の特徴を考慮していません。目的: この論文では、CXR 画像を使用する新しい高精度の COVID-19 検出方法を設計したいと考えています。これは、COVID-19 の独自の機能と、COVID-19 と他の肺炎との間の類似の機能を考慮することができます。方法: 私たちの方法は 2 つのフェーズで構成されます。 1 つは自己教師あり学習ベースの関係です。もう 1 つは、バッチ知識アンサンブル ベースの微調整です。自己教師あり学習ベースの事前トレーニングは、手動で注釈を付けたラベルなしで、CXR 画像から識別表現を学習できます。一方、バッチ知識アンサンブルベースの微調整では、視覚的特徴の類似性に従ってバッチ内の画像のカテゴリ知識を利用して、検出パフォーマンスを向上させることができます。以前の実装とは異なり、微調整フェーズにバッチ知識アンサンブルを導入し、自己教師あり学習で使用されるメモリを削減し、COVID-19 検出精度を向上させます。結果: 2 つの公開 COVID-19 CXR データセット、つまり大規模なデータセットと不均衡なデータセットで、私たちの方法は有望な COVID-19 検出性能を示しました。私たちの方法は、注釈付きの CXR トレーニング画像が大幅に削減された場合でも (たとえば、元のデータセットの 10% のみを使用して)、高い検出精度を維持します。さらに、私たちの方法はハイパーパラメータの変化に影響されません。
Problem: Detecting COVID-19 from chest X-Ray (CXR) images has become one of the fastest and easiest methods for detecting COVID-19. However, the existing methods usually use supervised transfer learning from natural images as a pretraining process. These methods do not consider the unique features of COVID-19 and the similar features between COVID-19 and other pneumonia. Aim: In this paper, we want to design a novel high-accuracy COVID-19 detection method that uses CXR images, which can consider the unique features of COVID-19 and the similar features between COVID-19 and other pneumonia. Methods: Our method consists of two phases. One is self-supervised learning-based pertaining; the other is batch knowledge ensembling-based fine-tuning. Self-supervised learning-based pretraining can learn distinguished representations from CXR images without manually annotated labels. On the other hand, batch knowledge ensembling-based fine-tuning can utilize category knowledge of images in a batch according to their visual feature similarities to improve detection performance. Unlike our previous implementation, we introduce batch knowledge ensembling into the fine-tuning phase, reducing the memory used in self-supervised learning and improving COVID-19 detection accuracy. Results: On two public COVID-19 CXR datasets, namely, a large dataset and an unbalanced dataset, our method exhibited promising COVID-19 detection performance. Our method maintains high detection accuracy even when annotated CXR training images are reduced significantly (e.g., using only 10% of the original dataset). In addition, our method is insensitive to changes in hyperparameters.
updated: Thu Mar 30 2023 12:32:37 GMT+0000 (UTC)
published: Mon Dec 19 2022 07:39:26 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アソシエイト