Search

In order to solve the problem that the Transformer model cannot apply to mobile devices with limited computational resources due to its huge number of parameters and computational complexity, this paper proposed a mobile-friendly lightweight image recognition network called FuseNet. FuseNet utilized convolutional neural network to extract local feature information and self-attention to excel in modeling global information, and it integrated the features of both local and global representations into a single feature extraction module, which efficiently combined the advantages of the two different structures to achieve a high accuracy with a small model size. Experiments demonstrate that FuseNet with different model sizes can achieve good performance without using pre-training and it can well apply to mobile devices. FuseNet-B achieves 80. 5% accuracy with 14. 8 M parameters on ImageNet-1 K dataset, the performance exceeds the same volume of Transformer models and convolutional neural networks. [ABSTRACT FROM AUTHOR]