UPANets: Learning from the Universal Pixel Attention Neworks
Ching-Hsun Tseng
(Department of Computer Science, The University of Manchester, Manchester M13 9PL, UK)
Shin-Jye Lee
(Institute of Management of Technology, National Chiao Tung University, Hsinchu 300, Taiwan)
Jianan Feng
(School of Software, Yunnan University, Kunming 650504, China)
Shengzhong Mao
(Department of Computer Science, The University of Manchester, Manchester M13 9PL, UK)
Yu-Ping Wu
(Department of Computer Science, The University of Manchester, Manchester M13 9PL, UK)
Jia-Yu Shang
(Department of Computer Science, The University of Manchester, Manchester M13 9PL, UK)
Xiao-Jun Zeng
(Department of Computer Science, The University of Manchester, Manchester M13 9PL, UK)
Journal: Entropy
•
• MDPI AG
• ISSN: 1099-4300
Abstract
With the successful development in computer vision, building a deep convolutional neural network (CNNs) has been mainstream, considering the character of shared parameters in a convolutional layer. Stacking convolutional layers into a deep structure improves performance, but over-stacking also ramps up the needed resources for GPUs. Seeing another surge of Transformers in computer vision, the issue has aroused severely. A resource-hungry model is hardly implemented for limited hardware or single-customers-based GPU. Therefore, this work focuses on these concerns and proposes an efficient but robust backbone, which equips with channel and spatial direction attentions, so the attentions help to expand receptive fields in shallow convolutional layers and pass the information to every layer. An attention-boosted network based on already efficient CNNs, Universal Pixel Attention Networks (UPANets), is proposed. Through a series of experiments, UPANets fulfil the purposes of learning global information with less needed resources and outshine many existing SOTAs in CIFAR-{10, 100}.