Reasoning Language Models: A Blueprint

  • 2025-01-22 17:49:37
  • Maciej Besta, Julia Barth, Eric Schreiber, Ales Kubicek, Afonso Catarino, Robert Gerstenberger, Piotr Nyczyk, Patrick Iff, Yueling Li, Sam Houliston, Tomasz Sternal, Marcin Copik, Grzegorz Kwaśniewski, Jürgen Müller, Łukasz Flis, Hannes Eberhard, Hubert Niewiadomski, Torsten Hoefler
  • 0

Abstract

Reasoning language models (RLMs), also known as Large Reasoning Models(LRMs), such as OpenAI's o1 and o3, DeepSeek-V3, and Alibaba's QwQ, haveredefined AI's problem-solving capabilities by extending LLMs with advancedreasoning mechanisms. Yet, their high costs, proprietary nature, and complexarchitectures - uniquely combining Reinforcement Learning (RL), searchheuristics, and LLMs - present accessibility and scalability challenges. Toaddress these, we propose a comprehensive blueprint that organizes RLMcomponents into a modular framework, based on a survey and analysis of all RLMworks. This blueprint incorporates diverse reasoning structures (chains, trees,graphs, and nested forms), reasoning strategies (e.g., Monte Carlo Tree Search,Beam Search), RL concepts (policy, value models and others), supervisionschemes (Outcome-Based and Process-Based Supervision), and other relatedconcepts (e.g., Test-Time Compute, Retrieval-Augmented Generation, agenttools). We provide detailed mathematical formulations and algorithmicspecifications to simplify RLM implementation. By showing how schemes likeLLaMA-Berry, QwQ, Journey Learning, and Graph of Thoughts fit as special cases,we demonstrate the blueprint's versatility and unifying potential. Toillustrate its utility, we introduce x1, a modular implementation for rapid RLMprototyping and experimentation. Using x1 and a literature review, we providekey insights, such as multi-phase training for policy and value models, and theimportance of familiar training distributions. Finally, we discuss scalable RLMcloud deployments and we outline how RLMs can integrate with a broader LLMecosystem. Our work demystifies RLM construction, democratizes advancedreasoning capabilities, and fosters innovation, aiming to mitigate the gapbetween "rich AI" and "poor AI" by lowering barriers to RLM development andexperimentation.

 

Quick Read (beta)

loading the full paper ...