Data Imputation with Iterative Graph Reconstruction
arXiv:2212.02810 · doi:10.1609/aaai.v37i9.26348
Abstract
Effective data imputation demands rich latent ``structure" discovery capabilities from ``plain" tabular data. Recent advances in graph neural networks-based data imputation solutions show their strong structure learning potential by directly translating tabular data as bipartite graphs. However, due to a lack of relations between samples, those solutions treat all samples equally which is against one important observation: ``similar sample should give more information about missing values." This paper presents a novel Iterative graph Generation and Reconstruction framework for Missing data imputation(IGRM). Instead of treating all samples equally, we introduce the concept: ``friend networks" to represent different relations among samples. To generate an accurate friend network with missing data, an end-to-end friend network reconstruction solution is designed to allow for continuous friend network optimization during imputation learning. The representation of the optimized friend network, in turn, is used to further optimize the data imputation process with differentiated message passing. Experiment results on eight benchmark datasets show that IGRM yields 39.13% lower mean absolute error compared with nine baselines and 9.04% lower than the second-best. Our code is available at https://github.com/G-AILab/IGRM.
Published in AAAI2023
References in corpus (12)
- Inductive Representation Learning on Large Graphs
- Categorical Reparameterization with Gumbel-Softmax
- Graph Convolutional Matrix Completion
- The Concrete Distribution: A Continuous Relaxation of Discrete Random Variables
- GAIN: Missing Data Imputation using Generative Adversarial Nets
- Matrix Completion and Low-Rank SVD via Fast Alternating Least Squares
- Missing Data Imputation with Adversarially-trained Graph Convolutional Networks
- Handling Missing Data with Graph Representation Learning
- Missing Data Imputation using Optimal Transport
- G2SAT: Learning to Generate SAT Formulas
- MIRACLE: Causally-Aware Imputation via Learning Missing Data Mechanisms
- Polynomial Matrix Completion for Missing Data Imputation and Transductive Learning