Rethinking the Value of Local Feature Fusion in Convolutional Neural Networks

NEURAL PROCESSING LETTERS(2023)

引用 0|浏览17
暂无评分
摘要
Traditional CNN head for classification tasks typically consists of a global average pooling layer before the last fully-connected classifier. However, such a simple and light-weighted head lacks the ability of feature fusion, and can’t give full play to the strong feature extraction ability of the network body. In the present work, we analyze the Basic Block and Bottleneck structure in ResNet in-depth and reveal the importance of performing feature fusion inside local patches via 1× 1 convolution. We propose a new head structure consisting of three stages with a series of 1× 1 convolution to replace global average pooling. With little additional FLOPs and inference speed drop, our new head improves the accuracy for ResNet18 by 3.6
更多
查看译文
关键词
CNN,CNN head,Network architecture design,Local feature fusion
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要