自动挖掘关系数据中的潜在特征,提升预测模型性能。
Graph-Based Feature Augmentation for Predictive Tasks on Relational Datasets
- 基于图神经网络构建跨表关系结构,自动提取有用特征
- 在10个真实和合成数据集上显著优于现有方法
- 适合需要自动化特征工程的金融、医疗等场景
数据已成为金融、医疗、电商等领域创新的核心资产。在这些领域中,基于关系表格的预测建模广泛应用,且对通过自动化机器学习(AutoML)减少人工工作量的需求日益增长。这引发了一个关键问题:能否自动化特征增强,并识别与任务相关的关联信号?为此,我们提出一个端到端的自动化特征增强框架ReCoGNN,通过从多个关系表中提取特征来增强原始数据,以支持预测任务。ReCoGNN首先通过建模表内属性间的关系捕捉语义依赖,将表划分为结构清晰、语义一致的段落;接着构建一个异构加权图,表示所有段落间的行间关系;最后利用消息传递图神经网络在图中传播信息,指导特征选择并扩充原始数据集。在十个真实和合成数据集上的大量实验表明,ReCoGNN在分类和回归任务上均持续优于现有方法。
原文摘要 · Abstract (English)
Data has become a foundational asset driving innovation across domains such as finance, healthcare, and e-commerce. In these areas, predictive modeling over relational tables is commonly employed, with increasing emphasis on reducing manual effort through automated machine learning (AutoML) techniques. This raises an interesting question: can feature augmentation itself be automated and identify and utilize task-related relational signals? To address this challenge, we propose an end-to-end automated feature augmentation framework, ReCoGNN, which enhances initial datasets using features extracted from multiple relational tables to support predictive tasks. ReCoGNN first captures semantic dependencies within each table by modeling intra-table attribute relationships, enabling it to partition tables into structured, semantically coherent segments. It then constructs a heterogeneous weighted graph that represents inter-row relationships across all segments. Finally, ReCoGNN leverages message-passing graph neural networks to propagate information through the graph, guiding feature selection and augmenting the original dataset. Extensive experiments conducted on ten real-life and synthetic datasets demonstrate that ReCoGNN consistently outperforms existing methods on both classification and regression tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。