山西大学学报(自然科学版)2026,Vol.49Issue(2):284-294,11.DOI:10.13451/j.sxu.ns.2024087
基于CNN与Transformer混合的轻量化皮肤病变分类网络
A Lightweight Classification Network for Skin Lesion Based on the Hybridization of CNN and Transformer
摘要
Abstract
Existing classification algorithms often suffer from a large number of parameters,high computational complexity,and sub-optimal classification performance in classifying skin lesion images,due to their uneven distribution attibutes.To address this issue,in this paper,we propose a lightweight transformer module and a novel strategy that combines Convolutional Neural Network(CNN)with Transformer to enhance network classification performance.Additionally,we adopt an inverse class loss function weighting scheme to mitigate the impact of imbalanced image category distribution during training.The lightweight transformer ex-tracts essential features from input sequences,and performs separable self-attention computations to capture global feature informa-tion from skin lesion regions.This approach addresses the computational limitations of traditional transformer.Furthermore,our new strategy effectively integrates shallow global detail features with deep semantic features,enhancing the network's expressive ability.Experimental results on the HAM10000 dataset demonstrate that our algorithm outperforms other comparative methods in terms of evaluation metrics.Remarkably,we achieve these results while maintaining a model size of only 2.3 million parameters,which holds significant promise for advancing automatic skin lesion classification tools.关键词
图像分类/皮肤病变分类/轻量化网络/混合网络/HAM10000Key words
image classification/skin lesions classification/lightweight network/hybrid network/HAM10000分类
信息技术与安全科学引用本文复制引用
徐健,赵欣,李鑫杰..基于CNN与Transformer混合的轻量化皮肤病变分类网络[J].山西大学学报(自然科学版),2026,49(2):284-294,11.基金项目
国家自然科学基金(61971424) (61971424)