为边缘设备设计的低延迟视觉编码器,让大模型更快更省资源。
UltraViT: Latency-Optimized On-device Vision Encoder for Large Vision-Language Models

- 基于实际设备延迟优化,采用分层架构与异构空间混合器。
- 在边缘设备上运行速度达基准方案1.7倍,性能显著提升。
- 适合需要高效部署大视觉语言模型的移动端应用。
大型视觉语言模型(LVLM)因计算开销巨大,难以部署在资源受限的边缘设备上。现有压缩方法多聚焦于减少视觉标记或使用小型语言模型,却忽视了视觉编码器本身——通常作为单一、高耗能特征提取器。此前尚无专为边缘设备延迟优化的视觉编码器设计。本文提出UltraViT,一种专为边缘设备性能优化的视觉编码器。通过考虑真实设备延迟,系统性地设计了一种分层架构,在宏观块级别融合并自适应异构空间混合器。此外,为预训练UltraViT,我们提出一种两阶段生成式预训练策略:先通过密集蒸馏培养丰富的空间特征,再通过容量混合的冻结大语言模型进行直接生成监督。相比传统对比学习与自监督方法,该策略在实现高级语义对齐方面更为有效,为后续生成式多模态对齐训练奠定基础。大量实验表明,结合延迟感知设计与定制训练策略,UltraViT在边缘设备上的编码效率达到新基准,显著优于现有以编码器为中心的基线,运行速度接近1.7倍提升。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。