Efficient learning with robust gradient descent

From MaRDI portal
Publication:2320583



Abstract: Minimizing the empirical risk is a popular training strategy, but for learning tasks where the data may be noisy or heavy-tailed, one may require many observations in order to generalize well. To achieve better performance under less stringent requirements, we introduce a procedure which constructs a robust approximation of the risk gradient for use in an iterative learning routine. Using high-probability bounds on the excess risk of this algorithm, we show that our update does not deviate far from the ideal gradient-based update. Empirical tests using both controlled simulations and real-world benchmark data show that in diverse settings, the proposed procedure can learn more efficiently, using less resources (iterations and observations) while generalizing better.


Gradient descent is a very popular optimization method with wide applications to many machine learning problems. However, gradient descent may not generalize well if the training data is noisy or heavy-tailed. To address this defect, this paper proposes an efficient robust gradient descent method. The idea is to use a robust estimate of the risk gradient rather than the risk itself, which is achieved by a soft-truncation of errant values. The proposed algorithm enjoys the computation ease of gradient descent, and is backed up by both theoretical and experimental analyses.





Describes a project that uses

Uses Software






This page was built for publication: Efficient learning with robust gradient descent

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q2320583)