Efficient adaptive ensembling for image classification

被引:5
作者
Antonio, Bruno [1 ]
Moroni, Davide [1 ]
Martinelli, Massimo [1 ]
机构
[1] Italian Natl Res Council, Inst Informat Sci & Technol, Via Giuseppe Moruzzi 1, I-56124 Pisa, Italy
关键词
convolutional neural networks; deep learning; EfficientNet; ensemble; image classification;
D O I
10.1111/exsy.13424
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In recent times, with the exception of sporadic cases, the trend in computer vision is to achieve minor improvements compared to considerable increases in complexity. To reverse this trend, we propose a novel method to boost image classification performances without increasing complexity. To this end, we revisited ensembling, a powerful approach, often not used properly due to its more complex nature and the training time, so as to make it feasible through a specific design choice. First, we trained two EfficientNet-b0 end-to-end models (known to be the architecture with the best overall accuracy/complexity trade-off for image classification) on disjoint subsets of data (i.e., bagging). Then, we made an efficient adaptive ensemble by performing fine-tuning of a trainable combination layer. In this way, we were able to outperform the state-of-the-art by an average of 0.5% on the accuracy, with restrained complexity both in terms of the number of parameters (by 5-60 times), and the FLoating point Operations Per Second FLOPS by 10-100 times on several major benchmark datasets.
引用
收藏
页数:12
相关论文
共 36 条
[1]  
Bonawitz K., 2019, Proc. Mach. Learn. Res, V1, P374
[2]  
Bossard L, 2014, LECT NOTES COMPUT SC, V8694, P446, DOI 10.1007/978-3-319-10599-4_29
[3]  
Brock A., 2021, High-performance large-scale image recognition without normalization. CoRR
[4]  
abs/2102.06171
[5]  
Dai Z., 2021, ABS210604803 CORR
[6]  
Darlow L.N., 2018, CINIC-10 Is Not ImageNet or CIFAR-10
[7]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[8]   The carbon impact of artificial intelligence [J].
Dhar, Payal .
NATURE MACHINE INTELLIGENCE, 2020, 2 (08) :423-425
[9]   A survey on ensemble learning [J].
Dong, Xibin ;
Yu, Zhiwen ;
Cao, Wenming ;
Shi, Yifan ;
Ma, Qianli .
FRONTIERS OF COMPUTER SCIENCE, 2020, 14 (02) :241-258
[10]  
Dosovitskiy A., 2021, INT C LEARN REPR