Dynamic Latent Routing
Not on the map. Why?
Abstract (arXiv)
We investigate the temporal concatenation of sub-policies in Markov Decision Processes (MDP) with time-varying reward functions. We introduce General Dijkstra Search (GDS), and prove that globally optimal goal-reaching policies can be recovered through temporal composition of intermediate optimal sub-policies. Motivated by the "search, select, update" principle underlying GDS, we propose Dynamic Latent Routing (DLR), a language-model post-training method that jointly learns discrete latent codes, routing policies, and model parameters through dynamic search in a single training stage. In low-data fine-tuning settings, DLR matches or outperforms supervised fine-tuning across four datasets and six models, achieving a mean gain of +6.6 percentage points, while prior discrete-latent baselines consistently underperform SFT. Mechanistic analyses and targeted code ablations show that DLR learns structured routing behaviors with distinct causal roles.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 12 of its 55 references that are in the map. First posted on arXiv 14 May 2026.
Builds on
12 of 55 references.
References · 12
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
- 2023Think before you speak: Training Language Models With Pause Tokenscited by 264
- 2025Token Assorted: Mixing Latent and Text Tokens for Improved Language Model Reasoningcited by 91
- 2021Temporal Abstraction in Reinforcement Learning with the Successor Representationcited by 54
- 2025RLP: Reinforcement as a Pretraining Objectivecited by 25
- 2024Understanding Addition and Subtraction in Transformerscited by 8
- 2026Thinking Without Words: Efficient Latent Reasoning with Abstract Chain-of-Thoughtcited by 8
- 2025Why Representation Engineering Works: A Theoretical and Empirical Study in Vision-Language Modelscited by 6
- 2020Zero-Shot Compositional Policy Learning via Language Groundingcited by 3
- 2025SteeringSafety: Benchmarking Representation Steering in LLMs Across Safety Perspectivescited by 2
Built on by
Not cited yet.