Effortless optimization through gradient flows – Machine Learning Research Blog
Optimization algorithms often rely on simple intuitive principles, but their analysis quickly leads to a lot of algebra, where the original idea is not transparent. In last month post, Adrien Taylor explained how convergence proofs could be automated. This month, I will show how proof sketches can be obtained easily for algorithms based on gradient descent. This will be done using vanishing step-sizes that lead to gradient flows. The intuitive principle behind gradient descent is the quest for local descent. We thus need to characterize the local behavior of the function we aim to optimize. This is what gradients are for. In this blog post, I will consider minimizing a function 𝑓 f over ℝ 𝑑 Rd . Assuming 𝑓 f is differentiable, a first order Taylor expansion of 𝑓 f around a point 𝑥 x leads to 𝑓(𝑥+𝛿)=𝑓(𝑥)+∇𝑓(𝑥 ) ⊤ 𝛿+𝑜(‖𝛿‖), f(x+δ)=f(x)+∇f(x)⊤δ+o(‖δ‖), for any norm ‖⋅‖ ‖⋅‖ on ℝ 𝑑 Rd , where ∇𝑓(𝑥)∈ ℝ 𝑑 ∇f(x)∈Rd is the gradient of 𝑓 f at 𝑥 x , composed
Optimization algorithms often rely on simple intuitive principles, but their analysis quickly leads to a lot of algebra, where the original idea is not transparent. In last month post , Adrien Taylor explained how convergence proofs could be automated. This month, I will show how proof sketches can be obtained easily for algorithms based on gradient descent. This will be done using vanishing step-sizes that lead to gradient flows . Gradient as local information The intuitive principle behind gradient descent is the quest for local descent. We thus need to characterize the local behavior of the
related reading
- Why Momentum Really Worksdistill.pub
- Keep the gradient flowingfa.bianp.net
- Differentiable Programming from Scratchthenumb.at
- An overview of gradient descent optimization algorithmsruder.io
- Understanding the Neural Tangent Kernel – EigenTaleseigentales.com
- Stochastic Gradient Descent: An intuitive proofmedium.com
- Some Math behind Neural Tangent Kernel | Lil'Loglilianweng.github.io
- Part 3: Intro to Policy Optimization - Spinning Up documentationspinningup.openai.com
- [1609.04747] An overview of gradient descent optimization algorithmsarxiv.org
- A Visual Explanation of Gradient Descent Methods (Momentum, AdaGrad, RMSProp, Adam) | Towards Data Sciencetowardsdatascience.com
- Machine Learning Research Blog – Francis Bachfrancisbach.com
- Calculus on Computational Graphs: Backpropagation -- colah's blogcolah.github.io