SynthraXCoreNet:一种用于皮肤镜图像分类的可解释且校准良好的六CNN集成模型

《IEEE Access》:SynthraXCoreNet: An Interpretable, Well-Calibrated Six-CNN Ensemble for Dermoscopic Skin-Lesion Classification

【字体: 时间:2025年12月09日 来源:IEEE Access 3.6

编辑推荐:

  本研究针对皮肤镜图像分类中类别相似性高、数据集不平衡以及设备域差异大等挑战,提出了一种名为SynthraXCoreNet的六模型CNN集成方法。该方法融合了ResNet50V2、ResNet101V2、ResNet152V2、DenseNet201、NASNetLarge和Xception六种架构,通过后验概率平均实现分类。研究在四个公开数据集(HAM10000、ISIC2019、BCN20000和DERM12345)上验证了模型性能,准确率均超过95%,同时采用集成梯度、Grad-CAM等多种可解释性方法生成归因图,并通过温度缩放技术改善概率校准。该研究为皮肤病变分类提供了高精度、可解释的决策支持工具,推动了深度学习在临床 dermatology 中的应用。

  
在皮肤癌诊断领域,准确分类皮肤病变图像一直面临着诸多挑战。不同类别的皮肤病变在视觉上具有高度相似性,比如黑色素瘤与色素痣往往难以区分;医疗数据集普遍存在类别不平衡问题,常见病变样本数量远超罕见病变;此外,不同医疗机构使用的成像设备差异导致图像特征分布不一致,给模型泛化带来困难。这些因素共同制约着深度学习模型在临床实践中的可靠应用。
为了解决这些问题,研究人员在《IEEE Access》上发表了题为"SynthraXCoreNet: An Interpretable, Well-Calibrated Six-CNN Ensemble for Dermoscopic Skin-Lesion Classification"的研究论文。该研究开发了一个包含六种卷积神经网络架构的集成模型,通过综合多种网络的互补优势,提高分类性能的同时确保概率输出的可靠性和决策过程的可解释性。
研究人员采用的关键技术方法包括:使用六种预训练的CNN架构(ResNet50V2、ResNet101V2、ResNet152V2、DenseNet201、NASNetLarge和Xception)构建集成模型;采用后验概率平均的软投票融合策略;应用温度缩放技术进行概率校准;结合多种可解释性方法(集成梯度、Grad-CAM及其变体等)生成归因图;在四个公开数据集(HAM10000、ISIC2019、BCN20000和DERM12345)上进行系统评估,包含超过50,000张皮肤镜图像和40个诊断类别。
模型架构与集成策略
SynthraXCoreNet的核心创新在于其集成设计。研究团队选择了六种具有不同 inductive biases(归纳偏置)的CNN架构:ResNet系列通过残差连接解决深度网络训练难题,DenseNet201采用密集连接促进特征重用,Xception利用深度可分离卷积提高效率,NASNetLarge则基于神经架构搜索优化网络结构。每个 backbone(骨干网络)都独立在特定数据集上进行训练,在推理阶段通过共享输入层将图像同时馈送给所有模型,然后对它们的后验概率进行平均。这种 late fusion(晚期融合)策略既保持了各模型的独立性,又通过 ensemble averaging(集成平均)降低了预测方差。
概率校准与可靠性评估
除了传统的准确率指标,研究团队特别关注模型输出的概率校准质量。他们使用 reliability diagrams(可靠性图)和 Expected Calibration Error(ECE,预期校准误差)来评估概率校准情况。结果显示,未经校准的模型存在轻微 over-confidence(过度自信)现象,而应用 temperature scaling(温度缩放)后,ECE值降低了40-60%,概率输出更加可靠。这种校准改进对于临床决策支持至关重要,因为医生需要依赖模型给出的风险概率来制定诊疗方案。
可解释性分析
在可解释性方面,研究综合了多种 attribution methods(归因方法),包括Grad-CAM、Grad-CAM++、LayerCAM、ScoreCAM、Faster-ScoreCAM、Vanilla Saliency、SmoothGrad和Integrated Gradients(IG)。通过 functionally grounded evaluation(功能基础评估),他们从 faithfulness(忠实度)、robustness(鲁棒性)、complexity(复杂性)和 axiomatic properties(公理属性)四个维度量化了不同解释方法的质量。结果显示,在 confident predictions(自信预测)情况下,各种方法产生的归因图高度一致,都集中在病变区域;而在 misclassified cases(误分类案例)中,归因图则呈现出分散、多焦点模式,反映了模型的不确定性。
多数据集性能验证
在四个公开数据集上的实验结果表明,SynthraXCoreNet consistently(一致地)超越了单个 backbone 模型的性能。在HAM10000数据集上,模型达到了98.85%的准确率(95% Wilson CI ±0.68),Macro-F1分数为0.99;在包含更多类别的ISIC2019数据集上,准确率为95.30%(±0.83),Macro-F1为0.95;在BCN20000数据集上,准确率为95.00%(±1.21);在最具挑战性的DERM12345数据集(包含40个子类)上,模型仍实现了96.63%的准确率(±1.00)和0.97的Macro-F1分数。这些结果证明了集成策略在 handling class imbalance(处理类别不平衡)和 dataset shift(数据集偏移)方面的有效性。
计算效率与临床可行性
考虑到临床部署的实际需求,研究团队还详细报告了模型的计算特性。在NVIDIA RTX 4070 Ti GPU上,单张图像的端到端推理延迟约为461毫秒(batch size为1,FP32精度),相当于每分钟处理约130张图像。这种效率水平支持交互式临床决策场景,医生可以在实际诊疗过程中获得近乎实时的辅助诊断建议。
错误模式分析
通过对 confusion matrices(混淆矩阵)的深入分析,研究人员识别出了系统性的错误模式。最常见的混淆发生在黑色素瘤(MEL)和色素痣(NV)之间,这反映了临床实践中也存在的诊断难题。其他常见的混淆包括良性角化样病变(BKL)与色素痣(NV)之间的误分类。这些发现为未来模型改进提供了明确方向,提示需要引入更多 discriminative features(判别性特征)或结合临床 metadata(元数据)来区分视觉上相似的病变。
消融实验与权重优化
研究还进行了 ablation study(消融实验),比较了 equal-weight soft voting(等权软投票)和 weighted soft voting(加权软投票)的效果。结果表明,通过在验证集上学习最优的融合权重,可以在保持准确率的同时进一步改善概率校准,ECE值有显著降低。这种加权融合策略为不同模型分配不同的重要性权重,使集成更加灵活和自适应。
与现有技术的比较
与 state-of-the-art(最先进)系统相比,SynthraXCoreNet在多个基准测试中都表现出竞争优势。特别是在细粒度分类任务(如DERM12345的40子类分类)上,集成模型的优势更加明显。研究团队强调,他们的贡献不仅在于提出新的架构,更重要的是建立了一个全面、可复现的评估框架,统一了预处理、数据分割、评估指标和可解释性分析的标准。
研究的讨论部分明确指出,SynthraXCoreNet旨在作为临床决策支持工具,而非完全自主的诊断系统。当前工作主要针对单病灶皮肤镜图像,多病灶情况下的处理和分割将是未来的扩展方向。此外,概率校准和可解释性分析为医生提供了模型决策的透明视图,有助于建立临床信任,但最终诊断责任仍然在医生身上。
研究意义与未来方向
SynthraXCoreNet的研究在多个方面推动了皮肤镜图像分析领域的发展。首先,它证明了精心设计的CNN集成策略可以显著提高分类性能,特别是在面对类别不平衡和领域偏移时。其次,研究强调了概率校准的重要性,并展示了如何通过温度缩放等简单技术改善概率质量。第三,研究提供了一套完整的可解释性分析框架,帮助用户理解模型决策依据。最后,详细的 computational characteristics(计算特性)报告为临床部署提供了实用参考。
未来工作将围绕几个关键方向展开:进行多中心前瞻性验证,评估模型在真实临床环境中的表现;通过知识蒸馏、剪枝和量化等技术优化模型效率,使其更适合边缘设备部署;扩展可解释性方法,引入反事实解释和概念基础解释;开展公平性审计,确保模型在不同肤色、设备和采集设置下的均衡性能;探索开放集识别和罕见类别建模,应对临床实践中不断出现的新挑战。
这项研究为皮肤病变分类建立了一个新的基准,通过集成学习、概率校准和可解释性分析的有机结合,为深度学习在医学影像中的可靠应用提供了重要范例。随着技术的不断成熟和临床验证的深入,这类系统有望成为 dermatologist(皮肤科医生)的得力助手,提高皮肤癌早期诊断的准确性和可及性。
相关新闻
生物通微信公众号
微信
新浪微博
  • 急聘职位
  • 高薪职位

知名企业招聘

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号