Saved in:
Bibliographic Details
Main Authors: Bruno, Antonio, Moroni, Davide, Martinelli, Massimo
Format: Preprint
Published: 2022
Subjects:
Online Access:https://arxiv.org/abs/2206.07394
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909815488380928
author Bruno, Antonio
Moroni, Davide
Martinelli, Massimo
author_facet Bruno, Antonio
Moroni, Davide
Martinelli, Massimo
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2206_07394
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Efficient Adaptive Ensembling for Image Classification
Bruno, Antonio
Moroni, Davide
Martinelli, Massimo
Computer Vision and Pattern Recognition
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.
title Efficient Adaptive Ensembling for Image Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2206.07394