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

统一对比学习与生成目标以实现视觉理解和文本到图像生成

Unifying Contrastive and Generative Objectives for Visual Understanding and Text-to-Image Generation

MIT Computer Science \& Artificial Intelligence Laboratory(Meta AI);Meta AI
Chao Li, Tianhong Li, Sai Vidyaranya Nuthalapati, Hong-You Chen, Satya Narayan Shukla, Jianpeng Cheng, Yonghuan Yang, Jun Xiao, Xiangjun Fan, Aashu Singh, Dina … 展开作者

Chao Li, Tianhong Li, Sai Vidyaranya Nuthalapati, Hong-You Chen, Satya Narayan Shukla, Jianpeng Cheng, Yonghuan Yang, Jun Xiao, Xiangjun Fan, Aashu Singh, Dina Katabi, Shlok Kumar Mishra

AI总结:

本文提出DREAM框架,通过Masking Warmup解决对比学习与文本到图像生成的矛盾,提升模型在多个任务上的性能。

摘要:

将文本-图像对比学习与文本到图像生成统一到一个端到端模型具有挑战性,因为两者需要不同的掩码策略:对比学习需要近完全可见的token,而掩码生成模型需要大量干扰。我们引入DREAM框架,通过Masking Warmup调度,在训练过程中逐步调整掩码分布的中心,使低和高掩码比率同时存在。这种共暴露使一个联合训练的编码器能够服务于两种目标。所得到的稳定优化解锁了语义对齐解码:在推理阶段,经过所有掩码比率训练的文本编码器可以评估部分生成的图像并选择最佳轨迹,仅需解码图像的12.5%,从而提高FID和吞吐量。DREAM在ImageNet线性探测(+1.1%)、5次转移(+4.1%)、ADE20K分割(+1.9%)和NYU深度估计(+6.25%)上优于CLIP,在CC12M FID上优于FLUID(+6.2%)的同时保持CLIP Score。这些收益表明,当正确统一文本-图像对比和生成目标时,它们是协同作用而非竞争。

英文摘要:

Unifying text-image contrastive learning and text-to-image (T2I) generation in a single end-to-end model is challenging because the two objectives demand opposing masking regimes: contrastive alignment needs near-complete visible tokens, while masked generative modeling needs heavy corruption. We introduce DREAM, a unified framework that resolves this conflict through Masking Warmup, a schedule that shifts the center of the masking distribution over training, so low and high masking ratios coexist at every step. This co-exposure lets a single jointly-trained encoder serve both objectives. The resulting stable optimization unlocks Semantically Aligned Decoding at inference: the text encoder, trained against visual embeddings at all masking ratios, can score partially generated images and select the best trajectory with as little as 12.5% of the image decoded, improving both FID and throughput. DREAM outperforms its single-objective baselines, CLIP and FLUID: on ImageNet linear-probing (+1.1%), 5-shot transfer (+4.1%), ADE20K segmentation (+1.9%), and NYU depth estimation (+6.25%) over CLIP, and on CC12M FID (+6.2%) over FLUID while maintaining CLIP Score. Together, these gains show that text-image contrastive and generative objectives, when properly unified, are synergistic rather than competing.

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

↑