Missing Data Imputation using Optimal Transport - Sorbonne Université Access content directly
Conference Papers Year : 2020

Missing Data Imputation using Optimal Transport


Missing data is a crucial issue when applying machine learning algorithms to real-world datasets. Starting from the simple assumption that two batches extracted randomly from the same dataset should share the same distribution, we leverage optimal transport distances to quantify that criterion and turn it into a loss function to impute missing data values. We propose practical methods to minimize these losses using end-to-end learning, that can exploit or not parametric assumptions on the underlying distributions of values. We evaluate our methods on datasets from the UCI repository, in MCAR, MAR and MNAR settings. These experiments show that OT-based methods match or out-perform state-of-the-art imputation methods, even for high percentages of missing values.
Fichier principal
Vignette du fichier
Missing_Data_Imputation_with_Optimal_Transport__ICML_.pdf (3.03 Mo) Télécharger le fichier
Origin : Files produced by the author(s)

Dates and versions

hal-03938752 , version 1 (13-01-2023)


  • HAL Id : hal-03938752 , version 1


Boris Muzellec, Julie Josse, Claire Boyer, Marco Cuturi. Missing Data Imputation using Optimal Transport. ICML 2020 - International Conference on Machine Learning, 2020, Online, France. ⟨hal-03938752⟩
19 View
37 Download


Gmail Facebook X LinkedIn More