用改进的Transformer模型提升猴痘皮损图像分类准确率
RSwinV2-MD: An Enhanced Residual SwinV2 Transformer for Monkeypox Detection from Skin Images
- 基于SwinV2构建分层结构,结合滑动窗口与多头注意力增强全局关联
- 在Kaggle数据集上达96.51%准确率和96.13%F1分数,优于CNN和原始SwinV2
- 引入逆残差模块缓解梯度消失,更好区分猴痘与水痘等相似病症
本文提出一种名为自定义残差SwinTransformer V2(RSwinV2)的深度学习方法,用于猴痘诊断,通过工具辅助视觉分析提升病灶分类能力。该方法根据输入维度、嵌入结构及输出目标定制了Transformer的分层架构,将输入图像分割为非重叠块,并利用滑动窗口与注意力机制处理,有效避免了非重叠区域间的局部性问题,同时保持计算高效。相较于SwinTransformer,RSwinV2引入了补丁嵌入与位置嵌入,结合多头注意力实现全局连接能力。此外,还设计并集成逆残差块(IRB),通过卷积跳跃连接缓解训练中的梯度消失问题,从而同时捕捉全局与局部模式特征。该方法显著提升了对猴痘、水痘、麻疹及牛痘病灶的区分能力。在Kaggle公开数据集上的测试表明,其准确率达到96.51%,F1分数为96.13%,优于标准CNN模型和原始SwinTransformer,验证了其作为猴痘皮损辅助判读工具的有效性。
原文摘要 · Abstract (English)
In this paper, a deep learning approach for Mpox diagnosis named Customized Residual SwinTransformerV2 (RSwinV2) has been proposed, trying to enhance the capability of lesion classification by employing the RSwinV2 tool-assisted vision approach. In the RSwinV2 method, a hierarchical structure of the transformer has been customized based on the input dimensionality, embedding structure, and output targeted by the method. In this RSwinV2 approach, the input image has been split into non-overlapping patches and processed using shifted windows and attention in these patches. This process has helped the method link all the windows efficiently by avoiding the locality issues of non-overlapping regions in attention, while being computationally efficient. RSwinV2 has further developed based on SwinTransformer and has included patch and position embeddings to take advantage of the transformer global-linking capability by employing multi-head attention in these embeddings. Furthermore, RSwinV2 has developed and incorporated the Inverse Residual Block (IRB) into this method, which utilizes convolutional skip connections with these inclusive designs to address the vanishing gradient issues during processing. RSwinV2 inclusion of IRB has therefore facilitated this method to link global patterns as well as local patterns; hence, its integrity has helped improve lesion classification capability by minimizing variability of Mpox and increasing differences of Mpox, chickenpox, measles, and cowpox. In testing SwinV2, its accuracy of 96.51 and an F1score of 96.13 have been achieved on the Kaggle public dataset, which has outperformed standard CNN models and SwinTransformers; the RSwinV2 vector has thus proved its validity as a computer-assisted tool for Mpox lesion observation interpretation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。