自动从表格数据生成高质量图结构,无需人工干预。
AutoG: Towards automatic graph construction from tabular data
- 用大模型自动构建图结构 schema,替代人工设计。
- 实验表明自动生成图性能接近人工专家水平。
- 适用于需要快速构建图数据的科研与工业场景。
近年来图机器学习(GML)取得显著进展,应用遍及多个领域。然而,研究重点多集中于模型本身,忽视了关键的前置步骤:从常见数据格式(如表格数据)中构建合适的图。这一过程对图模型应用至关重要,却长期缺乏系统研究与规范。本文旨在填补这一空白,正式定义图构建问题并提出有效解决方案。我们识别出两大挑战:1. 缺乏专门用于评估图构建方法的数据集;2. 现有自动方法仅适用于特定场景,高质量图仍需大量人工设计。为此,我们提出两项贡献:首先,构建了一组用于形式化和评估图构建方法的数据集;其次,提出基于大模型的 AutoG 框架,可自动生成高质量图结构,无需人工参与。实验表明,图的质量直接影响下游任务表现,AutoG 生成的图质量可媲美人工专家成果。代码已开源:https://github.com/amazon-science/Automatic-Table-to-Graph-Generation。
原文摘要 · Abstract (English)
Recent years have witnessed significant advancements in graph machine learning (GML), with its applications spanning numerous domains. However, the focus of GML has predominantly been on developing powerful models, often overlooking a crucial initial step: constructing suitable graphs from common data formats, such as tabular data. This construction process is fundamental to applying graph-based models, yet it remains largely understudied and lacks formalization. Our research aims to address this gap by formalizing the graph construction problem and proposing an effective solution. We identify two critical challenges to achieve this goal: 1. The absence of dedicated datasets to formalize and evaluate the effectiveness of graph construction methods, and 2. Existing automatic construction methods can only be applied to some specific cases, while tedious human engineering is required to generate high-quality graphs. To tackle these challenges, we present a two-fold contribution. First, we introduce a set of datasets to formalize and evaluate graph construction methods. Second, we propose an LLM-based solution, AutoG, automatically generating high-quality graph schemas without human intervention. The experimental results demonstrate that the quality of constructed graphs is critical to downstream task performance, and AutoG can generate high-quality graphs that rival those produced by human experts. Our code can be accessible from https://github.com/amazon-science/Automatic-Table-to-Graph-Generation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。