arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.01284cs.CVcs.LG
收藏

基于不平衡小数据集的单模型深度学习用于皮肤病变分类

Single Model Deep Learning on Imbalanced Small Datasets for Skin Lesion Classification

University of Science and Technology of China(中国科学技术大学);The Ohio State University Medical Center(俄亥俄州立大学医学中心)

Peng Yao, Shuwei Shen, Mengjuan Xu, Peng Liu, Fan Zhang, Jinyu Xing, Pengfei Shao, Benjamin Kaffenberger, Ronald X. Xu

摘要:

深度卷积神经网络(DCNN)模型已被广泛探索用于皮肤病诊断,其中一些模型已经实现了与皮肤科医生相当甚至更好的诊断结果。然而,DCNN在皮肤疾病检测中的广泛实施受到可访问皮肤病变数据集的小尺寸和数据不平衡的阻碍。本文提出了一种新的基于单模型的策略,用于小的和不平衡的数据集上的皮肤病变的分类。首先,在不同的小型和不平衡数据集上训练各种DCNN,以验证中等复杂度的模型优于较大的模型。其次,加入正则化DropOut和DropBlock来减少过拟合,并针对小数据集样本代表性不足的缺陷,提出了一种改进的RandAugment增强策略。最后,引入了一种新的多加权新损失(MWNL)函数和一种端到端累积学习策略(CLS),以克服样本量不均匀和分类困难的挑战,并减少异常样本对训练的影响。通过结合Modified RandAugment,MWNL和CLS,我们的单一DCNN模型方法在不同的皮肤镜图像数据集上实现了与多个集成模型相当或更好的分类精度。我们的研究表明,该方法能够以较低的计算资源和推理时间成本实现高分类性能,可能适合在移动设备中实现,用于在低资源设置中自动筛选皮肤病变和许多其他恶性肿瘤。

英文摘要:

Deep convolutional neural network (DCNN) models have been widely explored for skin disease diagnosis and some of them have achieved the diagnostic outcomes comparable or even superior to those of dermatologists. However, broad implementation of DCNN in skin disease detection is hindered by small size and data imbalance of the publically accessible skin lesion datasets. This paper proposes a novel single-model based strategy for classification of skin lesions on small and imbalanced datasets. First, various DCNNs are trained on different small and imbalanced datasets to verify that the models with moderate complexity outperform the larger models. Second, regularization DropOut and DropBlock are added to reduce overfitting and a Modified RandAugment augmentation strategy is proposed to deal with the defects of sample underrepresentation in the small dataset. Finally, a novel Multi-Weighted New Loss (MWNL) function and an end-to-end cumulative learning strategy (CLS) are introduced to overcome the challenge of uneven sample size and classification difficulty and to reduce the impact of abnormal samples on training. By combining Modified RandAugment, MWNL and CLS, our single DCNN model method achieved the classification accuracy comparable or superior to those of multiple ensembling models on different dermoscopic image datasets. Our study shows that this method is able to achieve a high classification performance at a low cost of computational resources and inference time, potentially suitable to implement in mobile devices for automated screening of skin lesions and many other malignancies in low resource settings.

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

↑