Bag of Tricks for Image Classification with Convolutional Neural Networks

被引:882
|
作者
He, Tong [1 ]
Zhang, Zhi [1 ]
Zhang, Hang [1 ]
Zhang, Zhongyue [1 ]
Xie, Junyuan [1 ]
Li, Mu [1 ]
机构
[1] Amazon Web Serv, Seattle, WA 98109 USA
关键词
D O I
10.1109/CVPR.2019.00065
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Much of the recent progress made in image classification research can be credited to training procedure refinements, such as changes in data augmentations and optimization methods. In the literature, however, most refinements are either briefly mentioned as implementation details or only visible in source code. In this paper, we will examine a collection of such refinements and empirically evaluate their impact on the final model accuracy through ablation study. We will show that, by combining these refinements together, we are able to improve various CNN models significantly. For example, we raise ResNet-50's top-1 validation accuracy from 75.3% to 79.29% on ImageNet. We will also demonstrate that improvement on image classification accuracy leads to better transfer learning performance in other application domains such as object detection and semantic segmentation.
引用
收藏
页码:558 / 567
页数:10
相关论文
共 50 条
  • [21] Improving the Performance of Convolutional Neural Networks for Image Classification
    Giveki, Davar
    [J]. OPTICAL MEMORY AND NEURAL NETWORKS, 2021, 30 (01) : 51 - 66
  • [22] Hyperspectral Image Classification using Convolutional Neural Networks
    Shambulinga, M.
    Sadashivappa, G.
    [J]. INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2021, 12 (06) : 702 - 708
  • [23] Deformable Convolutional Neural Networks for Hyperspectral Image Classification
    Zhu, Jian
    Fang, Leyuan
    Ghamisi, Pedram
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2018, 15 (08) : 1254 - 1258
  • [24] Analysis of Convolutional Neural Networks for Document Image Classification
    Tensmeyer, Chris
    Martinez, Tony
    [J]. 2017 14TH IAPR INTERNATIONAL CONFERENCE ON DOCUMENT ANALYSIS AND RECOGNITION (ICDAR), VOL 1, 2017, : 388 - 393
  • [25] Improved Convolutional Neural Networks for Hyperspectral Image Classification
    Kalita, Shashanka
    Biswas, Mantosh
    [J]. RECENT DEVELOPMENTS IN MACHINE LEARNING AND DATA ANALYTICS, 2019, 740 : 397 - 410
  • [26] Mirror invariant convolutional neural networks for image classification
    Lu, Shufang
    Li, Yan
    Wang, Minqian
    Gao, Fei
    [J]. IET IMAGE PROCESSING, 2022, 16 (06) : 1626 - 1635
  • [27] Evolving Deep Convolutional Neural Networks for Image Classification
    Sun, Yanan
    Xue, Bing
    Zhang, Mengjie
    Yen, Gary G.
    [J]. IEEE TRANSACTIONS ON EVOLUTIONARY COMPUTATION, 2020, 24 (02) : 394 - 407
  • [28] Hierarchical Transfer Convolutional Neural Networks for Image Classification
    Dong, Xishuang
    Wu, Hsiang-Huang
    Yan, Yuzhong
    Qian, Lijun
    [J]. 2019 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2019, : 2817 - 2825
  • [29] Quantum convolutional neural networks for multiclass image classification
    Shi, Shangshang
    Wang, Zhimin
    Li, Jiaxin
    Li, Yanan
    Shang, Ruimin
    Zhong, Guoqiang
    Gu, Yongjian
    [J]. QUANTUM INFORMATION PROCESSING, 2024, 23 (05)
  • [30] Whole Mammogram Image Classification With Convolutional Neural Networks
    Zhang, Xiaofei
    Zhang, Yi
    Han, Erik Y.
    Jacobs, Nathan
    Han, Qiong
    Wang, Xiaoqin
    Liu, Jinze
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE (BIBM), 2017, : 700 - 704