Papers for

software engineers optimizing ml inference

Papers whose findings have a practical use for this group, as judged from the abstract. Open a paper to read what it means in practice.

Multimodal large language models save compute with visual backtracking

ReTrace: Visual Backtracking for Efficient Multimodal Large Language Models

Abstract: Multimodal Large Language Models (MLLMs) encode images into dense visual token sequences, creating substantial attention and KV-cache overhead. Existing visual token compression methods reduce this cost by pruning low-saliency tokens in shallow layers, but this practice couples temporary computational deactivation with irreversible information deletion. We show that this coupling breaks an intrinsic inference pattern in MLLMs. Through layer-wise attention analysis, we identify \textbf{Visual Backtracking}: visual pivots remain spatially stable, yet in middle layers they actively retrieve context from tokens that appeared unimportant earlier. Thus, early low-saliency tokens can be inactive for computation while still being useful for later reasoning. Motivated by this finding, we propose \textbf{ReTrace}, a backtracking-aware compression framework that introduces a dormant token state. ReTrace caches non-pivot tokens outside expensive layer-wise computation, restores them during the backtracking window, and prunes only after their information has been absorbed. Experiments on Qwen2.5-VL and InternVL3 demonstrate nearly 90\% reductions in FLOPs and KV cache, a 2.83$\times$ prefilling speedup, and strong performance retention across complex vision-language benchmarks.

Sun 18 OctComputer Vision and Pattern RecognitionArtificial Intelligence
The gist
Images in large language models are broken down into many pieces that computers use when answering questions, which takes a lot of computing power. The authors discovered that often pieces thought unimportant early on actually become helpful later for understanding the image better. They created a way called ReTrace that temporarily sets aside less important pieces instead of deleting them, then brings them back when needed. This method reduces computing work significantly while keeping the model’s accuracy high.
Open → 2610.25005v1