arXiv cs.CV· Yuyao Zhang, Yuwei Hu, Ziyang Mai, Yu-Wing Tai·· 4 小时前AI 评分36
JAx：对齐预测而非表征，改进像素空间扩散模型训练
Just Align $\bm{x}$: Aligning Predictions, Not Representations
AI 导读
JAx（Just Align x）不对齐表征而只对齐干净图像预测，在 ImageNet 256x256 上改善了 JiT-B/16、L/16、H/16 的 FID 并加速收敛。它用马尔可夫退化把噪声更多的学生观测与更干净的观测耦合，无需外部编码器或改动架构与采样流程。梯度诊断显示 minibatch 梯度方差降低，消融表明收益非仅来自时间重加权。
来源：arXiv cs.CV · arxiv.org