Training diffusion neural networks by backpropagating differentiable rewards
Patent Information
- Authority / Receiving Office
- EP · EP
- Patent Type
- Applications
- Current Assignee / Owner
- GDM HOLDING LLC
- Filing Date
- 2024-09-30
- Publication Date
- 2026-06-24
AI Technical Summary
Diffusion neural networks struggle to align their training data distribution with desired post-training behavior, leading to a mismatch between modeled and desired output qualities.
The system employs a differentiable reward function to fine-tune the diffusion neural network through reinforcement learning, allowing it to generate data items with desired properties by backpropagating rewards through the sampling process.
This approach enables the diffusion neural network to produce outputs that meet specific quality criteria, such as aesthetic appeal, while reducing computational and memory costs through techniques like gradient checkpointing and truncated gradient computation.
Smart Images

Figure EP2024077489_03042025_PF_FP_ABST