Search papers, labs, and topics across Lattice.
DiSCTT, a difficulty-aware test-time adaptation framework, is introduced to improve LLM reasoning by dynamically allocating optimization strategies based on epistemic uncertainty. High-consensus inputs are fine-tuned with pseudo-labels, while low-consensus inputs are optimized via reinforcement learning with a consensus-regularized objective. Experiments on mathematical and general reasoning benchmarks show DiSCTT outperforms existing test-time adaptation methods with higher accuracy, reduced variance, and lower computational cost.
LLMs can adapt to reasoning tasks far more efficiently by focusing supervised fine-tuning on easy examples and using reinforcement learning to explore diverse solutions for hard ones.
Test-time adaptation offers a promising avenue for improving reasoning performance in large language models without additional supervision, but existing approaches often apply a uniform optimization objective across all inputs, leading to inefficient or unstable adaptation on heterogeneous reasoning problems. We propose DiSCTT, a difficulty-aware, consensus-guided self-curriculum framework that dynamically allocates test-time optimization strategies based on instance-level epistemic uncertainty estimated from agreement among sampled reasoning trajectories. Inputs with high consensus are consolidated via supervised fine-tuning using majority-agreed solutions as pseudo-labels, while low-consensus inputs are optimized via reinforcement learning with a consensus-regularized objective that encourages diversity under relevance constraints. Across a broad suite of mathematical and general reasoning benchmarks, DiSCTT consistently outperforms strong test-time adaptation baselines, achieving higher accuracy with reduced variance and substantially lower computation and wall-clock training times. These results demonstrate that explicitly accounting for instance difficulty and uncertainty enables more stable, efficient, and effective test-time adaptation for reasoning models.
