Data Imputation with Iterative Graph Reconstruction
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.
Code (1)
Tasks
Data Imputation 70%Graph GenerationGraph ReconstructionImputationMissing ValuesSimilar Papers 제목 키워드 기반
Clustering-Oriented Generative Attribute Graph Imputation
Attribute-missing graph clustering has emerged as a significant unsupervised task, where only attribute vectors of partial nodes are available and the graph structure is intact. The related models generally follow the tw…
Graph ClusteringGEDI: A Graph-based End-to-end Data Imputation Framework
Data imputation is an effective way to handle missing data, which is common in practical applications. In this study, we propose and test a novel data imputation process that achieve two important goals: (1) preserve the…
Graph structure learningImputationMeta-LearningPredictionSpatiotemporal Imputation with Graph-Informed Flow Matching
Missing data is a common challenge in spatiotemporal systems, arising in applications such as air quality monitoring and urban traffic management. Traditional machine learning approaches, like recurrent and graph neural …
Enabling Tensor Decomposition for Time-Series Classification via A Simple Pseudo-Laplacian Contrast
Tensor decomposition has emerged as a prominent technique to learn low-dimensional representation under the supervision of reconstruction error, primarily benefiting data inference tasks like completion and imputation, b…
Data AugmentationImputationTensor DecompositionTime Series+1IGANI: Iterative Generative Adversarial Networks for Imputation with Application to Traffic Data
Increasing use of sensor data in intelligent transportation systems calls for accurate imputation algorithms that can enable reliable traffic management in the occasional absence of data. As one of the effective imputati…
ImputationManagementTraffic Prediction