arXiv:2507.03279cs.LGcs.AI2025-07NeurIPS被引 9

用置信预测集优化大模型交互式问答,减少提问次数。

Conformal Information Pursuit for Interactively Guiding Large Language Models

  • 用置信预测集大小替代条件熵,更稳健估计不确定性。
  • 在20个问题游戏和医疗问答中,提问次数减少15%以上。
  • 适合需要可解释性的交互式决策场景,如医疗辅助诊断。

指令微调的大语言模型(LLM)常用于交互式问答任务,即通过逐轮向用户查询信息来推断答案,而非单次对话。本文研究如何最小化期望提问次数。现有方法如信息追求(IP)依赖于信息增益,但因大模型概率输出过于自信或不充分,难以准确估计互信息或条件熵,导致查询效率下降。为此,提出基于合规范畴预测集的共形信息追求(C-IP),利用预测集平均大小与条件熵的关系,在无需分布假设的前提下稳健估计不确定性。实验表明,在20个问题基准上,C-IP比传统IP和基于不确定性的思维链方法减少约15%的提问次数;在MediQ医疗问答数据集上,其性能接近单轮直接预测,同时提供更高可解释性。

原文摘要 · Abstract (English)

A significant use case of instruction-finetuned Large Language Models (LLMs) is to solve question-answering tasks interactively. In this setting, an LLM agent is tasked with making a prediction by sequentially querying relevant information from the user, as opposed to a single-turn conversation. This paper explores sequential querying strategies that aim to minimize the expected number of queries. One such strategy is Information Pursuit (IP), a greedy algorithm that at each iteration selects the query that maximizes information gain or equivalently minimizes uncertainty. However, obtaining accurate estimates of mutual information or conditional entropy for LLMs is very difficult in practice due to over- or under-confident LLM proba- bilities, which leads to suboptimal query selection and predictive performance. To better estimate the uncertainty at each iteration, we propose Conformal Information Pursuit (C-IP), an alternative approach to sequential information gain based on conformal prediction sets. More specifically, C-IP leverages a relationship between prediction sets and conditional entropy at each iteration to estimate uncertainty based on the average size of conformal prediction sets. In contrast to conditional entropy, we find that conformal prediction sets are a distribution-free and robust method of measuring uncertainty. Experiments with 20 Questions show that C-IP obtains better predictive performance and shorter query-answer chains compared to previous approaches to IP and uncertainty-based chain-of-thought methods. Furthermore, extending to an interactive medical setting between a doctor and a patient on the MediQ dataset, C-IP achieves competitive performance with direct single-turn prediction while offering greater interpretability.

交互式问答不确定性估计医疗AI

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