Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Temperature as a Meta-Policy: Adaptive Temperature in LLM Reinforcement Learning

About

Temperature is a crucial hyperparameter in large language models (LLMs), controlling the trade-off between exploration and exploitation during text generation. High temperatures encourage diverse but noisy outputs, while low temperatures produce focused outputs but may cause premature convergence. Yet static or heuristic temperature schedules fail to adapt to the dynamic demands of reinforcement learning (RL) throughout training, often limiting policy improvement. We propose Temperature Adaptive Meta Policy Optimization (TAMPO), a new framework that recasts temperature control as a learnable meta-policy. TAMPO operates through a hierarchical two-loop process. In the inner loop, the LLM policy is updated (e.g., using GRPO) with trajectories sampled at the temperature selected by the meta-policy. In the outer loop, meta-policy updates the distribution over candidate temperatures by rewarding those that maximize the likelihood of high-advantage trajectories. This trajectory-guided, reward-driven mechanism enables online adaptation without additional rollouts, directly aligning exploration with policy improvement. On five mathematical reasoning benchmarks, TAMPO outperforms baselines using fixed or heuristic temperatures, establishing temperature as an effective learnable meta-policy for adaptive exploration in LLM reinforcement learning. Accepted at ICLR 2026.

Haoran Dang, Cuiling Lan, Hai Wan, Xibin Zhao, Yan Lu• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval (test)
Pass@176.22
444
Mathematical ReasoningMinerva
Pass@127.9
138
Scientific ReasoningGPQA Diamond (test)
Accuracy36.05
32
Mathematical ReasoningOlympiadBench
Avg Success Rate @842.43
23
Mathematical ReasoningAIME 24
Avg@815
14
Mathematical ReasoningMinerva Math
Avg@822.79
14
Mathematical ReasoningOmni-MATH
Average Score @823.92
14
Mathematical ReasoningAverage Reasoning Benchmarks
Avg@838.83
14
Mathematical ReasoningAMC 23
Avg@850.62
14
Mathematical ReasoningMATH 500
Avg@8 Score78.23
14
Showing 10 of 16 rows

Other info

Follow for update