arXiv:2411.05269cs.CVcs.LG2024-11被引 1

用生成式AI创建皮肤癌数据集,缓解样本不均衡问题。

Cancer-Net SCa-Synth: An Open Access Synthetically Generated 2D Skin Lesion Dataset for Skin Cancer Classification

  • 基于Stable Diffusion和DreamBooth生成2D皮肤病变图像。
  • 加入合成数据后模型在ISIC 2020测试集上表现提升。
  • 开源数据集助力皮肤癌智能诊断研究。

美国皮肤癌发病率最高,早期发现对预防严重后果至关重要。尽管数据集构建与深度学习技术取得进展,但现有开源数据集存在显著类别不平衡问题,影响模型效果。生成式人工智能(AI)可通过合成数据补充少数类,提升模型性能。本文利用Stable Diffusion与DreamBooth等最新生成技术,构建了Cancer-Net SCa-Synth——一个公开可用的合成2D皮肤病变数据集,用于皮肤癌分类。通过对比使用与不使用合成图像训练同一模型在ISIC 2020测试集的表现,验证了合成数据的有效性。该数据集已开源,地址为https://github.com/catai9/Cancer-Net-SCa-Synth,旨在推动癌症医疗AI的开放协作。

原文摘要 · Abstract (English)

In the United States, skin cancer ranks as the most commonly diagnosed cancer, presenting a significant public health issue due to its high rates of occurrence and the risk of serious complications if not caught early. Recent advancements in dataset curation and deep learning have shown promise in quick and accurate detection of skin cancer. However, current open-source datasets have significant class imbalances which impedes the effectiveness of these deep learning models. In healthcare, generative artificial intelligence (AI) models have been employed to create synthetic data, addressing data imbalance in datasets by augmenting underrepresented classes and enhancing the overall quality and performance of machine learning models. In this paper, we build on top of previous work by leveraging new advancements in generative AI, notably Stable Diffusion and DreamBooth. We introduce Cancer-Net SCa-Synth, an open access synthetically generated 2D skin lesion dataset for skin cancer classification. Further analysis on the data effectiveness by comparing the ISIC 2020 test set performance for training with and without these synthetic images for a simple model highlights the benefits of leveraging synthetic data to improve performance. Cancer-Net SCa-Synth is publicly available at https://github.com/catai9/Cancer-Net-SCa-Synth as part of a global open-source initiative for accelerating machine learning for cancer care.

皮肤癌生成模型数据增强开源数据

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。