arXiv · 2105.09121
Single-Layer Vision Transformers for More Accurate Early Exits with Less Overhead
Abstract
Deploying deep learning models in time-critical applications with limited computational resources, for instance in edge computing systems and IoT networks, is a challenging task that often relies on dynamic inference methods such as early exiting. In this paper, we introduce a novel architecture for early exiting based on the vision transformer architecture, as well as a fine-tuning strategy that significantly increase the accuracy of early exit branches compared to conventional approaches while introducing less overhead. Through extensive experiments on image and audio classification as well as audiovisual crowd counting, we show that our method works for both classification and regression problems, and in both single- and multi-modal settings. Additionally, we introduce a novel method for integrating audio and visual modalities within early exits in audiovisual data analysis, that can lead to a more fine-grained dynamic inference.
Explore related subjects
Keep this discovery
Arian Bakhtiarnia, Qi Zhang, Alexandros Iosifidis. 2021-05-19. Single-Layer Vision Transformers for More Accurate Early Exits with Less Overhead. https://arxiv.org/abs/2105.09121
Cite the original work for its findings. Save a collection to share your selection of sources.