DeepGUM: Learning Deep Robust Regression with a Gaussian-Uniform Mixture Model

Stéphane Lathuilière (1, 2) , Pablo Mesejo (1, 3) , Xavier Alameda-Pineda (1) , Radu Horaud (1)

1 PERCEPTION - Interpretation and Modelling of Images and Videos
2 UNITN - Università degli Studi di Trento = University of Trento
3 UGR - Universidad de Granada = University of Granada

In this paper we address the problem of how to robustly train a Con-vNet for regression, or deep robust regression. Traditionally, deep regression employ the L2 loss function, known to be sensitive to outliers, i.e. samples that either lie at an abnormal distance away from the majority of the training samples, or that correspond to wrongly annotated targets. This means that, during back-propagation, outliers may bias the training process due to the high magnitude of their gradient. In this paper, we propose DeepGUM: a deep regression model that is robust to outliers thanks to the use of a Gaussian-uniform mixture model. We derive an optimization algorithm that alternates between the unsupervised detection of outliers using expectation-maximization, and the supervised training with cleaned samples using stochastic gradient descent. DeepGUM is able to adapt to a continuously evolving outlier distribution, avoiding to manually impose any threshold on the proportion of outliers in the training set. Extensive experimental evaluations on four different tasks (facial and fashion landmark detection, age and head pose estimation) lead us to conclude that our novel robust technique provides reliability in the presence of various types of noise and protection against a high percentage of outliers.

Mots clés: Outlier detection; Mixture model; Robust regression; Deep Neural Networks

Domaines: Vision par ordinateur et reconnaissance de formes [cs.CV]; Apprentissage [cs.LG]

Stéphane Lathuilière, Pablo Mesejo, Xavier Alameda-Pineda, Radu Horaud. DeepGUM: Learning Deep Robust Regression with a Gaussian-Uniform Mixture Model. ECCV 2018 - European Conference on Computer Vision, Sep 2018, Munich, Germany. pp.205-221, ⟨10.1007/978-3-030-01228-1_13⟩. ⟨hal-01851511⟩