让大模型更懂何时该拒绝回答,避免过度拒答。
Utilize the Flow before Stepping into the Same River Twice: Certainty Represented Knowledge Flow for Refusal-Aware Instruction Tuning
- 用回答置信度筛选数据,减少训练冲突。
- 引入预回放训练,缓解知识变化带来的不一致。
- 适合需要可靠拒答能力的对话系统研发者。
拒绝感知指令微调(RAIT)使大语言模型能够拒绝回答未知问题。通过将训练数据中未知问题的原始回答修改为“我不知道”等拒绝回应,RAIT提升了模型可靠性并减少了幻觉。然而,传统方法仅依据初始模型响应的正确性进行修改,易导致模型对本可回答的问题过度拒绝,即“过拒”问题。本文分析其根源:静态冲突指特征空间相似样本接收不同监督信号(原答案与修改后的“我不知道”);动态冲突则源于微调过程中模型知识更新,使原本无法回答的问题变得可解,但训练样本仍保留初始状态的“我不知道”标签,造成不一致。为此,本文提出确定性知识流拒绝感知指令微调(CRaFT),核心贡献有二:一、引入响应置信度筛选并选择性修改数据,降低静态冲突;二、实施预回放训练以刻画模型知识状态变化,缓解微调中的动态冲突。在开放式问答和多选题任务上的实验表明,CRaFT显著提升模型在RAIT过程中的整体表现。代码与数据将公开于 https://github.com/opendatalab/CRaFT。
原文摘要 · Abstract (English)
Refusal-Aware Instruction Tuning (RAIT) enables Large Language Models (LLMs) to refuse to answer unknown questions. By modifying responses of unknown questions in the training data to refusal responses such as "I don't know", RAIT enhances the reliability of LLMs and reduces their hallucination. Generally, RAIT modifies training samples based on the correctness of the initial LLM's response. However, this crude approach can cause LLMs to excessively refuse answering questions they could have correctly answered, the problem we call over-refusal. In this paper, we explore two primary causes of over-refusal: Static conflict occurs when similar samples within the LLM's feature space receive differing supervision signals (original vs. modified "I don't know"). Dynamic conflict arises as the LLM's evolving knowledge during SFT enables it to answer previously unanswerable questions, but the now-answerable training samples still retain the original "I don't know" supervision signals from the initial LLM state, leading to inconsistencies. These conflicts cause the trained LLM to misclassify known questions as unknown, resulting in over-refusal. To address this issue, we introduce Certainty Represented Knowledge Flow for Refusal-Aware Instructions Tuning (CRaFT). CRaFT centers on two main contributions: First, we additionally incorporate response certainty to selectively filter and modify data, reducing static conflicts. Second, we implement preliminary rehearsal training to characterize changes in the LLM's knowledge state, which helps mitigate dynamic conflicts during the fine-tuning process. We conducted extensive experiments on open-ended question answering and multiple-choice question task. Experiment results show that CRaFT can improve LLM's overall performance during the RAIT process. Code and data will be released at https://github.com/opendatalab/CRaFT .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。