ディープニューラルネットワーク(DNN)は、医療画像に大きな影響を与えます。皮膚がんの分類にDNNを採用することの1つの重要な問題は、既存のデータセットのクラス頻度が不均衡であることです。この問題は、堅牢で一般化されたモデルのトレーニングを妨げます。データ拡張は、既存のデータをより効果的に使用することでこれに対処します。ただし、標準のデータ拡張の実装は手動で設計され、限られた合理的な代替データのみを生成します。代わりに、Generative Adversarial Networks(GAN)を使用して、より広範な拡張を生成します。本論文では、自己注意メカニズムを使用したプログレッシブ生成的敵対ネットワーク(PGAN)の新しい拡張機能を提案します。自己注意メカニズムを使用して、機能マップの長距離依存関係を直接モデル化します。したがって、自己注意はPGANを補完して、臨床的に意味のある情報を含むきめの細かいサンプルを生成します。さらに、安定化手法は、拡張された生成モデルに適用されました。生成モデルをトレーニングするために、ISIC 2018の皮膚病変チャレンジデータセットを使用して、非常に現実的な皮膚病変サンプルを合成し、分類結果をさらに向上させました。 70.1%の精度を達成しました。これは、67.3%の非拡張精度よりも2.8%優れています。
Deep Neural Networks (DNNs) show a significant impact on medical imaging. One significant problem with adopting DNNs for skin cancer classification is that the class frequencies in the existing datasets are imbalanced. This problem hinders the training of robust and well-generalizing models. Data Augmentation addresses this by using existing data more effectively. However, standard data augmentation implementations are manually designed and produce only limited reasonably alternative data. Instead, Generative Adversarial Networks (GANs) is utilized to generate a much broader set of augmentations. This paper proposes a novel enhancement for the progressive generative adversarial networks (PGAN) using self-attention mechanism. Self-attention mechanism is used to directly model the long-range dependencies in the feature maps. Accordingly, self-attention complements PGAN to generate fine-grained samples that comprise clinically-meaningful information. Moreover, the stabilization technique was applied to the enhanced generative model. To train the generative models, ISIC 2018 skin lesion challenge dataset was used to synthesize highly realistic skin lesion samples for boosting further the classification result. We achieve an accuracy of 70.1% which is 2.8% better than the non-augmented one of 67.3%.