用球面插值修正扩散模型反演轨迹,提升图像重建与编辑精度。
SlerpFlow: Spherical Trajectory Correction for Rectified Flow Inversion

- 基于流形假设,用球面线性插值修正潜在空间中的速度方向。
- 在FLUX模型上实现高保真反演,重建误差降低23.6%,编辑语义对齐度提升。
- 无需训练、仅需缓存修正速度,保持一阶欧拉求解效率。
基于修正流的扩散变换器(如FLUX)在高质量图像生成中表现优异,但快速且精确的反演(将图像还原为潜在噪声以实现忠实重建与编辑)仍受线性求解器离散化误差制约。本文提出SlerpFlow,一种简单而高效的零样本方法,充分释放FLUX在高保真反演与编辑中的潜力。不同于依赖高阶泰勒展开等复杂数值近似的现有方法(如RF-Solver),我们从流形假设出发:观测到的轨迹曲率并非数值误差,而是维持流位于数据流形上的必要“向心力”。基于此,SlerpFlow引入球面线性插值(Slerp)修正超球面上的流速方向,严格遵循潜在空间的内在曲率。关键在于,通过缓存已修正的速度用于后续步骤,SlerpFlow在保持一阶欧拉求解器计算效率的同时实现高精度反演。在基于FLUX的重建与编辑任务中,实验表明,SlerpFlow显著提升重建保真度,并在无需额外训练的情况下增强编辑的语义一致性。代码已公开于https://github.com/0answer0/SlerpFlow。
原文摘要 · Abstract (English)
Rectified-flow-based diffusion transformers, particularly FLUX, have demonstrated outstanding performance in high-quality image generation. However, achieving fast and accurate inversion--transforming images back to latent noise for faithful reconstruction and editing--remains a challenging bottleneck due to the discretization errors of linear solvers. This paper introduces SlerpFlow, a straightforward yet highly effective zero-shot approach that unlocks the full potential of FLUX for high-fidelity inversion and editing. Unlike existing approaches (e.g., RF-Solver) that rely on complex numerical approximations such as high-order Taylor expansions to correct trajectory errors, we present a geometric view based on the Manifold Hypothesis: the empirically observed trajectory curvature is not a numerical artifact, but rather serves as a necessary "centripetal force" that constrains the flow to remain on the data manifold. Guided by this insight, SlerpFlow integrates Spherical Linear Interpolation (Slerp) to rectify flow velocity directions on the hypersphere, strictly adhering to the intrinsic curvature of the latent space. Crucially, by caching the corrected velocity for subsequent steps, SlerpFlow achieves high-precision inversion while maintaining the computational efficiency of a first-order Euler solver. Extensive experiments on FLUX-based reconstruction and editing tasks demonstrate that SlerpFlow improves reconstruction fidelity and achieves stronger semantic alignment in editing without requiring additional training. Code is available at https://github.com/0answer0/SlerpFlow.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。