cs.LGMay 12, 2026

DriftXpress: Faster Drifting Models via Projected RKHS Fields

Authors: Ali FalahatiElliot CreagerGautam KamathShubhankar Mohapatra

Organizations: Cheriton School of Computer Science, University of Waterloo and Vector Institute. · Department of Electrical and Computer Engineering, University of Waterloo and Vector Institute. · A*STAR Centre for Frontier AI Research, Singapore.

Abstract

Drifting Models have emerged as a new paradigm for one-step generative modeling, achieving strong image quality without iterative inference. The premise is to replace the iterative denoising process in diffusion models with a single evaluation of a generator. However, this creates a different trade-off: drifting reduces inference cost by moving much of the computation into training. We introduce DriftXpress, an accelerated formulation of drifting models based on projected RKHS fields. DriftXpress approximates the drifting kernel in a low-rank feature space. This preserves the attraction-repulsion structure of the original drifting field while reducing the cost of field evaluation. Across image-generation benchmarks, DriftXpress achieves comparable FID to standard drifting while reducing wall-clock training cost. These results show that the training-inference trade-off of drifting models can be pushed further without giving up their one-step inference advantage.

Explore similar work

CardsList
  1. Second Order Drifting Models

    Aug 8, 2026Drake Brown, Yuhao Huang, Shih-Hsin Wang +1Generative ModelsOrder

  2. Drift Flow Matching

    May 17, 2026Chenrui Ma, Xi Xiao, Lin Zhao +3Generative ModelsDiscrete Diffusion Model