paper-with-me

홈 › Papers

ThinkTwice: Jointly Optimizing Large Language Models for Reasoning and Self-Refinement

2026-04-02 · Difan Jiao, Qianfeng Wen, Blair Yang, Zhenwei Tang, Ashton Anderson arxiv

We introduce ThinkTwice, a simple two-phase framework that jointly optimizes LLMs to solve reasoning problems and refine the answers, based on Group Relative Policy Optimization (GRPO). In each pair of training steps, ThinkTwice first optimizes the model on solving reasoning problems, then optimizes it on refining its own solutions to the same problems, using the same binary correctness reward in both phases without correctness signals or critique annotations. Across five mathematical reasoning benchmarks and two model families including Qwen3-4B and Olmo3-7B, ThinkTwice substantially improves both reasoning and refinement performance over competitive online policy optimization baselines. Specifically, on Qwen3-4B, ThinkTwice outperforms GRPO on AIME by 5 percentage points before refinement and by 11.5 points after one self-refinement step, measured by pass@4. Analysis of the training dynamics of ThinkTwice reveals an implicit rectify-then-fortify curriculum: refinement predominantly corrects errors early in training and naturally shifts toward preserving already-correct solutions as the model improves, yielding a more rectified reward signal. Our work establishes joint training of reasoning and self-refinement as a principled and effective methodology for RLVR.

📄 PDF Abstract BibTeX arXiv:2604.01591

Code (0)

등록된 구현이 없습니다.

Tasks

Mathematical Reasoning

Similar Papers 제목 키워드 기반

Do not be greedy, Think Twice: Sampling and Selection for Document-level Information Extraction

2026-01-26 · Mikel Zubillaga, Oscar Sainz, Oier Lopez de Lacalle, Eneko Agirre arxiv

Document-level Information Extraction (DocIE) aims to produce an output template with the entities, relations, and events of interest occurring in the given document. Standard practices include prompting decoder-only LLM…

Information Extraction

Reasoning-Oriented and Analogy-Based Methods for Locating and Editing in Zero-Shot Event-Relational Reasoning

2025-01-01 · Jingyao Tang, Lishuang Li, Liteng Mi, Haiming Wu 외

Zero-shot event-relational reasoning is an important task in natural language processing, and existing methods jointly learn a variety of event-relational prefixes and inference-form prefixes to achieve such tasks. Howev…

Language ModelingLanguage ModellingRelational Reasoning

LLM Compression with Jointly Optimizing Architectural and Quantization choices

2026-06-02 · Hoang-Loc La, Truong-Thanh Le, Amir Taherkordi, Phuong Hoai Ha arxiv

Deploying large language models (LLMs) is challenging due to their significant memory and computational requirements. While some methods address this by developing small or tiny language models from scratch, these approa…

Neural Architecture Search

Stepwise Think-Critique: A Unified Framework for Robust and Interpretable LLM Reasoning

2025-12-17 · Jiaqi Xu, Cuiling Lan, Xuejin Chen, Yan Lu arxiv

Human beings solve complex problems through critical thinking, where reasoning and evaluation are intertwined to converge toward correct solutions. However, most existing large language models (LLMs) treat the reasoning …

Reinforcement LearningMathematical Reasoning

GeoFaith: A Spatio-Temporal Dual View of Faithful Chain-of-Thought

2026-05-26 · Weijiang Lv, Wentong Zhao, Jiayu Wang, Yuhao Wu 외 arxiv

Chain-of-Thought (CoT) reasoning has advanced large language models (LLMs), but outcome-based supervision leads to pervasive post-hoc rationalization, producing plausible yet unfaithful reasoning chains. Most prior faith…

Reinforcement Learning