cs.LGApr 13, 2026

Tracking High-order Evolutions via Cascading Low-rank Fitting

Authors: Zhao Song

Organizations: Simons Institute for Theory of Computing, UC Berkeley

Abstract

Diffusion models have become the de facto standard for modern visual generation, including well-established frameworks such as latent diffusion and flow matching. Recently, modeling high-order dynamics has emerged as a promising frontier in generative modeling. Rather than only learning the first-order velocity field that transports random noise to a target data distribution, these approaches simultaneously learn higher-order derivatives, such as acceleration and jerk, yielding a diverse family of higher-order diffusion variants. To represent higher-order derivatives, naive approaches instantiate separate neural networks for each order, which scales the parameter space linearly with the derivative order. To overcome this computational bottleneck, we introduce cascading low-rank fitting, an ordinary differential equation inspired method that approximates successive derivatives by applying a shared base function augmented with sequentially accumulated low-rank components. Theoretically, we analyze the rank dynamics of these successive matrix differences. We prove that if the initial difference is linearly decomposable, the generic ranks of high-order derivatives are guaranteed to be monotonically non-increasing. Conversely, we demonstrate that without this structural assumption, the General Leibniz Rule allows ranks to strictly increase. Furthermore, we establish that under specific conditions, the sequence of derivative ranks can be designed to form any arbitrary permutation. Finally, we present a straightforward algorithm to efficiently compute the proposed cascading low-rank fitting.

Explore similar work

CardsList
  1. Second Order Drifting Models

    Aug 8, 2026Drake Brown, Yuhao Huang, Shih-Hsin Wang +1One-Step Generative ModelingDrifting Models

  2. Reducing Diffusion Model Memorization with Higher Order Langevin Dynamics

    May 18, 2026Benjamin Sterling, Mónica F. Bugallo, Tom TirerLangevin DynamicsMemorization in Generative Models

  3. Breaking the Curse of Dimensionality: Diffusion Models Efficiently Learn Low-Dimensional Distributions

    Sep 4, 2024Peng Wang, Huijie Zhang, Zekai Zhang +3Intrinsic DimensionalitySample Complexity