Debiasing Stochastic Gradient Descent to handle missing values

Aude Sportisse (1) , Claire Boyer (1, 2) , Aymeric Dieuleveut (3) , Julie Josse (3, 4)

1 LPSM (UMR_8001) - Laboratoire de Probabilités, Statistique et Modélisation
2 DMA - Département de Mathématiques et Applications - ENS Paris
3 CMAP - Centre de Mathématiques Appliquées - Ecole Polytechnique
4 Inria Saclay - Ile de France

A major caveat of large scale data is their incom-pleteness. We propose an averaged stochastic gradient algorithm handling missing values in linear models. This approach has the merit to be free from the need of any data distribution modeling and to account for heterogeneous missing proportion. In both streaming and finite-sample settings, we prove that this algorithm achieves convergence rate of O(1 n) at the iteration n, the same as without missing values. We show the convergence behavior and the relevance of the algorithm not only on synthetic data but also on real data sets, including those collected from medical register.

Mots clés: missing data; stochastic approximation; optimization; supervised learning

Domaines: Statistiques [math.ST]

Aude Sportisse, Claire Boyer, Aymeric Dieuleveut, Julie Josse. Debiasing Stochastic Gradient Descent to handle missing values. 2020. ⟨hal-02483651v1⟩