English

Energy-Efficient Vision Transformer Inference for Edge-AI Deployment

Machine Learning 2025-12-01 v1

Abstract

The growing deployment of Vision Transformers (ViTs) on energy-constrained devices requires evaluation methods that go beyond accuracy alone. We present a two-stage pipeline for assessing ViT energy efficiency that combines device-agnostic model selection with device-related measurements. We benchmark 13 ViT models on ImageNet-1K and CIFAR-10, running inference on NVIDIA Jetson TX2 (edge device) and an NVIDIA RTX 3050 (mobile GPU). The device-agnostic stage uses the NetScore metric for screening; the device-related stage ranks models with the Sustainable Accuracy Metric (SAM). Results show that hybrid models such as LeViT_Conv_192 reduce energy by up to 53% on TX2 relative to a ViT baseline (e.g., SAM5=1.44 on TX2/CIFAR-10), while distilled models such as TinyViT-11M_Distilled excel on the mobile GPU (e.g., SAM5=1.72 on RTX 3050/CIFAR-10 and SAM5=0.76 on RTX 3050/ImageNet-1K).

Keywords

Cite

@article{arxiv.2511.23166,
  title  = {Energy-Efficient Vision Transformer Inference for Edge-AI Deployment},
  author = {Nursultan Amanzhol and Jurn-Gyu Park},
  journal= {arXiv preprint arXiv:2511.23166},
  year   = {2025}
}
R2 v1 2026-07-01T07:59:24.031Z