Self-Fusion Convolutional Neural Networks.

Shenjian Gong,Shanshan Zhang,Jian Yang,Pong Chi Yuen
DOI: https://doi.org/10.1016/j.patrec.2021.08.022
IF: 4.757
2021-01-01
Pattern Recognition Letters
Abstract:Efficiency is an important concern for practical applications, therefore, it is of great importance to build effective lightweight networks. This paper proposes a novel lightweight feature self-fusion convolutional (SFC) module, which consists of self-fusion and point-wise convolution. The core of SFC is a three-step self-fusion. First, each input feature map is expanded to a high dimensional space individually, prohibiting connections with other input channels. Then, in the second step, we fuse all features from the same input in the high dimensional space to enhance the representation ability. Finally, we compress high dimensional features to a low dimensional space. After self-fusion, we connect all features by one point-wise convolution. Compared to inverted bottleneck, SFC module decreases the number of parameters by replacing the dense connections among channels with self-fusion. To the best of our knowledge, SFC is the first method to build lightweight networks by feature self-fusion. We then build a new network namely SFC-Net, by stacking SFC modules. Experimental results on the CI -FAR and downsampled ImageNet datasets demonstrate our SFC-Net achieves better performance to some previous popular CNNs with fewer number of parameters and achieves comparable performance com-pared to other previous lightweight architectures. The code is available at https://github.com/Yankeegsj/ Self-fusion . (c) 2021 Elsevier B.V. All rights reserved.
What problem does this paper attempt to address?