We release Diamond Maps💎 unlocking accurate and efficient guidance for diffusion models. Our experiments show that our methods scale incredibly well. Excited to see what people will build with this!
Accurate guidance has been a notoriously hard problem, but in this work, we’re bringing TWO (!) solutions to the table. The recipe for success:
1️⃣ Speed: Use distilled models (flow maps, mean flows, consistency models).
2️⃣ Exploration: Inject stochasticity to properly explore your search space.
Because this fundamentally improves anything using flow matching and diffusion, we see a lot of potential for applications across audio, robotics, molecules, and beyond.
Paper:
arxiv.org/abs/2602.05993
Code:
github.com/PeterHolderrieth/…
Huge thanks to an amazing team: Douglas Chen,
@LucaEyring,
@ishin_shah, Giri Anantharaman,
@electronickale,
@zeynepakata, Tommi Jaakkola,
@nmboffi, and
@max_simchowitz. It was awesome bringing this to life together!