| 注册
首页|期刊导航|山西大学学报(自然科学版)|基于CNN与Transformer混合的轻量化皮肤病变分类网络

基于CNN与Transformer混合的轻量化皮肤病变分类网络

徐健 赵欣 李鑫杰

山西大学学报(自然科学版)2026,Vol.49Issue(2):284-294,11.
山西大学学报(自然科学版)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

徐健 1赵欣 1李鑫杰1

作者信息

  • 1. 大连大学 信息工程学院,辽宁 大连 116622
  • 折叠

摘要

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.

关键词

图像分类/皮肤病变分类/轻量化网络/混合网络/HAM10000

Key words

image classification/skin lesions classification/lightweight network/hybrid network/HAM10000

分类

信息技术与安全科学

引用本文复制引用

徐健,赵欣,李鑫杰..基于CNN与Transformer混合的轻量化皮肤病变分类网络[J].山西大学学报(自然科学版),2026,49(2):284-294,11.

基金项目

国家自然科学基金(61971424) (61971424)

山西大学学报(自然科学版)

0253-2395

访问量0
|
下载量0
段落导航相关论文