arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.06824eess.IVcs.CV
收藏

使用对抗训练和深度迁移学习检测黑色素瘤

Melanoma Detection using Adversarial Training and Deep Transfer Learning

Concordia University(康考迪亚大学)

Hasib Zunair, A. Ben Hamza

摘要:

皮肤病变数据集主要由正常样本组成,只有一小部分异常样本,从而导致类别不平衡问题。此外,由于类间变异性低,皮肤病变图像在整体外观上很大程度上相似。在本文中,我们提出了一个两阶段的框架,用于使用对抗训练和转移学习对黑色素瘤检测进行皮肤病变图像的自动分类。在第一阶段中,我们利用数据分布的类间变化进行条件图像合成,通过学习类间映射并使用未配对的图像到图像转换从过度代表的类中合成代表性不足的类样本。在第二阶段,我们使用原始训练集结合新合成的代表性不足的类样本来训练用于皮肤病变分类的深度卷积神经网络。该分类器的训练是通过最小化焦点损失函数来进行的,这有助于模型从困难的示例中学习,同时降低简单示例的权重。在皮肤病学图像基准上进行的实验表明,我们提出的方法优于几个标准的基线方法,实现了显着的性能改进。有趣的是,我们通过特征可视化和分析表明,我们的方法导致基于上下文的病变评估,可以达到专家皮肤科医生的水平。

英文摘要:

Skin lesion datasets consist predominantly of normal samples with only a small percentage of abnormal ones, giving rise to the class imbalance problem. Also, skin lesion images are largely similar in overall appearance owing to the low inter-class variability. In this paper, we propose a two-stage framework for automatic classification of skin lesion images using adversarial training and transfer learning toward melanoma detection. In the first stage, we leverage the inter-class variation of the data distribution for the task of conditional image synthesis by learning the inter-class mapping and synthesizing under-represented class samples from the over-represented ones using unpaired image-to-image translation. In the second stage, we train a deep convolutional neural network for skin lesion classification using the original training set combined with the newly synthesized under-represented class samples. The training of this classifier is carried out by minimizing the focal loss function, which assists the model in learning from hard examples, while down-weighting the easy ones. Experiments conducted on a dermatology image benchmark demonstrate the superiority of our proposed approach over several standard baseline methods, achieving significant performance improvements. Interestingly, we show through feature visualization and analysis that our method leads to context based lesion assessment that can reach an expert dermatologist level.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑