Unlocking Recursive Thinking of LLMs: Alignment via Refinement

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Haoke, Liang, Xiaobo, Wang, Cunxiang, Li, Juntao, Zhang, Min
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915331178496000
author Zhang, Haoke
Liang, Xiaobo
Wang, Cunxiang
Li, Juntao
Zhang, Min
author_facet Zhang, Haoke
Liang, Xiaobo
Wang, Cunxiang
Li, Juntao
Zhang, Min
contents The OpenAI o1-series models have demonstrated that leveraging long-form Chain of Thought (CoT) can substantially enhance performance. However, the recursive thinking capabilities of Large Language Models (LLMs) remain limited, particularly in the absence of expert-curated data for distillation. In this paper, we propose \textbf{AvR}: \textbf{Alignment via Refinement}, a novel method aimed at unlocking the potential of LLMs for recursive reasoning through long-form CoT. AvR introduces a refinement process that integrates criticism and improvement actions, guided by differentiable learning techniques to optimize \textbf{refinement-aware rewards}. As a result, the synthesized multi-round data can be organized as a long refinement thought, further enabling test-time scaling. Experimental results show that AvR significantly outperforms conventional preference optimization methods. Notably, with only 3k synthetic samples, our method boosts the performance of the LLaMA-3-8B-Instruct model by over 20\% in win rate on AlpacaEval 2.0. Our code is available at Github (https://github.com/Banner-Z/AvR.git).
format Preprint
id arxiv_https___arxiv_org_abs_2506_06009
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Unlocking Recursive Thinking of LLMs: Alignment via Refinement
Zhang, Haoke
Liang, Xiaobo
Wang, Cunxiang
Li, Juntao
Zhang, Min
Computation and Language
Artificial Intelligence
The OpenAI o1-series models have demonstrated that leveraging long-form Chain of Thought (CoT) can substantially enhance performance. However, the recursive thinking capabilities of Large Language Models (LLMs) remain limited, particularly in the absence of expert-curated data for distillation. In this paper, we propose \textbf{AvR}: \textbf{Alignment via Refinement}, a novel method aimed at unlocking the potential of LLMs for recursive reasoning through long-form CoT. AvR introduces a refinement process that integrates criticism and improvement actions, guided by differentiable learning techniques to optimize \textbf{refinement-aware rewards}. As a result, the synthesized multi-round data can be organized as a long refinement thought, further enabling test-time scaling. Experimental results show that AvR significantly outperforms conventional preference optimization methods. Notably, with only 3k synthetic samples, our method boosts the performance of the LLaMA-3-8B-Instruct model by over 20\% in win rate on AlpacaEval 2.0. Our code is available at Github (https://github.com/Banner-Z/AvR.git).
title Unlocking Recursive Thinking of LLMs: Alignment via Refinement
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2506.06009