arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.05196cs.LGcs.CV

去噪自回归表示学习

Denoising Autoregressive Representation Learning

Google DeepMind(谷歌DeepMind);xAI

Yazhe Li, Jorg Bornschein, Ting Chen

AI总结:

本文提出DARL,一种采用仅解码器Transformer自回归预测图像块的去噪自回归表示学习方法,通过结合MSE与扩散目标,在微调下性能接近最先进掩码预测模型,为视觉感知与生成统一模型迈出重要一步。

摘要:

本文探索了一种学习视觉表示的新生成方法。我们的方法DARL采用仅解码器Transformer自回归地预测图像块。我们发现仅使用均方误差(MSE)训练即可产生强大的表示。为增强图像生成能力,我们通过使用去噪图像块解码器,用扩散目标替换了MSE损失。我们表明,通过使用定制的噪声调度和在更大模型中更长时间的训练,可以改进学习到的表示。值得注意的是,最佳调度与标准图像扩散模型中使用的典型调度显著不同。总体而言,尽管架构简单,DARL在微调协议下的性能非常接近最先进的掩码预测模型。这标志着向能够同时进行视觉感知和生成的统一模型迈出了重要一步,有效结合了自回归和去噪扩散模型的优势。

英文摘要:

In this paper, we explore a new generative approach for learning visual representations. Our method, DARL, employs a decoder-only Transformer to predict image patches autoregressively. We find that training with Mean Squared Error (MSE) alone leads to strong representations. To enhance the image generation ability, we replace the MSE loss with the diffusion objective by using a denoising patch decoder. We show that the learned representation can be improved by using tailored noise schedules and longer training in larger models. Notably, the optimal schedule differs significantly from the typical ones used in standard image diffusion models. Overall, despite its simple architecture, DARL delivers performance remarkably close to state-of-the-art masked prediction models under the fine-tuning protocol. This marks an important step towards a unified model capable of both visual perception and generation, effectively combining the strengths of autoregressive and denoising diffusion models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑