Papers for

ai system engineers

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.

Vision token pruning improves by picking best method per image

Beyond One-Size-Fits-All: Sample-Adaptive Strategy Routing for Vision Token Pruning in MLLMs

Abstract: Multimodal large language models (MLLMs) process hundreds or thousands of visual tokens per image, incurring prohibitive inference costs. While existing vision token pruning methods mitigate this overhead, they implicitly assume that a single fixed pruning strategy can be applied uniformly across all inputs. Our analysis further reveals that ranking pruning methods by average benchmark accuracy conceals substantial sample-wise complementarity: although the average-best strategy excels overall, alternative strategies prove superior on a significant fraction of individual samples. To harness this diversity, we propose VIP-Router, a lightweight VIsion Pruning Router that adaptively selects the pruning strategy predicted to be best suited to each input at a specified pruning level. Conditioned on low-cost visual and textual features, VIP-Router identifies the most suitable candidate strategy while retaining full-token inference as an option when pruning is predicted to be unfavorable. Evaluated on a curated suite of pruning-sensitive visual perception benchmarks, VTC-Bench Group A, VIP-Router consistently outperforms the best fixed strategy baseline across all reduction ratios, achieving a 26.9% relative improvement in average accuracy, and a 22.0% relative increase in average utility after accounting for realized token cost. Crucially, VIP-Router operates in a plug-and-play manner without modifying underlying pruning algorithms or model weights, introducing trainable parameters equivalent to merely 0.017\% of the backbone. Furthermore, VIP-Router proves effective across various MLLM backbones and yields consistent gains on unseen benchmarks, highlighting the potential of sample adaptive routing for visual token pruning.

Wed 9 SeptComputer Vision and Pattern RecognitionArtificial Intelligence
The gist
Processing images with large AI models is expensive because they use many visual pieces called tokens. The authors found that no single pruning method works best for every image—some methods work better on certain pictures. They created Vip-Router, a small system that picks the best pruning method for each image using simple clues, saving effort without losing accuracy. This approach works well across different models and images without changing the main AI itself.
Open 2609.10346v1

Unitboost improves compound AI systems with transparent output merging

UnitBoost: Managing Compound LLM Systems with a Merge Operator, Not a Model

Abstract: Compound LLM systems often solve a coordination problem by adding a higher-level LLM. The resulting meta-agent reads workers' outputs, writes the final answer, allocates later calls, and decides when to stop. It is expressive, but it also concentrates three control decisions in an opaque, order-sensitive model call. We ask whether the manager needs to be generative at all. UnitBoost replaces that model with a defined meta-level operator: a task-given unit map turns worker outputs into slot-value proposals, a constrained argmax assembles the output, and the slots left unfilled or unsupported become an explicit residual for the next round. The operator is order-free, records unit provenance, and gives a simple guarantee: without coupling constraints, unit-wise maximization under the same admission score dominates selection of any complete candidate. On three held-out benchmarks, it exceeds the best single candidate chosen with gold labels by 0.060-0.195 absolute task-score points and input-matched generative managers by 0.048-0.076. Replacing only the management step improves six compound-system configurations by 0.013-0.182. Residual-directed rounds raise FanOutQA cell F1 from 0.4778 to 0.5524; matched controls show that the true residual outperforms random targets and ordinary rereading, while a label-free supply signal flags exhaustion after one unproductive round. The same analysis measures three conditions in which no such gain is available (one indivisible unit, unavailable unit identity, and an endpoint that charges for every emitted unit) and quantifies cross-unit coupling as a repair cost. The manager gives up semantic freedom and gains order invariance, unit provenance, and testable failure conditions.

Wed 9 SeptArtificial IntelligenceComputation and LanguageMultiagent Systems
The gist
Large language models (LLMs) that work together usually need a special manager model to combine their answers. The authors show that this manager doesn’t need to generate text to do its job. Instead, their approach called UnitBoost uses a clear method to pick the best parts from each model's answer, making the system easier to understand, less sensitive to order, and more reliable. Tests on several tasks showed UnitBoost gave better final answers than other ways of managing multiple LLMs.
Open 2609.09815v1

Procedural memory mismatch does not always disrupt web task behavior

Procedural Memory Under Change: Reuse and Interference in Controlled Web Tasks

Abstract: Procedural memory lets language agents reuse successful routines, but reuse presumes that a stored routine remains applicable. We study what happens when that presumption is deliberately violated. The study combines a retrospective, human-assisted interface-adaptation case from BrowserGym TimeWarp with controlled frozen-memory comparisons on synthetic shopping decisions. During the documented WebShop V1-V6 development path, interface-specific code was adapted while the separately stored high-level procedure was not reported to change; this phase does not constitute an autonomous memory-agent evaluation. In the controlled phase, an early pilot produced one task on which two memory conditions selected a more expensive item while the no-memory condition selected the reference minimum. Follow-up probes did not establish a recurring row-order or identity-binding pattern. We then tested four forms of mismatch: changed quantities, a different evidence representation, a conflict between local and global optimization, and distributed promotion evidence, across 32 formal cells. Each cell used one temperature-0 generation with the same local qwen3:8b configuration and no adaptive retry. Across these pairs, none of the predefined diagnostic interference signatures appeared on the tasks for which they were defined when current-task evidence was explicit and sufficient. The result identifies a tested region of non-interference: a procedural memory can be mismatched without becoming behaviorally disruptive. It does not establish general safety or a mechanism. The remaining question is which additional conditions turn applicability mismatch into observable, memory-caused error.

Wed 9 SeptArtificial Intelligence
The gist
Procedural memory helps language agents reuse routines they learned before, but this only works if the routines still apply. The authors tested what happens when the stored routines no longer exactly match the current task in web-based shopping decisions. They found that even when there are mismatches, these do not necessarily cause errors or disruptions in behavior. This shows that procedural memory can sometimes handle changes without breaking functionality, though the authors did not determine what conditions cause errors.
Open 2609.09774v1

Combined method improves long input response speed and accuracy

Fine-Tuning a KV Cache Concatenation-Aware Model or Recomputing KV Caches? Why Not Both?

Abstract: In Retrieval-Augmented Generation (RAG) systems, a large number of retrieved chunks are concatenated to form the input context so that users can receive high-quality responses based on external knowledge. As a result, the input context length increases substantially, leading to a larger prefill workload and, in turn, a longer time to first token (TTFT). While previous works that reuse precomputed key-value (KV) caches effectively reduce TTFT for long-context inputs, it remains unclear whether response quality is preserved when the input context becomes very long. In this paper, we propose a combined approach that (i) fine-tunes the model while taking KV cache concatenation into account and (ii) selectively recomputes a subset of the KV caches. By applying both techniques, we demonstrate improved accuracy for long-context inputs. Experiments on the RULER benchmark show that, for a 124k-token input, our method improves the RULER score by 9.7 point over the baseline that recomputes KV caches only. Moreover, TTFT is reduced by 80% compared with full attention.

Wed 9 SeptMachine LearningArtificial IntelligenceComputation and Language
The gist
When language models answer questions using lots of information, processing everything takes a long time before the first answer word appears. The authors looked into ways to speed this up without losing answer quality by teaching the model to handle joined information chunks better and only updating some parts as needed. Their approach made responses more accurate and much faster when dealing with very long inputs. This could help systems that rely on pulling in large external knowledge collections to give better and quicker answers.
Open 2609.09768v1

Deposon offers auditable energy-conserving checks on AI reasoning paths

Deposon: An Auditable, Conservation-Guaranteed, Game-Theoretically Tested Scattering Layer over LLM Reasoning Paths

Abstract: Multi-step LLM reasoning lacks a machine-recheckable ledger: discarded reasoning paths leave no auditable record. We propose the Deposon scattering layer, which binds each node of an LLM-generated concept-decomposition graph to a two-parameter Deposon state; paths undergo three-channel scattering -- transmission, reflection, irreversible dissipation -- obeying T+R+A=1 for arbitrary parameters, with a maximum per-path energy-audit deviation of 2.2E-16 (machine epsilon). We report all three evidence tiers honestly. On synthetic trap benchmarks the path-filtering gain is closed (pre-registered): unified reaches 100% versus a decoy-capture baseline at 7%/10%. On real benchmarks the layer is indistinguishable from a trivial six-keyword rule filter (GSM8K 0.87 >= 0.85, McNemar p=0.5; StrategyQA 0.899 = 0.899); no difference is detected here, so we sharpen the claim to "the differential value lies solely in machine verifiability." Fusion yields a second negative result: convex combinations with a semantic prior never improve (physics 0.484 -> 0.452), and the apparent lambda=2 gain is an anti-field artifact; any fusion gain must be nonlinear. Modeling the reverse dynamics as a potential game on the graph, we evidence an auditable scalar's monotonicity and near-gradientness and quantify the empirical coordination ratio (ECR). The three formalized dynamical-equivalence propositions (P1a/P1b/T-P1c) are falsified under the pre-registered kill protocol, and the potential-game claim is downgraded to approximate (cyclic-graph median residual 0.669): only consistency-level evidence survives at the dynamical level. Code: github.com/zeroandcat/Deposon.

Tue 8 SeptArtificial IntelligenceMachine Learning
The gist
AI systems that think through multi-step problems often lose track of discarded steps, making it hard to check their reasoning later. The authors introduce Deposon, a method that tracks and verifies parts of an AI's thought process with very precise mathematical guarantees. They tested it on artificial puzzles and real tasks, showing it records verifiable reasoning without improving task accuracy. They also explored theoretical properties of their approach but found some expected game-theory connections only approximate.
Open 2609.09001v1

History aware dynamic routing improves large language model efficiency

Do Dynamic Routers Need Memory? HeRo: History-Aware Routing for Efficient LLM Inference

Abstract: Dynamic layer routing reduces the inference cost of Large Language Models (LLMs) by learning to skip layers for individual tokens. Existing methods, however, treat each routing decision as a local operation conditioned solely on the current hidden state which is a formulation that overlooks the sequential, path-dependent nature of routing across depth: earlier decisions shape the representations seen by downstream routers, and the layer-usage objective couples all decisions jointly. We propose History-Aware Routing (HeRo), a dynamic routing framework that resolves this mismatch by introducing a router memory mechanism to maintain an explicit routing state across model depth. The memory is constructed via linear attention, incrementally aggregating preceding routing scores and their induced residual updates into a compact history representation. At each routed layer, the router conditions jointly on this accumulated state and the current hidden representation to select the executed branch. Instantiated for token-wise FFN routing, HeRo trains only lightweight routers and adapters on a frozen backbone, requiring no modification to pretrained parameters. Across Llama 3.1-8B, Llama 2-7B, and Llama 2-13B, HeRo consistently achieves the highest aggregate performance retention among ten baselines. On Llama 3.1-8B, it bypasses 26.87% of model parameters while achieving 100.24% of dense model performance across seven benchmarks, and retains 97.01% while bypassing 38.82% of model parameters under a tighter computation budget. Ablation studies confirm that removing routing history consistently degrades performance, most notably on multistep reasoning and code generation, validating that explicit routing memory enables more accurate and adaptive dynamic routing than solely conditioning on hidden state.

Tue 8 SeptArtificial IntelligenceComputation and Language
The gist
Large language models are very big and slow to use because they process every word through many layers. Some methods try to speed this up by skipping layers for each word, but they make decisions based only on the current word’s data. The authors found that remembering past routing decisions helps the model make better choices about which layers to use. They created a system called HeRo that keeps a memory of earlier decisions to improve layer skipping and make the model faster without losing accuracy. Their tests show HeRo works well on multiple versions of Llama models and keeps performance high while using fewer parameters.
Open 2609.08189v1

Large language models show different decision paths despite similar choices

The Internal Anatomy of Strategic Choice in Large Language Models

Abstract: Large language models act as strategic agents and models of human choice, yet choosing like a strategic agent does not mean computing like one. We recorded activations from four open-weight models --- dense and mixture-of-experts, including a matched base--instruct pair --- in one-shot play of 144 strict ordinal $2\times2$ games. We followed a prespecified incentive from prompt, through activations, to choice. Dense models mirrored the unadjusted human decline with game complexity. Incentive and choice were detectable in every model, but models differed in whether incentive reached the choice, aligned with it and, where tested, whether strengthening it shifted preference. The base and instruction-tuned Qwen2.5 models chose almost identically at baseline yet differed in whether incentive reached choice. Fixed decision cues were distinguishable internally but changed choices selectively. Similar behaviour can rest on different computation; post-training can reshape the path from represented incentive to decision while leaving behaviour and decodable information largely intact.

Mon 7 SeptArtificial IntelligenceComputer Science and Game Theory
The gist
Sometimes large language models make decisions like humans do, but how they process information inside their 'brains' can be quite different. The researchers looked at how these models play simple games and found that while the end choices looked alike, the steps inside the models varied. Changing how the models were trained affected how information about incentives moved inside but didn’t always change the final decisions. This means similar behaviors in AI can come from very different internal thinking.
Open 2609.07478v1

Cedar speeds up long context attention with smarter token routing

CEDAR: Error-Bounded Residual Routing for Efficient Long-Context Attention

Abstract: Post-hoc sparse attention accelerates long-context prefill by routing each query to a small set of token-level interactions. Hard selection, however, assigns zero probability to every omitted chunk: a routing miss cannot be recovered, and a fixed expansion budget spends the same work on easy and ambiguous queries. We introduce Coarse-to-fine Error-aware Dynamic Attention Routing (CEDAR), a coarse-to-fine method that keeps the language model frozen while preserving global coverage. Each semantic chunk contributes a cheap key--value summary to a residual attention path; chunks with high estimated approximation error are then expanded to exact token attention. Exact and summarized contributions are combined in a single softmax normalization, so refinement replaces, rather than duplicates, coarse evidence. We derive an output-error bound governed by within-chunk key/value dispersion and use it to allocate a variable refinement budget. A controlled clustered-attention study shows that residual summaries reduce reconstruction error by more than 98% relative to hard dropping at equal exact-chunk budgets. Experiments on long-context benchmarks demonstrate that CEDAR recovers most of the quality lost by hard sparse routing while maintaining approximately $3\times$ kernel speedup at 128K context.

Mon 7 SeptComputation and Language
The gist
Handling very long pieces of text with AI language models is slow and memory-heavy because every word needs to pay attention to many others. The authors created CEDAR, a method that quickly summarizes groups of words and only looks closely at smaller parts that matter most for the current task. This approach keeps the model's quality high while running about three times faster for very long texts. It also smartly decides where to spend extra effort based on how much detail is needed, rather than treating all parts equally.
Open 2609.07237v1