Score: 0

RayFlow: Instance-Aware Diffusion Acceleration via Adaptive Flow Trajectories

Published: March 10, 2025 | arXiv ID: 2503.07699v2

By: Huiyang Shao , Xin Xia , Yuhong Yang and more

Potential Business Impact:

Makes AI create pictures much faster and better.

Business Areas:
Autonomous Vehicles Transportation

Diffusion models have achieved remarkable success across various domains. However, their slow generation speed remains a critical challenge. Existing acceleration methods, while aiming to reduce steps, often compromise sample quality, controllability, or introduce training complexities. Therefore, we propose RayFlow, a novel diffusion framework that addresses these limitations. Unlike previous methods, RayFlow guides each sample along a unique path towards an instance-specific target distribution. This method minimizes sampling steps while preserving generation diversity and stability. Furthermore, we introduce Time Sampler, an importance sampling technique to enhance training efficiency by focusing on crucial timesteps. Extensive experiments demonstrate RayFlow's superiority in generating high-quality images with improved speed, control, and training efficiency compared to existing acceleration techniques.

Page Count
23 pages

Category
Computer Science:
Machine Learning (CS)