开源 Flow-WLS 采样器:用 Savitzky-Golay 正则化让 DiT 10 步出图零伪影

YakovSava · reddit · 2026-10-13

Reddit 研究发布帖:作者探究高阶扩散采样器(DPM-Solver++、Adams-Bashforth 4+ 阶)在 Flow Matching DiT 模型上崩溃的数学根源,并开发出免训练新求解器 Flow-WLS,在消费级 RTX 5060 上实现 11 秒的干净 10 步生成。

为什么高阶采样器会崩:

解决方案:Flow-WLS 不强拟振荡高阶多项式,而是用加权最小二乘(WLS,指数记忆衰减 γ=0.85)对 3–8 个历史步拟合刚性正则化的二次抛物线:二次曲线物理上无法振荡(零 Runge 现象);权重范数随历史扩展反而衰减(Σ|wᵢ|=0.066 vs 经典 AB 的 0.54),历史变成平滑滤波器且不拖慢推理。

基准(Anima DiT 720×1280,RTX 5060):10 步 flowwls8 达 SSIM 0.8003(零伪影);30 步 0.7541 超过内置 unipc(0.7459);经典 6 阶法 30 步崩为噪声(0.1341);15 步即可获得完美手部/手指解剖。

使用:代码开源并注册 Zenodo DOI(10.5281/zenodo.23337579),ComfyUI 用户把自定义节点放进 customnodes 即可在标准 KSampler 选 flowwls8/5。

原文链接 →

「多模态」频道最新

更多「多模态」频道 AI 资讯 →