Reward Transport Enables Property Control in Flow Matching via Noise Alignment
July 13, 2026
Reward Transport utilizes optimal transport coupling during training to align noise-space coordinates with scalar molecular rewards. This allows for distribution-level steering at inference time without requiring an external reward model, gradient guidance, or additional computational overhead.
HOW THIS AFFECTS YOU
●
builderYou can implement controllable molecular generation without the latency of an oracle reward model at inference time.
●
researcherYou can steer generative distributions using noise-space coordinates instead of iterative gradient guidance.