跳到正文
arXiv cs.LG· Hamed Shirzad, Frederik Wenkel, Dominique Beaini, Danica J. Sutherland, Emmanuel Noutahi·· 5 小时前AI 评分22

SeedER:用 Graph Transformer 与强化学习实现高效知识图谱检索

SeedER: Seed-Expand-Retrieve for Efficient Knowledge Graph Retrieval

AI 导读

研究者提出 SeedER(Seed-Expand-Retrieve),一种基于 Graph Transformer 和强化学习的一阶段知识图谱检索器,推理时可在 CPU 上运行,训练时每次只需少量节点的反馈。该方法在检索效果上可与微调 LLM 的方案竞争,同时显著降低内存与计算开销。

来源:arXiv cs.LG · arxiv.org