LLM Router: Prefill is All You Need

arXiv cs.CL / 3/24/2026

💬 OpinionSignals & Early TrendsIdeas & Deep AnalysisModels & Research

Key Points

  • The paper argues that an “oracle” router could outperform single LLMs by selecting among models according to their complementary strengths across different task subsets.
  • It proposes a more robust routing signal by using internal prefill activations and Encoder-Target Decoupling, separating the component generating predictive signals from the component whose performance is being estimated.
  • The method uses mathematical probes—Fisher Separability and Effective Dimensionality—to identify optimal layer-wise signals that form the basis for the SharedTrunkNet routing architecture.
  • SharedTrunkNet is reported to recover up to 45.58% of the accuracy gap between the best standalone model and the oracle router while reducing cost, achieving 74.31% cost savings relative to the highest-cost model.
  • Overall, the work shifts router design away from brittle external semantic cues toward internal activation-based signals intended to support optimized heterogeneous model pairing.

Abstract

LLMs often share comparable benchmark accuracies, but their complementary performance across task subsets suggests that an Oracle router--a theoretical selector with perfect foresight--can significantly surpass standalone model accuracy by navigating model-specific strengths. While current routers rely on fragile semantic signals, we propose using internal prefill activations via Encoder-Target Decoupling--a functional separation between the model providing the predictive signal (the Encoder) and the model whose performance is being estimated (the Target). This allows optimized heterogeneous pairing between unique encoders and target models. We utilize Fisher Separability (J) and Effective Dimensionality (d_eff) as mathematical probes to isolate optimal layer-wise signals, providing the predictive foundation for our SharedTrunkNet architecture. SharedTrunkNet captures up to 45.58% of the accuracy gap between the strongest standalone model and the Oracle while achieving 74.31% cost savings relative to the highest-cost model.