Timezone: »

Handling Missing Data with Graph Representation Learning
Jiaxuan You · Xiaobai Ma · Yi Ding · Mykel J Kochenderfer · Jure Leskovec

Wed Dec 09 09:00 PM -- 11:00 PM (PST) @ Poster Session 4 #1168

Machine learning with missing data has been approached in many different ways, including feature imputation where missing feature values are estimated based on observed values and label prediction where downstream labels are learned directly from incomplete data. However, existing imputation models tend to have strong prior assumptions and cannot learn from downstream tasks, while models targeting label predictions often involve heuristics and can encounter scalability issues. Here we propose GRAPE, a framework for feature imputation as well as label prediction. GRAPE tackles the missing data problem using graph representation, where the observations and features are viewed as two types of nodes in a bipartite graph, and the observed feature values as edges. Under the GRAPE framework, the feature imputation is formulated as an edge-level prediction task and the label prediction as a node-level prediction task. These tasks are then solved with Graph Neural Networks. Experimental results on nine benchmark datasets show that GRAPE yields 20% lower mean absolute error for imputation tasks and 10% lower for label prediction tasks, compared with existing state-of-the-art methods.

Author Information

Jiaxuan You (Stanford University)

2nd CS PhD student in Stanford

Xiaobai Ma (Stanford University)
Daisy Ding (Stanford University)
Mykel J Kochenderfer (Stanford University)
Jure Leskovec (Stanford University/Pinterest)

More from the Same Authors