用迁移学习与领域自适应提升跨肤色皮肤疾病诊断准确率
Equitable Skin Disease Prediction Using Transfer Learning and Domain Adaptation
- 融合多源预训练模型,增强对不同肤色的诊断鲁棒性
- 在DDI数据集上,Med-ViT模型表现最优,跨肤色准确率显著提升
- 适合关注医疗AI公平性与跨种族泛化能力的研究者
皮肤病诊断依赖专业医师经验,准确识别从癌症到炎症性疾病至关重要。现有AI模型在不同肤色患者中表现不均,尤其在深肤色人群中性能明显下降。公开、无偏见的数据集稀缺,限制了包容性AI工具的发展。为此,本文采用迁移学习策略,整合来自通用图像及特定医学图像的多种预训练模型,以增强皮肤疾病预测的鲁棒性与包容性。在包含罕见与常见肤色的Diverse Dermatology Images(DDI)数据集上进行严格评估,结果表明,基于多源特征学习的Med-ViT表现最佳。进一步通过HAM10000等数据集进行领域自适应,显著提升了所有模型在各类肤色上的表现。
原文摘要 · Abstract (English)
In the realm of dermatology, the complexity of diagnosing skin conditions manually necessitates the expertise of dermatologists. Accurate identification of various skin ailments, ranging from cancer to inflammatory diseases, is paramount. However, existing artificial intelligence (AI) models in dermatology face challenges, particularly in accurately diagnosing diseases across diverse skin tones, with a notable performance gap in darker skin. Additionally, the scarcity of publicly available, unbiased datasets hampers the development of inclusive AI diagnostic tools. To tackle the challenges in accurately predicting skin conditions across diverse skin tones, we employ a transfer-learning approach that capitalizes on the rich, transferable knowledge from various image domains. Our method integrates multiple pre-trained models from a wide range of sources, including general and specific medical images, to improve the robustness and inclusiveness of the skin condition predictions. We rigorously evaluated the effectiveness of these models using the Diverse Dermatology Images (DDI) dataset, which uniquely encompasses both underrepresented and common skin tones, making it an ideal benchmark for assessing our approach. Among all methods, Med-ViT emerged as the top performer due to its comprehensive feature representation learned from diverse image sources. To further enhance performance, we conducted domain adaptation using additional skin image datasets such as HAM10000. This adaptation significantly improved model performance across all models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。