PeLK: Parameter-efficient Large Kernel ConvNets with Peripheral Convolution

被引:12
作者
Chen, Honghao [1 ,2 ,5 ]
Chu, Xiangxiang [3 ]
Ren, Yongjian [1 ,2 ]
Zhao, Xin [1 ,2 ]
Huang, Kaiqi [1 ,2 ,4 ]
机构
[1] Chinese Acad Sci, Inst Automat, Beijing, Peoples R China
[2] Univ Chinese Acad Sci, Sch Artificial Intelligence, Beijing, Peoples R China
[3] Meituan, Beijing, Peoples R China
[4] CAS Ctr Excellence Brain Sci & Intelligence Techn, Beijing, Peoples R China
[5] Meituan Inc, Beijing, Peoples R China
来源
2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2024 | 2024年
基金
中国国家自然科学基金; 国家重点研发计划;
关键词
D O I
10.1109/CVPR52733.2024.00531
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recently, some large kernel convnets strike back with appealing performance and efficiency. However, given the square complexity of convolution, scaling up kernels can bring about an enormous amount of parameters and the proliferated parameters can induce severe optimization problem. Due to these issues, current CNNs compromise to scale up to 51 x 51 in the form of stripe convolution ( i.e., 51 x 5 + 5 x 51) and start to saturate as the kernel size continues growing. In this paper, we delve into addressing these vital issues and explore whether we can continue scaling up kernels for more performance gains. Inspired by human vision, we propose a human-like peripheral convolution that efficiently reduces over 90% parameter count of dense grid convolution through parameter sharing, and manage to scale up kernel size to extremely large. Our peripheral convolution behaves highly similar to human, reducing the complexity of convolution from O(K-2) to O(log K) without backfiring performance. Built on this, we propose Parameter-efficient Large Kernel Network (PeLK). Our PeLK outperforms modern vision Transformers and ConvNet architectures like Swin, ConvNeXt, RepLKNet and SLaK on various vision tasks including ImageNet classification, semantic segmentation on ADE20K and object detection on MS COCO. For the first time, we successfully scale up the kernel size of CNNs to an unprecedented 101 x 101 and demonstrate consistent improvements.
引用
收藏
页码:5557 / 5567
页数:11
相关论文
共 53 条
  • [51] ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices
    Zhang, Xiangyu
    Zhou, Xinyu
    Lin, Mengxiao
    Sun, Ran
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 6848 - 6856
  • [52] Cascade R-CNN: Delving into High Quality Object Detection
    Cai, Zhaowei
    Vasconcelos, Nuno
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 6154 - 6162
  • [53] Semantic Understanding of Scenes Through the ADE20K Dataset
    Zhou, Bolei
    Zhao, Hang
    Puig, Xavier
    Xiao, Tete
    Fidler, Sanja
    Barriuso, Adela
    Torralba, Antonio
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2019, 127 (03) : 302 - 321