arXiv:2506.18414cs.CVcs.AI2025-06

用生成模型解释皮肤癌诊断不确定,让AI的犹豫变成可看的依据。

Latent Space Analysis for Interpretable Uncertainty in Melanoma Classification

  • 用对抗训练的VAE将皮肤病变映射到连续潜空间,保持相似图像邻近。
  • 在潜空间上训练XGBoost,AUC达0.868,接近顶尖模型表现。
  • 支持临床通过图像检索比对可疑病灶,提升诊断可信度。

黑色素瘤是一种高度侵袭性皮肤癌,早期准确诊断至关重要。尽管深度学习在皮肤病变分类中表现优异,但传统“黑箱”模型难以解释诊断不确定性,限制了临床信任。本文提出一种混合框架,结合类感知对抗变分自编码器与XGBoost分类器,超越简单二分类,利用生成潜空间实现可解释决策支持。通过对抗训练,模型学习皮肤病变的视觉特征,并将其投影到连续潜空间,确保相似图像在空间中邻近。在该潜空间上训练的XGBoost分类器取得0.868的稳健AUC,与当前先进模型相当。对于边界病例,框架支持基于内容的图像检索,使临床医生可直观对比可疑病灶与经活检确认的先例,同时作为早期预警信号——边缘分类提示病灶兼具痣与黑色素瘤特征,可能需密切监测。本方法将算法迟疑转化为透明、基于证据的可视化支持,弥合预测性能与临床信任之间的鸿沟。

原文摘要 · Abstract (English)

Melanoma is a highly aggressive skin cancer, making early and accurate diagnosis critical. While deep learning excels in skin lesion classification, standard ``black-box" models struggle to explain diagnostic uncertainty, limiting clinical trust. This work introduces a hybrid framework combining a class-aware adversarial Variational Autoencoder and an XGBoost classifier, transcending simple binary classification by leveraging a generative latent space for interpretable decision support. Guided by adversarial training, the model learns the visual characteristics of skin lesions and projects them into a continuous latent space, ensuring that similar images are grouped closely together. Trained on this latent space, the XGBoost classifier achieves a robust AUC of 0.868, competing closely with state-of-the-art models. For borderline cases, the framework enables clinicians to leverage the latent topology through Content-Based Image Retrieval. This provides a dual benefit: it allows the clinician to visually compare an ambiguous lesion against biopsy-confirmed precedents and acts as an early warning sign since a borderline classification can indicate that a lesion shares features of both nevi and melanomas, potentially requiring close monitoring. Our approach translates algorithmic hesitation into transparent, evidence-based visual support, bridging the gap between predictive performance and clinical trust.

皮肤癌可解释性潜空间医学影像

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