arXiv reaDer
ディープ ラーニング COVID-19 画像分類のための GAN ベースの拡張機能のパフォーマンス
Performance of GAN-based augmentation for deep learning COVID-19 image classification
深層学習を医療分野に適用する際の最大の課題は、トレーニング データの可用性です。データ拡張は、限られたデータ セットに直面した場合に機械学習で使用される典型的な方法論です。古典的なアプローチでは、画像変換、つまり回転、クロッピング、明るさの変更が使用されます。この作業では、Generative Adversarial Networks の StyleGAN2-ADA モデルが、限られた COVID-19 胸部 X 線画像セットでトレーニングされます。生成された画像の品質を評価した後、それらを使用してトレーニング データ セットを増やし、クラス間のバランスを改善します。文献でまだ完全に調査されていない COVID-19 陽性クラスを含む、胸部 X 線画像のマルチクラス分類問題を検討します。 COVID-19 胸部 X 線画像の転移学習ベースの分類の結果が表示されます。いくつかの深い畳み込みニューラル ネットワーク モデルのパフォーマンスが比較されます。回転、クロッピング、明るさの変化など、従来の画像増強の検出性能への影響が研究されています。さらに、従来の画像拡張は、GAN ベースの拡張と比較されます。最も正確なモデルは、90.2% の精度を持つ EfficientNet-B0 で、単純なクラス バランシングを使用してデータセットでトレーニングされています。 GAN拡張アプローチは、考慮されているデータセットの従来の方法に劣ることがわかりました。
The biggest challenge in the application of deep learning to the medical domain is the availability of training data. Data augmentation is a typical methodology used in machine learning when confronted with a limited data set. In a classical approach image transformations i.e. rotations, cropping and brightness changes are used. In this work, a StyleGAN2-ADA model of Generative Adversarial Networks is trained on the limited COVID-19 chest X-ray image set. After assessing the quality of generated images they are used to increase the training data set improving its balance between classes. We consider the multi-class classification problem of chest X-ray images including the COVID-19 positive class that hasn't been yet thoroughly explored in the literature. Results of transfer learning-based classification of COVID-19 chest X-ray images are presented. The performance of several deep convolutional neural network models is compared. The impact on the detection performance of classical image augmentations i.e. rotations, cropping, and brightness changes are studied. Furthermore, classical image augmentation is compared with GAN-based augmentation. The most accurate model is an EfficientNet-B0 with an accuracy of 90.2 percent, trained on a dataset with a simple class balancing. The GAN augmentation approach is found to be subpar to classical methods for the considered dataset.
updated: Tue Apr 18 2023 15:39:58 GMT+0000 (UTC)
published: Tue Apr 18 2023 15:39:58 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アソシエイト