论文

SlerpFlow:用于整流流反演的球形轨迹校正

SlerpFlow: Spherical Trajectory Correction for Rectified Flow Inversion

应用与实践内容创作

摘要

基于整流流的扩散Transformer,特别是FLUX,在高质量图像生成方面表现出了出色的性能。然而,由于线性求解器的离散化误差,实现快速准确的反演(将图像转换回潜在噪声以进行忠实的重建和编辑)仍然是一个具有挑战性的瓶颈。本文介绍了 SlerpFlow,这是一种简单而高效的零样本方法,可释放 FLUX 在高保真反演和编辑方面的全部潜力。与依赖复杂数值近似(例如高阶泰勒展开式)来纠正轨迹误差的现有方法(例如 RF-Solver)不同,我们提出了基于流形假设的几何视图:经验观察到的轨迹曲率不是数值伪影,而是作为必要的“向心力”来约束流保留在数据流形上。在这一见解的指导下,SlerpFlow 集成了球面线性插值 (Slerp) 来校正超球面上的流速方向,严格遵守潜在空间的固有曲率。至关重要的是,通过缓存后续步骤的校正速度,SlerpFlow 实现了高精度反演,同时保持了一阶欧拉求解器的计算效率。基于 FLUX 的重建和编辑任务的大量实验表明,SlerpFlow 提高了重建保真度,并在编辑中实现了更强的语义对齐,而无需额外的训练。代码可在 https://github.com/0answer0/SlerpFlow 获取。