arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.04641cs.LGmath.STstat.MLstat.TH
收藏

对比预训练与多模态生成式AI的统计理论

A Statistical Theory of Contrastive Pre-training and Multimodal Generative AI

UC Berkeley(加州大学伯克利分校)

Kazusato Oko, Licong Lin, Yuhang Cai, Song Mei

AI总结:

本文提出近似充分统计量与联合生成层次模型,从统计理论上解释对比预训练表示为何适配多模态下游任务,并给出样本复杂度保证。

摘要:

多模态生成式AI系统(例如结合视觉与语言的系统)依赖对比预训练来学习跨不同模态的表示。尽管其实践收益已得到广泛认可,但对对比预训练框架的严格理论理解仍然有限。本文建立了一个理论框架,用于解释对比预训练在零样本分类、条件扩散模型和视觉-语言模型等下游任务中取得成功的原因。我们引入近似充分统计量这一概念,它是经典充分统计量的推广,并证明对比预训练损失的近极小值点具有近似充分性,因此能够适应多种下游任务。我们进一步提出用于图像与文本联合分布的联合生成层次模型(Joint Generative Hierarchical Model),表明 transformers 可以通过信念传播高效近似该模型中的相关函数。基于这一框架,我们推导了利用对比预训练表示进行多模态学习的样本复杂度保证。数值模拟验证了这些理论发现,展示了经对比预训练的 transformers 在各类多模态任务中的强泛化性能。

英文摘要:

Multi-modal generative AI systems, such as those combining vision and language, rely on contrastive pre-training to learn representations across different modalities. While their practical benefits are widely acknowledged, a rigorous theoretical understanding of the contrastive pre-training framework remains limited. This paper develops a theoretical framework to explain the success of contrastive pre-training in downstream tasks, such as zero-shot classification, conditional diffusion models, and vision-language models. We introduce the concept of approximate sufficient statistics, a generalization of the classical sufficient statistics, and show that near-minimizers of the contrastive pre-training loss are approximately sufficient, making them adaptable to diverse downstream tasks. We further propose the Joint Generative Hierarchical Model for the joint distribution of images and text, showing that transformers can efficiently approximate relevant functions within this model via belief propagation. Building on this framework, we derive sample complexity guarantees for multi-modal learning based on contrastive pre-trained representations. Numerical simulations validate these theoretical findings, demonstrating the strong generalization performance of contrastively pre-trained transformers in various multi-modal tasks.

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

↑