arXiv:2410.13018cs.AIcs.CL2024-10被引 1

让模型学会跨结构推理,提升对新知识和复杂问题的泛化能力。

Learning Representations for Reasoning: Generalizing Across Diverse Structures

  • 用动态规划与神经算子学习新实体路径表示
  • 通过关系图将新关系转为新实体,实现结构泛化
  • 构建系统支持大规模结构数据训练,加速开发

推理是人类逻辑推断能力的体现,与感知共同构成人工智能两大核心。尽管深度学习在感知任务上已超越人类水平,但在推理领域进展缓慢。主要原因是推理问题常具有灵活的知识与查询结构,现有模型仅在训练时见过的结构上表现良好。本文旨在突破推理模型的泛化边界,提出可跨知识与查询结构泛化的算法及加速结构化数据开发的系统。第一部分提出框架,利用动态规划中的神经算子学习新实体的路径表示,并通过构建关系图将新关系映射为新实体。第二部分针对知识图谱与文本的多步查询,分别采用多轮图神经网络与模糊逻辑操作,以及显式文本规则学习来提升大模型性能。第三部分提出两个系统:一个将结构化数据作为一等公民,降低算法开发门槛;另一个节点嵌入系统解决嵌入矩阵的GPU内存瓶颈,支持百亿节点图的扩展。

原文摘要 · Abstract (English)

Reasoning, the ability to logically draw conclusions from existing knowledge, is a hallmark of human. Together with perception, they constitute the two major themes of artificial intelligence. While deep learning has pushed the limit of perception beyond human-level performance, the progress in reasoning domains is way behind. One fundamental reason is that reasoning problems usually have flexible structures for both knowledge and queries, and many existing models only perform well on structures seen during training. Here we aim to push the boundary of reasoning models by devising algorithms that generalize across knowledge and query structures, as well as systems that accelerate development on structured data. This thesis consists of three parts. In Part I, we study models that can inductively generalize to unseen knowledge graphs with new entity and relation vocabularies. For new entities, we propose a framework that learns neural operators in a dynamic programming algorithm computing path representations. For relations, we construct a relation graph to capture the interactions between relations, thereby converting new relations into new entities. In Part II, we propose two solutions for generalizing across multi-step queries on knowledge graphs and text respectively. For knowledge graphs, we show that multi-step queries can be solved by multiple calls of graph neural networks and fuzzy logic operations. For text, we devise an algorithm to learn explicit knowledge as textual rules to improve large language models on multi-step queries. In Part III, we propose two systems to facilitate machine learning development on structured data. Our library treats structured data as first-class citizens and removes the barrier for developing algorithms on structured data. Our node embedding system solves the GPU memory bottleneck of embedding matrices and scales to graphs with billion nodes.

知识图谱多步推理结构泛化图神经网络

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