Tiny Models are the Computational Saver for Large Models

26 Mar 2024  ·  Qingyuan Wang, Barry Cardiff, Antoine Frappé, Benoit Larras, Deepu John ·

This paper introduces TinySaver, an early-exit-like dynamic model compression approach which employs tiny models to substitute large models adaptively. Distinct from traditional compression techniques, dynamic methods like TinySaver can leverage the difficulty differences to allow certain inputs to complete their inference processes early, thereby conserving computational resources. Most existing early exit designs are implemented by attaching additional network branches to the model's backbone. Our study, however, reveals that completely independent tiny models can replace a substantial portion of the larger models' job with minimal impact on performance. Employing them as the first exit can remarkably enhance computational efficiency. By searching and employing the most appropriate tiny model as the computational saver for a given large model, the proposed approaches work as a novel and generic method to model compression. This finding will help the research community in exploring new compression methods to address the escalating computational demands posed by rapidly evolving AI models. Our evaluation of this approach in ImageNet-1k classification demonstrates its potential to reduce the number of compute operations by up to 90%, with only negligible losses in performance, across various modern vision models. The code of this work will be available.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Image Classification ImageNet TinySaver(Swin_large, 0.5 Acc drop) Top 1 Accuracy 85.74 # 198
Image Classification ImageNet TinySaver(Swin_large, 1.0 Acc drop) Top 1 Accuracy 85.24 # 237
Image Classification ImageNet TinySaver(EfficientFormerV2_l, 0.01 Acc drop) Top 1 Accuracy 83.52 # 390
Image Classification ImageNet TinySaver(ConvNeXtV2_h, 0.5 Acc drop) Top 1 Accuracy 85.75 # 197
GFLOPs 19.41 # 366
Image Classification ImageNet TinySaver(ConvNeXtV2_h, 0.01 Acc drop) Top 1 Accuracy 86.24 # 163
GFLOPs 31.17 # 392

Methods