Look Less, Think Faster: Joint Token-Compute Adaptation for Multimodal LLMs

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

SmartVL is a unified adaptive inference framework designed to address the high inference costs of Multimodal Large Language Models (MLLMs), which stem from numerous input visual tokens and intensive LLM computation. Unlike previous approaches that independently optimize visual token pruning or LLM layer/head skipping, SmartVL jointly controls the number of vision tokens and the LLM's compute capability. It integrates a vision-side token controller for selecting informative visual tokens and an LLM-side compute controller for adaptively adjusting LLM computation. These controllers are trained to coordinate using a shared budget encoding and a differentiable latency estimator, ensuring the overall inference cost meets a target budget. This joint scheduling enables SmartVL to learn cross-stage allocation strategies that adapt to both input complexity and runtime compute constraints. Benchmarking across multiple MLLM tasks demonstrates that SmartVL consistently surpasses prior adaptive methods, achieving superior accuracy-efficiency Pareto frontiers.

Key takeaway

For Machine Learning Engineers deploying Multimodal LLMs, you should consider unified adaptive inference frameworks like SmartVL to manage high inference costs. By jointly controlling visual token numbers and LLM compute, you can achieve better accuracy-efficiency trade-offs than independent optimization. This approach allows your models to dynamically adapt to varying input complexities and compute budgets, ensuring practical deployment without sacrificing performance.

Key insights

SmartVL jointly optimizes visual token count and LLM compute for MLLMs, adapting to input and budget.

Principles

Method

SmartVL employs a vision-side token controller and an LLM-side compute controller, coordinated via a shared budget encoding and differentiable latency estimator, for end-to-end training to meet target compute budgets.

In practice

Topics

Best for: Research Scientist, AI Engineer, Computer Vision Engineer, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.