Check our work on scaling diffusion models at test time by self-guidance. It only uses a diffusion model on its own with no network/memory overhead.
This presents how test-time scaling law in @OpenAI o1 model succeeds in diffusion models.
The report: https://t.co/Iu1MZgWBeK.
OpenAI @OpenAI used Reinforcement FineTuning (ReFT) for COT in LLM.
Last week, we submitted a report https://t.co/yZ4RFTzt5b where we presented the ReFT for diffusion and flow based models.
Diffusion time process is like COT reasoning step by step on how to generate images.
A brave idea of training with "reinforcement learning from reward" to dynamically schedule the diffusion time.
TPDM samples images more rapidly than the benchmark flux. The longer and more complex prompt, the more slowly the diffusion time decreases to generate more details.