中文

FLAME:激活冻结大语言模型进行知识图谱补全

计算与语言 2026-01-30 v5

摘要

传统知识图谱补全(KGC)方法仅依赖结构信息,难以处理稀疏性问题;而大型语言模型(LLM)通过丰富的世界知识和强大的上下文建模能力克服了这些限制。微调 LLM 效果显著,但成本高昂;而不微调的 LLM 则高效但表现次优。为解决这一权衡问题,本文提出 FLAME 框架,从冻结 LLM 中间层的上下文感知隐藏状态中提取信息,以训练数据高效的 KGC 分类器。我们通过基于子图的实体描述桥接 LLM-KG 语义差距,并采用切片互信息(SMI)量化表示中的任务相关信息。实验表明,FLAME 在非微调 LLM 基线上实现 47% 的改进,并且据称是首次以 188×188\times 内存效率和 26.11×26.11\times 加速实现达到微调水平。

关键词

引用

@article{arxiv.2408.06787,
  title  = {FLAME: Empowering Frozen LLMs for Knowledge Graph Completion},
  author = {Bo Xue and Yi Xu and Bolei Ma and Yunchong Song and Jiaxin Ding and Luoyi Fu and Xinbing Wang},
  journal= {arXiv preprint arXiv:2408.06787},
  year   = {2026}
}