用图注意力模型提升数据库自动补全准确率,尤其擅长处理文本字段。
RelGT-AC: A Relational Graph Transformer for Autocomplete Tasks in Relational Databases

- 基于关系图结构建模数据库,通过掩码目标列防止过拟合
- 统一头设计支持分类与回归任务,在3个数据集上全胜基线
- 融合TF-IDF编码自由文本列,文本类任务最高提升10%性能
关系型数据库支撑现代企业、科研与医疗系统,但其多表、异构和时序结构使预测学习极具挑战。关系深度学习将数据库表示为异构图并直接应用图神经网络。RelBench v2引入了自动补全任务——根据上下文预测已有列值,类似智能表单填充。本文提出RelGT-AC,基于RelGT架构,提出三项改进:(1) 列掩码策略,在子图编码时屏蔽目标列以避免平凡解;(2) 统一任务头,支持二分类、多分类和回归任务;(3) 引入TF-IDF文本编码器,自动识别并编码自由文本列,恢复类别编码器丢失的词汇信号。在覆盖3个RelBench v2数据集(rel-trial, rel-f1, rel-stack)的7个任务中,RelGT-AC在全部3个回归任务上超越GraphSAGE基线,且在文本密集型资格筛选任务中,通过TF-IDF编码实现最高+10 AUROC提升。
原文摘要 · Abstract (English)
Relational databases underpin modern enterprise, scientific, and healthcare systems, yet predictive machine learning on such data remains challenging due to their multi-table, heterogeneous, and temporal structure. Relational Deep Learning (RDL) addresses this by representing databases as heterogeneous graphs and applying graph neural networks (GNNs) directly. RelBench v2 recently introduced autocomplete tasks -- a practically motivated task type where the goal is to predict an existing column value from relational context, analogous to an intelligent form-filling assistant. We propose RelGT-AC (Relational Graph Transformer for Autocomplete), extending the RelGT architecture with three targeted contributions: (1) a column masking strategy that prevents trivial solutions by masking the target column during subgraph encoding; (2) a unified task head supporting binary classification, multiclass classification, and regression autocomplete tasks within a single model; and (3) a TF-IDF text encoder that automatically detects and encodes free-text columns, recovering strong lexical signal that categorical encoders discard. Across 7 tasks spanning 3 RelBench v2 datasets (rel-trial, rel-f1, rel-stack), RelGT-AC outperforms the GraphSAGE baseline on all 3 regression autocomplete tasks and achieves up to +10 AUROC points on text-heavy eligibility tasks via the TF-IDF encoder.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。