arXiv:2508.12484cs.CV2025-08被引 10

用混合模型提升皮肤癌分类准确率,融合局部与全局特征。

Skin Cancer Classification: Hybrid CNN-Transformer Models with KAN-Based Fusion

  • CNN+Transformer并行融合,结合柯尔莫哥洛夫网络实现非线性特征整合。
  • 在多个数据集上达到超91%准确率,最高达97.83%。
  • 适合医学图像分析、临床辅助诊断等场景的模型设计参考。

皮肤癌分类是医学图像分析的关键任务,精确区分恶性与良性病灶对早期诊断和治疗至关重要。本文探索了序列与并行混合的CNN-Transformer模型,并引入卷积型柯尔莫哥洛夫-阿诺德网络(CKAN)。方法结合迁移学习与大量数据增强,其中CNN提取局部空间特征,Transformer建模全局依赖关系,而CKAN通过可学习激活函数实现非线性特征融合,提升表征能力。为评估泛化性能,模型在多个基准数据集(HAM10000、BCN20000和PAD-UFES)上测试,涵盖不同数据分布与类别不平衡情况。实验表明,混合架构能有效捕捉空间与上下文特征,提升分类性能;引入CKAN后,特征融合更优,生成更具判别性的表示。所提方法在各数据集表现优异:于HAM10000上达92.81%准确率与92.47% F1-score,PAD-UFES上为97.83%准确率与F1-score,BCN20000上为91.17%准确率与91.79% F1-score,验证了模型跨数据集的有效性与泛化能力。研究强调特征表示与模型设计在推动鲁棒、精准医学图像分类中的关键作用。

原文摘要 · Abstract (English)

Skin cancer classification is a crucial task in medical image analysis, where precise differentiation between malignant and non-malignant lesions is essential for early diagnosis and treatment. In this study, we explore Sequential and Parallel Hybrid CNN-Transformer models with Convolutional Kolmogorov-Arnold Network (CKAN). Our approach integrates transfer learning and extensive data augmentation, where CNNs extract local spatial features, Transformers model global dependencies, and CKAN facilitates nonlinear feature fusion for improved representation learning. To assess generalization, we evaluate our models on multiple benchmark datasets (HAM10000,BCN20000 and PAD-UFES) under varying data distributions and class imbalances. Experimental results demonstrate that hybrid CNN-Transformer architectures effectively capture both spatial and contextual features, leading to improved classification performance. Additionally, the integration of CKAN enhances feature fusion through learnable activation functions, yielding more discriminative representations. Our proposed approach achieves competitive performance in skin cancer classification, demonstrating 92.81% accuracy and 92.47% F1-score on the HAM10000 dataset, 97.83% accuracy and 97.83% F1-score on the PAD-UFES dataset, and 91.17% accuracy with 91.79% F1- score on the BCN20000 dataset highlighting the effectiveness and generalizability of our model across diverse datasets. This study highlights the significance of feature representation and model design in advancing robust and accurate medical image classification.

皮肤癌分类CNN-TransformerCKAN医学图像

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