Differentiable Programming \`a la Moreau
Optimization and Control
2022-12-13 v2 Machine Learning
Machine Learning
Abstract
The notion of a Moreau envelope is central to the analysis of first-order optimization algorithms for machine learning. Yet, it has not been developed and extended to be applied to a deep network and, more broadly, to a machine learning system with a differentiable programming implementation. We define a compositional calculus adapted to Moreau envelopes and show how to integrate it within differentiable programming. The proposed framework casts in a mathematical optimization framework several variants of gradient back-propagation related to the idea of the propagation of virtual targets.
Cite
@article{arxiv.2012.15458,
title = {Differentiable Programming \`a la Moreau},
author = {Vincent Roulet and Zaid Harchaoui},
journal= {arXiv preprint arXiv:2012.15458},
year = {2022}
}
Comments
Short version appeared in ICASSP 2022 - 2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)