cs.CVJun 21, 2026

Curvature-Adaptive Consistency Flow Matching: Autonomous Trajectory Optimization via Reinforcement Learning

Authors: Songtao TianGuhan ChenBohan LiJingyi MaZixiong Yu

Organizations: 1Tsinghua University, Beijing, China · 2Huawei Large Model Data Technology Lab, Shenzhen, China

Abstract

Consistency distillation has significantly accelerated diffusion-model inference, but its sampling dynamics remain underexplored. We reveal an asymmetry: although Logit-Normal sampling priors work well for standard iterative generation, consistency distillation exhibits a different difficulty profile (e.g., U-shaped), with optimization bottlenecks concentrated at the boundary stages rather than intermediate steps. To address the limitations of static sampling under evolving learning demands, we propose Curvature-Adaptive Consistency Flow Matching (CACFM). By formulating distillation as a dynamic decision process, CACFM uses a lightweight reinforcement learning agent to probe Probability Flow ODE trajectories and construct an efficiency-oriented curriculum that prioritizes critical regions without manual scheduling. Combined with Flow-adapted DMD and adversarial consistency objectives, our RL-based scheduler achieves state-of-the-art results on large-scale models such as FLUX and SDXL, mitigating structural deformities and preserving high-frequency details in extreme few-step regimes.

Explore similar work

CardsList
  1. Dual-End Consistency Model

    Feb 11, 2026Linwei Dong, Ruoyu Guo, Ge Bai +3Generative Flow Networks