跳到正文
arXiv cs.AI· Anita Kriz, Antonin Berthon, Nicol\'as Astorga, Mihaela van der Schaar·· 2 天前AI 评分24

将序列贝叶斯实验设计摊销进推理 LLM

Amortizing Sequential Experimental Design in Reasoning LLMs

AI 导读

研究将序列贝叶斯实验设计(BED)摊销进 LLM,通过信息增益奖励显式训练引导动作选择的推理过程,实验生成模型独立于 LLM 自身信念。在 Twenty Questions 上,所得策略提升信息获取与任务表现,以显著更低计算成本超越推理时和训练时 BED 基线,并泛化到未见类别。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:尽管大型语言模型(LLM)拥有广泛的预训练知识和强大的推理能力,但在实验设计任务中仍难以自适应地收集信息。我们研究如何通过使用信息增益奖励显式训练指导动作选择的推理,将序列贝叶斯实验设计(BED)摊销到 LLM 中。我们的方法独立于 LLM 自身的信念来指定实验生成模型,使我们能够在所选模型下优化期望信息增益。我们研究了摊销的设计选择,包括替代生成模型构建、奖励估计、假设空间规范、多步目标,以及相对于训练时和推理时 BED 基线的计算成本。在 Twenty Questions 上,所得策略改善了信息获取和任务表现,以显著更低的计算成本优于推理时和训练时 BED 基线,并泛化到未见类别。我们进一步展示了从 Twenty Questions 训练到 Medi-Q(一个真实世界临床对话基准)的正向迁移。这些结果表明,信息收集可以作为 LLM 中一种可泛化的推理技能来学习。
学科: 机器学习(cs.LG);人工智能(cs.AI)
引用为: arXiv:2610.20832 [cs.LG]
  (或 arXiv:2610.20832v1 [cs.LG] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.20832

arXiv 颁发的 DOI 通过 DataCite(待注册)

提交历史

来自:Antonin Berthon [查看电子邮件]
[v1] 2026年10月14日星期三 19:15:58 UTC(1,676 KB)

来源:arXiv cs.AI · arxiv.org