2606.06840

Total: 1

#1 Characterize Then Distill: Mechanistic Reasoning in Large Output Spaces [PDF] [Copy] [Kimi] [REL]

Authors: Debjyoti Saha Roy, Byron C. Wallace, Javed A. Aslam

Modern reasoning models offer surprisingly strong zero-shot performance on challenging multi-label tasks that require selecting a small set of relevant options from hundreds of thousands to millions of candidate labels. We investigate how they achieve this mechanistically. We characterize reasoning as a two-phase process: A broad "shortlisting" of candidates followed by fine-grained reasoning over the resulting set. We provide evidence across a range of datasets that these steps can be isolated and are complementary. Using this characterization, we develop a mechanistic distillation strategy that consistently outperforms standard distillation.

Subjects: Computation and Language , Artificial Intelligence , Machine Learning

Publish: 2026-06-05 02:32:24 UTC