English

Multi-Exit Vision Transformer for Dynamic Inference

Computer Vision and Pattern Recognition 2021-10-25 v3

Abstract

Deep neural networks can be converted to multi-exit architectures by inserting early exit branches after some of their intermediate layers. This allows their inference process to become dynamic, which is useful for time critical IoT applications with stringent latency requirements, but with time-variant communication and computation resources. In particular, in edge computing systems and IoT networks where the exact computation time budget is variable and not known beforehand. Vision Transformer is a recently proposed architecture which has since found many applications across various domains of computer vision. In this work, we propose seven different architectures for early exit branches that can be used for dynamic inference in Vision Transformer backbones. Through extensive experiments involving both classification and regression problems, we show that each one of our proposed architectures could prove useful in the trade-off between accuracy and speed.

Keywords

Cite

@article{arxiv.2106.15183,
  title  = {Multi-Exit Vision Transformer for Dynamic Inference},
  author = {Arian Bakhtiarnia and Qi Zhang and Alexandros Iosifidis},
  journal= {arXiv preprint arXiv:2106.15183},
  year   = {2021}
}

Comments

Accepted by the 2021 British Machine Vision Conference (BMVC 2021)

R2 v1 2026-06-24T03:42:16.964Z