arXiv:2509.20162cs.CLcs.AI2025-09EMNLP被引 1

用强化学习让大模型更懂专业领域知识,效果远超传统方法。

Embedding Domain Knowledge for Large Language Models via Reinforcement Learning from Augmented Generation

  • 通过生成增强的强化学习迭代优化模型,聚焦关键知识点。
  • 在医学、法律等4个领域数据集上准确率显著提升。
  • 适合需要精准专业推理的应用场景,如医疗问答与法律分析。

大语言模型在特定领域任务中表现受限,主要因训练数据中专业知识分布不均且静态。知识缺失与滞后导致应用时出现认知缺口。虽可通过领域数据微调嵌入知识,但现有方法存在不足:持续预训练(CPT)对文档中所有标记同等对待,忽略关键信息;监督微调(SFT)依赖问答对,难以构建复杂推理所需的连贯知识结构。为此,本文提出增强生成强化学习(RLAG),通过生成采样与奖励驱动的迭代优化,有效嵌入关键且上下文一致的领域知识。我们以最高对数概率的生成结果作为采样输出,并设计三项定制化奖励指标引导优化。为全面评估领域专长,我们考察正确回答的问题的答案准确率及解释合理性。在医学、法律、天文学和时事四类数据集上的实验表明,该方法显著优于基线模型。代码与数据已开源至 https://github.com/ChaojunNie/RLAG。

原文摘要 · Abstract (English)

Large language models (LLMs) often exhibit limited performance on domain-specific tasks due to the natural disproportionate representation of specialized information in their training data and the static nature of these datasets. Knowledge scarcity and temporal lag create knowledge gaps for domain applications. While post-training on domain datasets can embed knowledge into models, existing approaches have some limitations. Continual Pre-Training (CPT) treats all tokens in domain documents with equal importance, failing to prioritize critical knowledge points, while supervised fine-tuning (SFT) with question-answer pairs struggles to develop the coherent knowledge structures necessary for complex reasoning tasks. To address these challenges, we propose Reinforcement Learning from Augmented Generation (RLAG). Our approach iteratively cycles between sampling generations and optimizing the model through calculated rewards, effectively embedding critical and contextually coherent domain knowledge. We select generated outputs with the highest log probabilities as the sampling result, then compute three tailored reward metrics to guide the optimization process. To comprehensively evaluate domain expertise, we assess answer accuracy and the rationality of explanations generated for correctly answered questions. Experimental results across medical, legal, astronomy, and current events datasets demonstrate that our proposed method significantly outperforms baseline approaches. Our code and data are open sourced at https://github.com/ChaojunNie/RLAG.

大模型强化学习领域知识推理

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