UltraViT: Latency-Optimized On-device Vision Encoder for Large Vision-Language Models
Abstract
Large Vision-Language Models (LVLMs) remain bottlenecked by massive computational footprints, precluding their deployment on resource-constrained edge devices. While efforts to compress LVLMs focus heavily on vision token reduction or smaller language models, the vision encoder is largely overlooked, typically deployed as a monolithic, computationally heavy feature extractor. Moreover, there is no previous effort that designs a vision encoder for LVLMs directly optimized for on-device latency. In this paper, we present UltraViT, a vision encoder for LVLMs, explicitly designed and optimized for on-device performance. Specifically, by taking into account real on-device latencies, we systematically design a pyramidal architecture that strategically integrates and adapts heterogeneous spatial mixers at the macro-block level. Furthermore, to pre-train UltraViT, we propose a novel two-stage generative pre-training strategy: cultivating rich spatial features via dense distillation, followed by direct generative supervision from a capacity-mixed frozen LLM. Compared to standard contrastive and SSL, we show that our pre-training is much more effective for achieving high-level semantic grounding for UltraViT needed for the subsequent generative multimodal alignment of LVLM training. Extensive experiments demonstrate that our on-device latency-informed design combined with our tailored training strategy establishes a new state-of-the-art for efficient LVLM encoding, significantly outperforming existing encoder-centric baselines while operating on-device at nearly 1.7xthe speed.
Community
Existing approaches for improving the efficiency of Large Vision–Language Models (LVLMs) primarily focus on reducing visual tokens or adopting smaller language models, while largely overlooking the vision encoder, which often remains a monolithic and computationally expensive component. In this work, we address this limitation by introducing UltraViT, a vision encoder explicitly designed and optimized for low-latency deployment on resource-constrained edge devices. Specifically, UltraViT uses real, on-device latency measurements to guide the systematic design of a pyramidal architecture that strategically integrates and adapts heterogeneous spatial mixers at the macro-block level. To effectively pre-train this efficient architecture, we further propose a two-stage generative pre-training strategy: dense distillation first captures rich spatial representations, followed by direct generative supervision from a frozen, capacity-mixed LLM to improve high-level semantic grounding. Compared with standard contrastive and self-supervised learning objectives, this strategy better prepares UltraViT for the generative, multimodal alignment required in LVLM training. Extensive experiments demonstrate that combining latency-aware architecture design with tailored generative pre-training establishes a new state of the art among efficient LVLM vision encoders, significantly outperforming existing encoder-centric baselines while achieving nearly 1.7× faster on-device inference.
Get this paper in your agent:
hf papers read 2607.23373 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper