arXiv:2410.12375cs.AIcond-mat.dis-nn2024-10被引 37

小模型通过自我迭代推理,实现深度思考与持续优化。

PRefLexOR: Preference-based Recursive Language Modeling for Exploratory Optimization of Reasoning and Agentic Thinking

  • 用偏好优化+递归思维框架,让模型在推理中不断修正中间步骤。
  • 30亿参数小模型也能实现跨领域生物材料问题的深度推理。
  • 适合想提升模型反思能力的研究者,尤其关注推理生成场景。

PRefLexOR 结合偏好优化与强化学习思想,使模型通过迭代推理自我改进。该方法采用多阶段训练:第一阶段通过优化优选与非优选响应的对数几率,使模型推理对齐正确决策路径;同时构建动态知识图谱,从随机文本片段生成问题并检索上下文信息。第二阶段利用拒绝采样持续生成现场训练数据,掩码推理步骤以精细调整推理质量。在思维令牌框架内引入递归优化,形成迭代反馈回路,显著提升推理的一致性、连贯性与适应性。尽管仅使用30亿参数的小型语言模型,仍可实现深度反思式推理。方法实现简单,可嵌入任意预训练大模型。重点应用于生物材料科学,涵盖同域与跨域案例研究。通过思考与反思双重策略,构建多智能体递归自提升推理系统,在推理时通过重复采样逐步优化输出。

原文摘要 · Abstract (English)

PRefLexOR (Preference-based Recursive Language Modeling for Exploratory Optimization of Reasoning) combines preference optimization with concepts from Reinforcement Learning to enable models to self-teach through iterative reasoning improvements. We propose a recursive learning approach that engages the model in multi-step reasoning, revisiting, and refining intermediate steps before producing a final output in training and inference phases. Through multiple training stages, the model first learns to align its reasoning with accurate decision paths by optimizing the log odds between preferred and non-preferred responses. During this process, PRefLexOR builds a dynamic knowledge graph by generating questions from random text chunks and retrieval-augmentation to contextualize relevant details from the entire training corpus. In the second stage, preference optimization enhances model performance by using rejection sampling to fine-tune reasoning quality by continually producing in-situ training data while masking the reasoning steps. Recursive optimization within a thinking token framework introduces iterative feedback loops, where the model refines reasoning, achieving deeper coherence, consistency, and adaptability. Implemented in small language models with only 3 billion parameters, we should that even tiny models can iteratively teach themselves to reason with greater depth and reflectivity. Our implementation is straightforward and can be incorporated into any existing pretrained LLM. We focus our examples on applications in biological materials science and demonstrate the method in a variety of case studies that range from in-domain to cross-domain applications. Using reasoning strategies that include thinking and reflection modalities we build a multi-agent recursive self-improving inference approach to successively improve responses via repeated sampling in inference time.

推理优化小模型自我改进递归推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。