🚀 Introducing DiffusionART: efficient & stable & Scalable diffusion RL.
Understanding forward-process RL first: 1. ❓Positive branch and negative branch provides the same ideal gradient direction, but why do reward collapses without any of them? 2. ❓Why does EMA off-policy strategy stablize training?
⚡ On HunyuanVideo-13B T2V: DiffusionART is 12.02× / 11.72× more efficient than DiffusionNFT and Flow-GRPO-Fast.
📄 Arxiv: https://t.co/mBjBE4b3ps
💻 GitHub:https://t.co/1mtCdu7cxT
🤗 Hugging Face Papers:https://t.co/OLrLjZ95KV
After nearly 800 wandb runs, we found some surprising answers. 🧵👇
I'm increasingly convinced that MAI-Thinking-1 and Qwen3.8-Flash-Next are two must-read technical reports for anyone who isn't on a frontier pretraining team but wants to learn the practical know-how behind training frontier models.
wandb 0.31.0 is out now, and import wandb and the CLI start about twice as fast.
That's for builders like @Vanzllll, whose DiffusionART team "found some surprising answers" after nearly 800 wandb runs.
Now find yours.
It also grows W&B LEET, the terminal UI for exploring W&B runs. wandb leet inspect --summary reads a run's local .wandb file, and --json prints that as JSON for a coding agent.
pip install -U wandb
DiffusionART has received a lot of attention since public, and I am really happy to hear about that the some frontier VideoGen companies of China have been trying to put it to use for their post-training pipeline and get a good result.
I'd like to provide more insight about the meaning of "aggregation-first" since some researchers reached out to me for discussing this confusing term: NFT-style guidance to each state contains an harmful consistency force and the dual branch cancel them only partially via cross-sample gradient aggregation (a.k.a loss = mean(per_sample_loss)). ART-style training uses the Covariance of endpoint and reward to aggregate a clean reward-improving direction before applying this guidance to each state, systematically cancelling the harmful force.
Interestingly, we can regard DiffusionART as the combination of Forward-Process RL and Reverse-Process RL. Meanwhile, the term "guidance" is inspired by CFG (classifier-free guidance), and shares some basic principle. So another way for interpretation is ART is "internalizing" the "reward-improving guidance" to model weight via MSE loss.
🚀 Introducing DiffusionART: efficient & stable & Scalable diffusion RL.
Understanding forward-process RL first: 1. ❓Positive branch and negative branch provides the same ideal gradient direction, but why do reward collapses without any of them? 2. ❓Why does EMA off-policy strategy stablize training?
⚡ On HunyuanVideo-13B T2V: DiffusionART is 12.02× / 11.72× more efficient than DiffusionNFT and Flow-GRPO-Fast.
📄 Arxiv: https://t.co/mBjBE4b3ps
💻 GitHub:https://t.co/1mtCdu7cxT
🤗 Hugging Face Papers:https://t.co/OLrLjZ95KV
After nearly 800 wandb runs, we found some surprising answers. 🧵👇
https://t.co/A1dlSj77cN
Thats what we do in this work, FineVLA.
The upper bound of VLA might be the image/video gen, which is very good when the prompt is quite well (usually polished with a llm refiner)
And until now, the only one intelligence I see like human is the CoT lol.
🚀 Introducing DiffusionART: efficient & stable & Scalable diffusion RL.
Understanding forward-process RL first: 1. ❓Positive branch and negative branch provides the same ideal gradient direction, but why do reward collapses without any of them? 2. ❓Why does EMA off-policy strategy stablize training?
⚡ On HunyuanVideo-13B T2V: DiffusionART is 12.02× / 11.72× more efficient than DiffusionNFT and Flow-GRPO-Fast.
📄 Arxiv: https://t.co/mBjBE4b3ps
💻 GitHub:https://t.co/1mtCdu7cxT
🤗 Hugging Face Papers:https://t.co/OLrLjZ95KV
After nearly 800 wandb runs, we found some surprising answers. 🧵👇
Our analysis suggests it's not primarily about reward scale.
Interestingly, both branches provide aligned reward guidance, but their hidden forward–reverse consistency forces point in opposite directions.
Joint training partially cancels these forces, while single-branch training leaves a strong residual that can cause gradient spikes and destabilize training.
So the key issue is the uncanceled consistency force, rather than simply insufficient reward signal.
💡 Building on these insights, we introduce DiffusionART: Aggregate reward guidance first, restore as the policy moves.
① Aggregation-First Guidance: Aggregate reward signals into a shared reward-improving direction via Cov(x₀, r), then guide actual reverse-ODE rollout states using frozen MSE targets—eliminating the hidden consistency force by construction.
② Adaptive Restoration: A trust-region-masked second update introduces drift-adaptive restoration, without EMA tuning!
✅ No ODE-to-SDE conversion
✅ No differentiable reward model required
✅ Supports black-box & pairwise rewards
✅ Robust to smaller prompt group sizes
✅ Stable and efficient across image & video generation, without per-task tuning
📈 On HunyuanVideo-13B, DiffusionART achieves +28.8% / +53.0% higher mean training rewards vs DiffusionNFT / Flow-GRPO-Fast.
🔓 Everything is open-sourced!
📄 Paper: https://t.co/mBjBE4bBf0
💻 Code: https://t.co/1mtCdu7Knr
🤗 HF Papers: https://t.co/OLrLjZ9DAt
⭐ Stars, upvotes, and feedback are greatly appreciated!
🔍 Two informative insights:
1️⃣ Hidden Consistency Force
NFT-style forward-process objectives contains an additional term: a forward–reverse consistency force beyond reward guidance.
Why we need both positive and negative branches? Dual-branch oppose this force, and heuristically cancel this unwanted force. But can only cancel it by cross-sample gradient aggregation—not state by state—leaving destabilizing residuals and gradient spikes.
2️⃣ Why EMA Really Works
EMA stabilizes forward RL through old-policy restoration in the loss, NOT off-policy rollouts themselves.
Surprisingly, this restoration force can dominate parameter gradients despite being much weaker in output space! The fundamental difference between on-policy and off-policy training in continuous-space diffusion RL is restoration. Off-policy training can provide restoration.
I do think the 60000 Iclr submission shows that the era of researcher is coming back, as research itself can be disentangled from paper outputs and this evaluation criteria, and the researcher can pursue the groundtruth.
Students sometimes ask me if it still makes sense, in this accelerating age, to pursue a PhD in AI.
Perhaps counterintuitively, I think it's a great time to do so.
I wrote up some thoughts on this here: https://t.co/RRqGR0qi1R
@phillip_isola I think the 60000 Iclr submission shows that the era of researcher is coming back, as research itself can be disentangled from paper outputs, and the researcher can pursue the groundtruth.
I’d recommend everyone working on multimodality to read this Roboflow post on GPT-6 Astra:
https://t.co/eyyLp3poWi
After working on multimodality for half a year, I increasingly feel that some of the most important problems are still the old ones:
OCR. Grounding. Segmentation.
They rarely make the fanciest demos, but they are the foundation of everything downstream.
2D grounding is fundamental to Computer-Use Agents; 3D grounding will be fundamental to Robot-Use Agents.
These capabilities need both targeted training and end-to-end optimization inside agents. Astra is one of the clearest examples I’ve seen of the two coming together.
And that is exactly why Astra can pull off so many fancy Blender and robot-control demos: the flashy behavior is built on very unflashy fundamentals.
Multimodality is not the destination. In the agent era, perception matters because it enables action.
Everything ultimately comes down to taking action.