让大模型生成更多样解法,提升复杂任务表现
GuidedSampling: Steering LLMs Towards Diverse Candidate Solutions at Inference-Time
- 分探索与生成两阶段,引导模型发现多种解题思路
- 比传统重复采样提升21.6%的pass@50指标
- 适合需要多角度解法的复杂推理任务
重复采样(RS)是一种简单的推理时算法,已被证明能提升大模型在复杂任务上的表现。尽管能通过延长推理时间来增强性能,但其常产生重复性解法,依赖相似思路导致候选解缺乏多样性。为此,我们提出新算法GuidedSampling,将推理过程拆分为探索与生成两个阶段:探索阶段识别多种可利用的解题概念,生成阶段则基于特定概念输出最终解。我们首先推导了GuidedSampling的理论边界,并实证表明,相比RS,其在多个基准测试上平均提升基模型的pass@50达21.6%。此外,使用GuidedSampling轨迹训练的模型,在pass@5上平均提升9.7%,且每实例平均概念数从1.67增至3.03,显著提高解法多样性。
原文摘要 · Abstract (English)
Repeated Sampling (RS) is a simple inference-time algorithm that has been shown to improve model performance on complex tasks. Although it is an effective way of scaling inference time, it often struggles to generate diverse solution candidates, frequently relying on the same underlying approach to solve the problem and thus producing redundant samples. To address this limitation, we propose a new inference algorithm, GuidedSampling, which decouples the exploration and generation phases during inference, increasing diversity of generated candidate solutions. The exploration phase identifies multiple concepts that can be utilized to solve the problem, while the generation phase applies a specific concept to provide final solution candidates. We first define the theoretical bounds of GuidedSampling and then empirically demonstrate that it improves the performance of base model at pass@50 by on an average ~21.6% across various benchmarks compared to RS. Furthermore, models trained on trajectories of GuidedSampling exhibit substantial performance improvements at pass@5 by on an average ~9.7%, compared to models trained on traditional RS. Additionally, models trained with GuidedSampling increases the average number of concepts per instance (1.67 -> 3.03), yielding a diverse set of candidates than traditional RS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。