arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.19009cs.LGcs.AI

统一生成、重建与表示:具有自适应潜在编码-解码器的广义扩散模型

Unified Generation, Reconstruction, and Representation: Generalized Diffusion with Adaptive Latent Encoding-Decoding

MBZUAI(穆罕默德·本·扎耶德人工智能大学);UC San Diego(加利福尼亚大学圣迭戈分校);University of Tokyo(东京大学);Stanford University(斯坦福大学);CUHK-Shenzhen(香港中文大学(深圳));CMU(卡内基梅隆大学)

Guangyi Liu, Yu Wang, Zeyu Feng, Qiyu Wu, Liping Tang, Yuan Gao, Zhen Li, Shuguang Cui, Julian McAuley, Zichao Yang, Eric P. Xing, Zhiting Hu

AI总结:

针对现有生成模型难以兼顾生成、重建与表示的问题,提出广义编码-解码扩散概率模型(EDDPMs),通过引入参数化编码-解码器推广标准扩散,兼容现有训练方法,在文本、蛋白质和图像上展现出广泛适用性与性能提升。

摘要:

深度生成模型的广泛应用基于三个核心能力——生成新实例、重建输入以及学习紧凑表示——涵盖各种数据类型,例如离散的文本/蛋白质序列和连续的图像。现有的模型家族,如变分自编码器(VAEs)、生成对抗网络(GANs)、自回归模型和(潜在)扩散模型,通常在特定能力和数据类型上表现出色,但在其他方面存在不足。我们引入了广义编码-解码扩散概率模型(EDDPMs),它整合了这些核心能力,以实现广泛的适用性和增强的性能。EDDPMs通过引入参数化的编码-解码器,对标准扩散中的高斯加噪-去噪过程进行了推广。关键的是,EDDPMs与成熟的扩散模型目标和训练方法兼容,允许将编码器-解码器参数与扩散过程联合有效学习。通过选择合适的编码器/解码器(例如大型语言模型),EDDPMs自然适用于不同的数据类型。在文本、蛋白质和图像上的大量实验表明,EDDPMs具备处理多样数据和任务的灵活性,并相较于各种现有模型取得了显著的改进。

英文摘要:

The vast applications of deep generative models are anchored in three core capabilities -- generating new instances, reconstructing inputs, and learning compact representations -- across various data types, such as discrete text/protein sequences and continuous images. Existing model families, like variational autoencoders (VAEs), generative adversarial networks (GANs), autoregressive models, and (latent) diffusion models, generally excel in specific capabilities and data types but fall short in others. We introduce Generalized Encoding-Decoding Diffusion Probabilistic Models (EDDPMs) which integrate the core capabilities for broad applicability and enhanced performance. EDDPMs generalize the Gaussian noising-denoising in standard diffusion by introducing parameterized encoding-decoding. Crucially, EDDPMs are compatible with the well-established diffusion model objective and training recipes, allowing effective learning of the encoder-decoder parameters jointly with diffusion. By choosing appropriate encoder/decoder (e.g., large language models), EDDPMs naturally apply to different data types. Extensive experiments on text, proteins, and images demonstrate the flexibility to handle diverse data and tasks and the strong improvement over various existing models.

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

↑