arXiv · 2602.13334
Ask the Expert: Collaborative Inference for Vision Transformers with Near-Edge Accelerators
Abstract
Deploying Vision Transformers on edge devices is challenging due to their high computational complexity, while full offloading to cloud resources presents significant latency overheads. We propose a novel collaborative inference framework, which orchestrates a lightweight generalist ViT on an edge device and multiple medium-sized expert ViTs on a near-edge accelerator. A novel routing mechanism uses the edge model's Top-$\mathit{k}$ predictions to dynamically select the most relevant expert for samples with low confidence. We further design a progressive specialist training strategy to enhance expert accuracy on dataset subsets. Extensive experiments on the CIFAR-100 dataset using a real-world edge and near-edge testbed demonstrate the superiority of our framework. Specifically, the proposed training strategy improves expert specialization accuracy by 4.12% on target subsets and enhances overall accuracy by 2.76% over static experts. Moreover, our method reduces latency by up to 45% compared to edge execution, and energy consumption by up to 46% compared to just near-edge offload.
Explore related subjects
Keep this discovery
Hao Liu, Suhaib A. Fahmy. 2026-02-11. Ask the Expert: Collaborative Inference for Vision Transformers with Near-Edge Accelerators. https://arxiv.org/abs/2602.13334
Cite the original work for its findings. Save a collection to share your selection of sources.