用生成模型合成暗肤色皮肤癌图像,提升AI诊断公平性
A Generative AI Approach for Reducing Skin Tone Bias in Skin Cancer Classification
- 用LoRA微调Stable Diffusion生成暗肤色皮肤癌图像
- 分割任务指标提升,分类准确率达92.14%
- 适合关注医疗AI公平性的研究者与临床开发者
皮肤癌是全球最常见的癌症之一,早期检测对治疗至关重要。然而,当前AI诊断工具多基于以浅肤色为主的训练数据,导致深肤色人群的准确率下降。国际皮肤成像合作组织(ISIC)数据集超过70%为浅肤色图像,深肤色不足8%,严重制约医疗公平。本文提出一种生成式增强方法:使用低秩适配(LoRA)微调预训练Stable Diffusion模型,在ISIC数据集中深肤色子集上生成受病变类型和肤色条件控制的合成皮肤镜图像。在下游任务中,分割模型在真实测试图像上的交并比(IoU)、Dice系数和边界精度均显著提升;二分类任务中,EfficientNet-B0模型在增强数据集上达到92.14%准确率。结果验证了合成数据的有效性,表明生成式AI可有效缓解皮肤色调偏差,提升皮肤病学诊断的公平性,并为未来研究提供新方向。
原文摘要 · Abstract (English)
Skin cancer is one of the most common cancers worldwide and early detection is critical for effective treatment. However, current AI diagnostic tools are often trained on datasets dominated by lighter skin tones, leading to reduced accuracy and fairness for people with darker skin. The International Skin Imaging Collaboration (ISIC) dataset, one of the most widely used benchmarks, contains over 70% light skin images while dark skins fewer than 8%. This imbalance poses a significant barrier to equitable healthcare delivery and highlights the urgent need for methods that address demographic diversity in medical imaging. This paper addresses this challenge of skin tone imbalance in automated skin cancer detection using dermoscopic images. To overcome this, we present a generative augmentation pipeline that fine-tunes a pre-trained Stable Diffusion model using Low-Rank Adaptation (LoRA) on the image dark-skin subset of the ISIC dataset and generates synthetic dermoscopic images conditioned on lesion type and skin tone. In this study, we investigated the utility of these images on two downstream tasks: lesion segmentation and binary classification. For segmentation, models trained on the augmented dataset and evaluated on held-out real images show consistent improvements in IoU, Dice coefficient, and boundary accuracy. These evalutions provides the verification of Generated dataset. For classification, an EfficientNet-B0 model trained on the augmented dataset achieved 92.14% accuracy. This paper demonstrates that synthetic data augmentation with Generative AI integration can substantially reduce bias with increase fairness in conventional dermatological diagnostics and open challenges for future directions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。