Chinese contentRiT 架构用原始输入旁路修复像素空间 Flow Matching 的 v-prediction 失效问题
做扩散或 Flow Matching 训练的工程和研究同学值得看这篇。它用 patch-wise PCA 给出了 v-prediction 失效的机理,提出的 RiT 不用改成 x-prediction 也能跑通像素空间训练,在 ImageNet 256×256 上验证有效。
做扩散或 Flow Matching 训练的工程和研究同学值得看这篇。它用 patch-wise PCA 给出了 v-prediction 失效的机理,提出的 RiT 不用改成 x-prediction 也能跑通像素空间训练,在 ImageNet 256×256 上验证有效。