计算机科学
变压器
卷积神经网络
人工智能
计算复杂性理论
模式识别(心理学)
机器学习
算法
电压
工程类
电气工程
作者
Mengxing Li,Ying Song,Bo Wang
标识
DOI:10.1109/vrw55335.2022.00041
摘要
In the process of metaverse construction, in order to achieve better interaction, it is necessary to provide clear semantic information for each object. Image classification technology plays a very important role in this process. Based on CMT transformer and improved Cross-Shaped Window Self-Attention, this paper presents an improved Image classification framework combining CNN and transformers, which is called CWCT transformer. Due to the high resolution of the image, vision transformers will lead to too high model complexity and too much calculation. To solve this problem, CWCT captures local features by using optimized Cross-Window Self-Attention mechanism and global features by using convolutional neural networks (CNN) stack. This structure has the flexibility to model at various scales and has linear computational complexity concerning image size. Compared with the original CMT network, the classification accuracy has been improved on ImageNet-1k and randomly screened Tiny-ImageNet dataset. Thanks to the optimized Cross-Window Self-Attention, the CWCT proposed in this paper has a significant improvement in operation speed and model complexity compared with CMT.
科研通智能强力驱动
Strongly Powered by AbleSci AI