AutoThink is an inference-time reasoning method that sets a query-specific reasoning budget and guides a model toward better reasoning paths. section 15.2 section 15.5
It combines query-complexity classification with steering-vector guidance. A classifier decides whether a query needs high- or low-complexity reasoning, and control vectors steer the model during inference. section 15.2 section 15.5
The treatment breaks AutoThink into two phases:
- Preprocessing collects query-response pairs, finds pivotal tokens, derives vectors, and trains a query-complexity classifier. section 15.2
- Inference loads a pretrained classifier, selects a token budget, and applies control vectors while decoding. section 15.2 section 15.5
Evaluation setup
A local small language model is evaluated on GSM8K samples with AutoThink, using OptiLLM’s autothink_decode function and a modest token budget. section 15.2 section 15.5
