论文

ResilPhase:面向扩散加速的即插即用相位映射与噪声稳健宏观轨迹外推

ResilPhase: Plug-and-Play Phase Mapping and Noise-Resilient Macro-Trajectory Extrapolation for Diffusion Acceleration

模型推理推理加速

摘要

强大扩散模型的采用受其显著推理延迟阻碍。近期“缓存后预报”方案经基于导数的多项式加速DiT缓解该问题——但在高加速比下 suffer 严重质量退化。我们的分析揭示其根因:在与连续扩散轨迹错位且数值不稳定的表示上执行离散外推。因此加速DiT承受累积空间误差、噪声导数放大与高阶不稳定。因此——我们把加速推理重新表述为常微分方程(ODE)空间中的稳定宏观轨迹外推。我们不预测中间特征——而是把预报对齐到模型的全局漂移(GD)——即端到端状态演化——从而消除特征不一致与内存开销。但即使这条平滑宏观轨迹仍易受导数谬误影响:其高阶时间导数本质带噪。因此——我们引入免导数的重心拉格朗日外推器——有效绕过导数不稳定与近似误差。我们进一步提出有界相位映射——正则化外推域——抑制振荡误差增长。这些元素共同构成ResilPhase——噪声稳健加速框架。在FLUX.1-dev与HunyuanVideo上的实验展示激进加速比下的SOTA保真度。

ResilPhase:面向扩散加速的即插即用相位映射与噪声稳健宏观轨迹外推:论文配图
图 1:通过相位映射抑制数值不稳定。 (a) 均匀时间步长上的标准轨迹外推会受到龙格现象的影响,导致混沌边缘振荡。 (b) 我们的方法将离散步骤映射到有界相空间,最小化稳定拟合的误差范围。