blog.alexalemi.com KL is All You Need
Modern machine learning is a sea of initialisms: VAE, VIB, VDM, BBB, VB, etc. But, the more time I spend working in this field the more I come to appreciate that the core of essentially all modern machine learning methods is a single universal objective: Kullback-Leibler (KL) divergence minimization. Even better, there is a very simple universal recipe you can follow to rederive most of the named objectives out there. Understand KL, understand the recipe, and you'll understand all of these methods and be well on your way to deriving your own. In the past I've discussed some of the special properties of KL divergence, and how you can derive VAEs or Diffusion Models by means of a simple KL objective. What follows is an extension of those ideas, essentially a written version of a recent talk [slides] I gave at the InfoCog Workshop at NeurIPS 2024. 1 Before we get into it, we need to make sure we're all starting on the same page. Because KL divergence is so fundamental and special (as I've
KL is All You Need Alexander A. Alemi. 2024-01-08 Modern machine learning is a sea of initialisms: VAE, VIB, VDM, BBB, VB, etc. But, the more time I spend working in this field the more I come to appreciate that the core of essentially all modern machine learning methods is a single universal objective: Kullback-Leibler (KL) divergence minimization. Even better, there is a very simple universal recipe you can follow to rederive most of the named objectives out there. Understand KL, understand the recipe, and you'll understand all of these methods and be well on your way to deriving your own. I
Explore this link on the map →related reading
- Six (and a half) intuitions for KL divergence — LessWronglesswrong.com
- Approximating KL Divergencejoschu.net
- Kullback–Leibler divergence - Wikipediaen.wikipedia.org
- Mediumandacdemir-94699.medium.com
- Andy Jonesandrewcharlesjones.github.io
- Eric Jang: A Beginner's Guide to Variational Methods: Mean-Field Approximationblog.evjang.com
- Visual Information Theory -- colah's blogcolah.github.io
- Tutorial - What is a variational autoencoder? – Jaan Lı 李jaan.io
- RL with KL penalties is better seen as Bayesian inference — LessWronglesswrong.com
- Exponential families from a single KL identityarxiv.org
- What are Diffusion Models? | Lil'Loglilianweng.github.io
- blog.alexalemi.com Why KL?blog.alexalemi.com