用自生成奖励提升大模型关系抽取的准确率与可解释性
The Answer Lies Within: Self-Derived Rewards Enable Explainable Relation Extraction
- 模仿人类文本处理流程,分步推理关系抽取
- 通过自动提取正确答案中的关键词生成奖励,提升模型对关键信息的聚焦
- 显著改善结果可解释性,适合需要透明推理的场景
尽管大语言模型具备强大的推理能力,但在无预定义关系标签的一次性关系抽取任务中仍表现不佳。我们识别出两个关键问题:模型常被无关词汇误导,而非关注蕴含关系的语义;且推理层级与人工标注者期望不一致。为此提出新框架,包含两项核心设计:(1) COGRE——受认知启发的推理框架,将关系抽取分解为类人化的多阶段处理过程;(2) HIT@DICT——基于强化学习的中间奖励策略,通过奖励推理中与关系相关的短语来引导模型对齐关系标签,奖励信号源自从正确预测中自动提取的关键词词典。实验表明,该框架同时提升了准确率与解释质量。例如,Qwen2.5-14B-Instruct 搭配 COGRE 在 One-shot NYT29 上达 24.65% F1,超越现有基于推理的方法;进一步结合 HIT@DICT 强化学习后性能提升 +23.46 个百分点。人工评估显示,最优模型生成的关系短语与标准答案高度一致,解释质量评分相对提升 54%。
原文摘要 · Abstract (English)
Despite the remarkable reasoning capabilities of large language models, they still struggle with one-shot relation extraction without predefined relation labels. We identify two pitfalls: models are often misled by irrelevant tokens instead of relation-conveying semantics, and they often fail to align with the abstraction level human annotators expect. We introduce a novel framework that closes this gap with two components: (1) COGRE, a cognitively-inspired reasoning framework that structures RE into a series of processes mimicking human text-processing; and (2) HIT@DICT, a reinforcement learning intermediate reward strategy that encourages reasoning to align with relational labels by rewarding relation-relevant phrases in reasoning. The reward is derived on a credit dictionary automatically extracted from correct predictions. Our experiments show that our framework improves both accuracy and explanation quality by addressing these two pitfalls. For example, COGRE with Qwen2.5-14B-Instruct on One-shot NYT29 achieves 24.65% F1, surpassing prior reasoning-based designs. Optimizing this approach with RL using HIT@DICT further improves performance by +23.46% points. Finally, human evaluation shows that our best model generates relational phrases closely aligned with gold labels, increasing human explanation quality ratings by 54% (relative).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。