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 条
  • [1] Bag of Tricks for Electrocardiogram Classification With Deep Neural Networks
    Min, Seonwoo
    Choi, Hyun-Soo
    Han, Hyeongrok
    Seo, Minji
    Kim, Jin-Kook
    Park, Junsang
    Jung, Sunghoon
    Oh, Il-Young
    Lee, Byunghan
    Yoon, Sungroh
    [J]. 2020 COMPUTING IN CARDIOLOGY, 2020,
  • [2] Convolutional Neural Networks for image classification
    Jmour, Nadia
    Zayen, Sehla
    Abdelkrim, Afef
    [J]. 2018 INTERNATIONAL CONFERENCE ON ADVANCED SYSTEMS AND ELECTRICAL TECHNOLOGIES (IC_ASET), 2017, : 397 - 402
  • [3] Bag of Tricks for Long-Tailed Visual Recognition with Deep Convolutional Neural Networks
    Zhang, Yongshun
    Wei, Xiu-Shen
    Zhou, Boyan
    Wu, Jianxin
    [J]. THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 3447 - 3455
  • [4] Convolutional neural networks for hyperspectral image classification
    Yu, Shiqi
    Jia, Sen
    Xu, Chunyan
    [J]. NEUROCOMPUTING, 2017, 219 : 88 - 98
  • [5] Convolutional Neural Networks for Document Image Classification
    Kang, Le
    Kumar, Jayant
    Ye, Peng
    Li, Yi
    Doermann, David
    [J]. 2014 22ND INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2014, : 3168 - 3172
  • [6] CONVOLUTIONAL NEURAL NETWORKS IN THE TASK OF IMAGE CLASSIFICATION
    Zelenina, Larisa
    Khaimina, Liudmila
    Khaimin, Evgenii
    Khripunov, D.
    Zashikhina, Inga
    [J]. MATHEMATICS AND INFORMATICS, 2022, 65 (01): : 19 - 29
  • [7] Hyperspectral Image Classification with Convolutional Neural Networks
    Slavkovikj, Viktor
    Verstockt, Steven
    De Neve, Wesley
    Van Hoecke, Sofie
    Van de Walle, Rik
    [J]. MM'15: PROCEEDINGS OF THE 2015 ACM MULTIMEDIA CONFERENCE, 2015, : 1159 - 1162
  • [8] Preprocessing for Image Classification by Convolutional Neural Networks
    Pal, Kuntal Kumar
    Sudeep, K. S.
    [J]. 2016 IEEE INTERNATIONAL CONFERENCE ON RECENT TRENDS IN ELECTRONICS, INFORMATION & COMMUNICATION TECHNOLOGY (RTEICT), 2016, : 1778 - 1781
  • [9] Clifford Convolutional Neural Networks for Lymphoblast Image Classification
    Vieira, Guilherme
    Valle, Marcos Eduardo
    Lopes, Wilder
    [J]. ADVANCED COMPUTATIONAL APPLICATIONS OF GEOMETRIC ALGEBRA, ICACGA 2022, 2024, 13771 : 75 - 87
  • [10] Convolutional Neural Networks based Pornographic Image Classification
    Zhou, KaiLong
    Zhou, Li
    Geng, Zhen
    Zhang, Jing
    Li, Xiao Guang
    [J]. 2016 IEEE SECOND INTERNATIONAL CONFERENCE ON MULTIMEDIA BIG DATA (BIGMM), 2016, : 206 - 209