Count Counts: Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2025
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866918166459842560 |
|---|---|
| author | Zhang, Xuan Li, Ruixiao Zhou, Zhijian Li, Long Qin, Yulei Li, Ke Sun, Xing Tan, Xiaoyu Qu, Chao Qi, Yuan |
| author_facet | Zhang, Xuan Li, Ruixiao Zhou, Zhijian Li, Long Qin, Yulei Li, Ke Sun, Xing Tan, Xiaoyu Qu, Chao Qi, Yuan |
| contents | Reinforcement Learning (RL) has become a compelling way to strengthen the multi step reasoning ability of Large Language Models (LLMs). However, prevalent RL paradigms still lean on sparse outcome-based rewards and limited exploration, which often drives LLMs toward repetitive and suboptimal reasoning patterns. In this paper, we study the central question of how to design exploration for LLM reasoning and introduce MERCI (Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards), a novel RL algorithm that augments policy optimization with a principled intrinsic reward. Building on the idea of count-based exploration, MERCI leverages a lightweight Coin Flipping Network (CFN) to estimate the pseudo count and further epistemic uncertainty over reasoning trajectories, and converts them into an intrinsic reward that values novelty while preserving the learning signal from task rewards. We integrate MERCI into some advanced RL frameworks like Group Relative Policy Optimization (GRPO). Experiments on complex reasoning benchmarks demonstrate that MERCI encourages richer and more varied chains of thought, significantly improves performance over strong baselines, and helps the policy escape local routines to discover better solutions. It indicates that our targeted intrinsic motivation can make exploration reliable for language model reasoning. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2510_16614 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Count Counts: Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards Zhang, Xuan Li, Ruixiao Zhou, Zhijian Li, Long Qin, Yulei Li, Ke Sun, Xing Tan, Xiaoyu Qu, Chao Qi, Yuan Artificial Intelligence Reinforcement Learning (RL) has become a compelling way to strengthen the multi step reasoning ability of Large Language Models (LLMs). However, prevalent RL paradigms still lean on sparse outcome-based rewards and limited exploration, which often drives LLMs toward repetitive and suboptimal reasoning patterns. In this paper, we study the central question of how to design exploration for LLM reasoning and introduce MERCI (Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards), a novel RL algorithm that augments policy optimization with a principled intrinsic reward. Building on the idea of count-based exploration, MERCI leverages a lightweight Coin Flipping Network (CFN) to estimate the pseudo count and further epistemic uncertainty over reasoning trajectories, and converts them into an intrinsic reward that values novelty while preserving the learning signal from task rewards. We integrate MERCI into some advanced RL frameworks like Group Relative Policy Optimization (GRPO). Experiments on complex reasoning benchmarks demonstrate that MERCI encourages richer and more varied chains of thought, significantly improves performance over strong baselines, and helps the policy escape local routines to discover better solutions. It indicates that our targeted intrinsic motivation can make exploration reliable for language model reasoning. |
| title | Count Counts: Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards |
| topic | Artificial Intelligence |
| url | https://arxiv.org/abs/2510.16614 |