PartitionLens: Identifying Trace-to-Route Alignment with Counterfactual Controls
Abstract
Routed reasoning distillation faces an attribution challenge: an accuracy gain can reflect conditional capacity, a changed supervision pool, or trace-to-route grouping. PartitionLens resolves the comparison design by fixing the routed student, teacher-token budget, trainable-parameter budget, router depth, routing granularity, and paired evaluation pool, then using Latent-Routing MoA as the routed-capacity reference. On the reported five-domain primary pool, sequence-level PartitionLens-Open reaches A1, exceeding Latent-Routing MoA by pp (paired BCa CI ). The advantage is positive in every reported domain, and its W/L/T accounting gives a tie-split win rate over paired outcomes. The result provides a precise matched-control reference for evaluating trace-grouped routed distillation.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.