Can we learn how a population evolves just by observing it? ๐๐๐
With @lazar_atan and @k_neklyudov, we came up with Wasserstein Lagrangian Mechanics (WLM), which generalizes least action to the population level.
See the ๐งต for our #ICML2026 spotlight paper and some fun gifs๐ฆ
Population dynamics (eg murmuration of birds ๐ฆ๐ฆ๐ฆ) is notoriously hard to learn; choosing the right model for the dynamics is even harder.
In our #ICML2026 spotlight, we introduce Wasserstein Lagrangian Mechanics (WLM) for learning population dynamics from observations, which
- Covers both first-order (gradient descent) and second-order dynamics (e.g. oscillations)
- Allows learning more expressive dynamics (including complex interactions) with fewer assumptions
- Generalizes in space (across different initial conditions) and time (beyond the training time snapshots)
[1/n] ๐งต
There are many posters at @icmlconf, and one of them is for our spotlight paper on learning population dynamics (done with @lazar_atan and @k_neklyudov). Come by poster #1515 in Hall A today from 5pm-6:45pm for any questions or complaints
A boid in the hand is better than two in the bush.๐ฆ
Fortunately, @guanton_soup and I have plenty more than one boid at our Wasserstein Lagrangian Mechanics poster @icmlconf. (5pm, Hall A, #1515)
If you're interested in population dynamics and OT (๐๏ธ๐งฌ๐), come flock by!
Riemannian MeanFlow ๐๐ will be at #ICML2026! (donโt worry though, itโs actually a pretty NiceFlow ๐)
RMF extends the MeanFlow framework to Riemannian manifolds by learning average velocities directly on the manifold -- this enables fast generation of biomolecules in high dimensions. โก๏ธ
RMF can generate discrete DNA sequences on the simplex manifold โ language flow maps in biology! ๐งฌ
We also show that reward guidance can generate proteins and DNA with structures or functions that we want using fewer steps. ๐
Huge congrats to @dywoo1247 who led this project and worked incredibly hard to identify the best losses and practices for training these challenging models ๐ฅ had an awesome time working on this with @hyuunnnnnn, @k_neklyudov, and @sungsoo_ahn_ ๐ค
also, if you like this work, check out GFM from Davis et al. who approach the same problem from a different angle :)
Excited to finally share Wasserstein Lagrangian Mechanics (WLM๐๏ธ)!
We developed a way to learn population mechanics directly from snapshot data! It was an absolute blast working with @k_neklyudov and @guanton_soup.
Check the ๐งต for the deep dive. See you at #ICML2026!