TabGNN: Multiplex Graph Neural Network for Tabular Data Prediction
Not on the map. Why?
Abstract (arXiv)
Tabular data prediction (TDP) is one of the most popular industrial applications, and various methods have been designed to improve the prediction performance. However, existing works mainly focus on feature interactions and ignore sample relations, e.g., users with the same education level might have a similar ability to repay the debt. In this work, by explicitly and systematically modeling sample relations, we propose a novel framework TabGNN based on recently popular graph neural networks (GNN). Specifically, we firstly construct a multiplex graph to model the multifaceted sample relations, and then design a multiplex graph neural network to learn enhanced representation for each sample. To integrate TabGNN with the tabular solution in our company, we concatenate the learned embeddings and the original ones, which are then fed to prediction models inside the solution. Experiments on eleven TDP datasets from various domains, including classification and regression ones, show that TabGNN can consistently improve the performance compared to the tabular solution AutoFE in 4Paradigm.
Key facts. Cited by 26 papers in the map (data through 14 Oct 2026). Builds on 18 of its 36 references that are in the map. First posted on arXiv 20 Aug 2021.
Builds on
18 of 36 references.
References · 18
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2018How Powerful are Graph Neural Networks?cited by 4,393
- 2016Geometric deep learning: going beyond Euclidean datacited by 1,048
- 2016Wide & Deep Learning for Recommender Systemscited by 747
- 2017DeepFM: A Factorization-Machine based Neural Network for CTR Predictioncited by 676
- 2017Deep Interest Network for Click-Through Rate Predictioncited by 544
- 2019Behavior Sequence Transformer for E-commerce Recommendation in Alibabacited by 130
- 2019Representation Learning for Attributed Multiplex Heterogeneous Networkcited by 91
- 2020AutoFIS: Automatic Feature Interaction Selection in Factorization Models for Click-Through Rate Predictioncited by 52
6 more
- 2020Molecular Mechanics-Driven Graph Neural Network with Multiplex Graph for Molecular Structurescited by 45
- 2018Product-based Neural Networks for User Response Prediction over Multi-field Categorical Datacited by 44
- 2019AutoCross: Automatic Feature Crossing for Tabular Data in Real-World Applicationscited by 40
- 2020Abstract Diagrammatic Reasoning with Multiplex Graph Networkscited by 37
- 2018Multi-dimensional Graph Convolutional Networkscited by 15
- 2020MTBRN: Multiplex Target-Behavior Relation Enhanced Network for Click-Through Rate Predictioncited by 2
Built on by
Citing papers per year: 2022: 3; 2023: 6; 2024: 7; 2025: 8; 2026: 2.
26 papers.
Sort:
By area:
- 2024Retrieval-Augmented Generation with Graphs (GraphRAG)cited by 172
- 2022TransTab: Learning Transferable Tabular Transformers Across Tablescited by 132
- 2025Griffin: Towards a Graph-Centric Relational Database Foundation Modelcited by 47
- 2024SwitchTab: Switched Autoencoders Are Effective Tabular Learnerscited by 38
- 2025Representation Learning for Tabular Data: A Comprehensive Surveycited by 36
- 2022Hopular: Modern Hopfield Networks for Tabular Datacited by 18
- 2024A Data-Centric Perspective on Evaluating Machine Learning Models for Tabular Datacited by 15
- 2026Relational In-Context Learning via Synthetic Pre-training with Structural Priorcited by 14
- 2024Graph Neural Networks for Tabular Data Learning: A Survey with Taxonomy and Directionscited by 13
- 2023Graph Neural Network contextual embedding for Deep Learning on Tabular Datacited by 8
- 2025AutoG: Towards automatic graph construction from tabular datacited by 7
- 2022Learning Enhanced Representations for Tabular Data via Neighborhood Propagationcited by 6
14 more
- 2023MuG: A Multimodal Classification Benchmark on Game Data with Tabular, Textual, and Visual Fieldscited by 6
- 2025From Features to Structure: Task-Aware Graph Construction for Relational and Tabular Learning with GNNscited by 6
- 2023Interpretable Graph Neural Networks for Tabular Datacited by 5
- 2023TabGSL: Graph Structure Learning for Tabular Data Predictioncited by 3
- 2024Novel Representation Learning Technique using Graphs for Performance Analyticscited by 2
- 2025TabGLM: Tabular Graph Language Model for Learning Transferable Representations Through Multi-Modal Consistency Minimizationcited by 2
- 2023Between-Sample Relationship in Learning Tabular Data Using Graph and Attention Networkscited by 1
- 2023AdaMedGraph: Adaboosting Graph Neural Networks for Personalized Medicinecited by 1
- 2024Interpretable Graph Neural Networks for Heterogeneous Tabular Datacited by 1
- 2024Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Datasetcited by 1
- 2025IVGAE: Handling Incomplete Heterogeneous Data with a Variational Graph Autoencodercited by 1
- 2025Can Graphs Improve Tabular Foundation Models?cited by 0
- 2025A Roadmap for Applying Graph Neural Networks to Numerical Data: Insights from Cementitious Materialscited by 0
- 2026TAROT: Task-Adaptive Refinement of LLM-prior Graphs for Few-shot Tabular Learningcited by 0
- 2026TAROT: Task-Adaptive Refinement of LLM-prior Graphs for Few-shot Tabular Learning10 Jun 2026
- 2026Relational In-Context Learning via Synthetic Pre-training with Structural Prior4 Mar 2026
- 2025A Roadmap for Applying Graph Neural Networks to Numerical Data: Insights from Cementitious Materials16 Dec 2025
- 2025Can Graphs Improve Tabular Foundation Models?13 Dec 2025
- 2025IVGAE: Handling Incomplete Heterogeneous Data with a Variational Graph Autoencoder27 Nov 2025
- 2025From Features to Structure: Task-Aware Graph Construction for Relational and Tabular Learning with GNNs2 Jun 2025
- 2025Griffin: Towards a Graph-Centric Relational Database Foundation Model8 May 2025
- 2025Representation Learning for Tabular Data: A Comprehensive Survey17 Apr 2025
- 2025TabGLM: Tabular Graph Language Model for Learning Transferable Representations Through Multi-Modal Consistency Minimization26 Feb 2025
- 2025AutoG: Towards automatic graph construction from tabular data25 Jan 2025
- 2024Retrieval-Augmented Generation with Graphs (GraphRAG)31 Dec 2024
- 2024Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset17 Dec 2024
14 more
- 2024Interpretable Graph Neural Networks for Heterogeneous Tabular Data14 Aug 2024
- 2024A Data-Centric Perspective on Evaluating Machine Learning Models for Tabular Data2 Jul 2024
- 2024Novel Representation Learning Technique using Graphs for Performance Analytics19 Jan 2024
- 2024Graph Neural Networks for Tabular Data Learning: A Survey with Taxonomy and Directions4 Jan 2024
- 2024SwitchTab: Switched Autoencoders Are Effective Tabular Learners4 Jan 2024
- 2023AdaMedGraph: Adaboosting Graph Neural Networks for Personalized Medicine24 Nov 2023
- 2023Interpretable Graph Neural Networks for Tabular Data17 Aug 2023
- 2023Between-Sample Relationship in Learning Tabular Data Using Graph and Attention Networks11 Jun 2023
- 2023TabGSL: Graph Structure Learning for Tabular Data Prediction25 May 2023
- 2023Graph Neural Network contextual embedding for Deep Learning on Tabular Data11 Mar 2023
- 2023MuG: A Multimodal Classification Benchmark on Game Data with Tabular, Textual, and Visual Fields6 Feb 2023
- 2022Learning Enhanced Representations for Tabular Data via Neighborhood Propagation14 Jun 2022
- 2022Hopular: Modern Hopfield Networks for Tabular Data1 Jun 2022
- 2022TransTab: Learning Transferable Tabular Transformers Across Tables19 May 2022
- 2024Retrieval-Augmented Generation with Graphs (GraphRAG)cited by 172
- 2024Retrieval-Augmented Generation with Graphs (GraphRAG)31 Dec 2024
- 2022TransTab: Learning Transferable Tabular Transformers Across Tablescited by 132
- 2025Griffin: Towards a Graph-Centric Relational Database Foundation Modelcited by 47
- 2024SwitchTab: Switched Autoencoders Are Effective Tabular Learnerscited by 38
- 2025Representation Learning for Tabular Data: A Comprehensive Surveycited by 36
- 2022Hopular: Modern Hopfield Networks for Tabular Datacited by 18
- 2024A Data-Centric Perspective on Evaluating Machine Learning Models for Tabular Datacited by 15
- 2026Relational In-Context Learning via Synthetic Pre-training with Structural Priorcited by 14
- 2024Graph Neural Networks for Tabular Data Learning: A Survey with Taxonomy and Directionscited by 13
- 2023Graph Neural Network contextual embedding for Deep Learning on Tabular Datacited by 8
- 2025AutoG: Towards automatic graph construction from tabular datacited by 7
- 2022Learning Enhanced Representations for Tabular Data via Neighborhood Propagationcited by 6
- 2023MuG: A Multimodal Classification Benchmark on Game Data with Tabular, Textual, and Visual Fieldscited by 6
12 most cited of 25 All 25 (JSON)
- 2026TAROT: Task-Adaptive Refinement of LLM-prior Graphs for Few-shot Tabular Learning10 Jun 2026
- 2026Relational In-Context Learning via Synthetic Pre-training with Structural Prior4 Mar 2026
- 2025A Roadmap for Applying Graph Neural Networks to Numerical Data: Insights from Cementitious Materials16 Dec 2025
- 2025Can Graphs Improve Tabular Foundation Models?13 Dec 2025
- 2025IVGAE: Handling Incomplete Heterogeneous Data with a Variational Graph Autoencoder27 Nov 2025
- 2025From Features to Structure: Task-Aware Graph Construction for Relational and Tabular Learning with GNNs2 Jun 2025
- 2025Griffin: Towards a Graph-Centric Relational Database Foundation Model8 May 2025
- 2025Representation Learning for Tabular Data: A Comprehensive Survey17 Apr 2025
- 2025TabGLM: Tabular Graph Language Model for Learning Transferable Representations Through Multi-Modal Consistency Minimization26 Feb 2025
- 2025AutoG: Towards automatic graph construction from tabular data25 Jan 2025
- 2024Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset17 Dec 2024
- 2024Interpretable Graph Neural Networks for Heterogeneous Tabular Data14 Aug 2024
12 newest of 25 All 25 (JSON)