动态检测小模型幻觉,适时调用大模型提升问答准确率
Invoke Interfaces Only When Needed: Adaptive Invocation for Large Language Models in Question Answering
- 用注意力累积得分实时评估小模型幻觉风险
- 在复杂问题上准确率提升,且无需额外训练
- 适合资源受限场景下高效部署问答系统
大模型与小模型协同的问答范式在性能与成本间取得平衡,但核心挑战在于精准识别小模型产生幻觉时的调用时机。以往优化多采用独立于推理过程的后处理方法,计算开销大且效果有限。本文提出一种实用的调用评估指标 AttenHScore,通过计算小模型生成过程中幻觉的积累与传播,持续放大潜在推理错误。结合动态阈值调整,实现大模型更精准的实时调用。同时,针对小模型推理能力有限的问题,引入基于不确定性的知识重组织机制,帮助其更好捕获不同文本片段的关键信息。大量实验表明,AttenHScore 在多个 QA 数据集上显著优于多数基线方法,尤其在复杂查询场景下表现突出。所提策略无需额外训练,可灵活适配多种基于 Transformer 的小模型。
原文摘要 · Abstract (English)
The collaborative paradigm of large and small language models (LMs) effectively balances performance and cost, yet its pivotal challenge lies in precisely pinpointing the moment of invocation when hallucinations arise in small LMs. Previous optimization efforts primarily focused on post-processing techniques, which were separate from the reasoning process of LMs, resulting in high computational costs and limited effectiveness. In this paper, we propose a practical invocation evaluation metric called AttenHScore, which calculates the accumulation and propagation of hallucinations during the generation process of small LMs, continuously amplifying potential reasoning errors. By dynamically adjusting the detection threshold, we achieve more accurate real-time invocation of large LMs. Additionally, considering the limited reasoning capacity of small LMs, we leverage uncertainty-aware knowledge reorganization to assist them better capture critical information from different text chunks. Extensive experiments reveal that our AttenHScore outperforms most baselines in enhancing real-time hallucination detection capabilities across multiple QA datasets, especially when addressing complex queries. Moreover, our strategies eliminate the need for additional model training and display flexibility in adapting to various transformer-based LMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。