融合分割与梯度提升树,用合成图像提升皮肤癌分类效果
Hybrid Ensemble of Segmentation-Assisted Classification and GBDT for Skin Cancer Detection with Engineered Metadata and Synthetic Lesions from ISIC 2024 Non-Dermoscopic 3D-TBP Images
- 用ViT和自研混合模型提取特征,结合分割辅助定位病变区域
- 在pAUC>80% TPR下达到0.1755,优于所有对比配置
- 适合远程医疗和资源有限场景的可解释性皮肤癌筛查
皮肤癌是全球最常见且威胁生命的疾病之一,早期检测对患者预后至关重要。本文提出一种基于机器学习与深度学习的混合方法,用于分类恶性与良性皮肤病变,使用来自ISIC 2024的SLICE-3D数据集,该数据集包含401,059张从3D全身体表摄影(TBP)中提取的病灶图像,模拟非皮肤镜、类智能手机拍摄条件。方法结合视觉变压器(EVA02)与自研卷积型ViT混合模型(EdgeNeXtSAC),通过分割辅助分类流程增强病灶定位能力。模型预测结果与由工程化特征及患者相关关系指标强化的梯度提升决策树(GBDT)集成。为缓解类别不平衡并提升泛化能力,采用Stable Diffusion生成合成恶性病灶,并应用诊断启发式重标注策略,将外部数据统一为三分类格式。以高于80%真阳性率(TPR)的局部AUC(pAUC)为评估标准,本方法取得0.1755的pAUC,为所有配置中最高。结果表明,该混合可解释人工智能系统在远程医疗和资源受限环境中具有皮肤癌初筛潜力。
原文摘要 · Abstract (English)
Skin cancer is among the most prevalent and life-threatening diseases worldwide, with early detection being critical to patient outcomes. This work presents a hybrid machine and deep learning-based approach for classifying malignant and benign skin lesions using the SLICE-3D dataset from ISIC 2024, which comprises 401,059 cropped lesion images extracted from 3D Total Body Photography (TBP), emulating non-dermoscopic, smartphone-like conditions. Our method combines vision transformers (EVA02) and our designed convolutional ViT hybrid (EdgeNeXtSAC) to extract robust features, employing a segmentation-assisted classification pipeline to enhance lesion localization. Predictions from these models are fused with a gradient-boosted decision tree (GBDT) ensemble enriched by engineered features and patient-specific relational metrics. To address class imbalance and improve generalization, we augment malignant cases with Stable Diffusion-generated synthetic lesions and apply a diagnosis-informed relabeling strategy to harmonize external datasets into a 3-class format. Using partial AUC (pAUC) above 80 percent true positive rate (TPR) as the evaluation metric, our approach achieves a pAUC of 0.1755 -- the highest among all configurations. These results underscore the potential of hybrid, interpretable AI systems for skin cancer triage in telemedicine and resource-constrained settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。