GPT-4o掀起全模态热潮!一文梳理全模态大模型最新研究进展
科技
2024-12-22 00:01
吉林
MLNLP社区是国内外知名的机器学习与自然语言处理社区,受众覆盖国内外NLP硕博生、高校老师以及企业研究人员。
社区的愿景是促进国内外自然语言处理,机器学习学术界、产业界和广大爱好者之间的交流和进步,特别是初学者同学们的进步。随着 Gemini 和 GPT-4o 等系列的发布,本文尝试总结一些最近全模态相关的工作。采用联合文本、图像、语音和视频进行训练,用一个端到端的模型可以支持文本、语音和图像三个模态,在支持图生文的同时也支持文生图。Emu
Emu: Generative Pretraining in Multimodalityhttps://arxiv.org/abs/2307.05222https://github.com/baaivision/Emua. Visual Encoder:先用 EVA-CLIP 进行图像编码,然后通过 Causal Transformer 进行 2D 信息建模,query 是自定义的向量,key 和 value 来自于视觉编码。b. LLM 阶段训练:用 LLM 进行 next-token 学习,文本 Token 算交叉熵,图像 Token 算 L2 loss。c. Visual Decoder:采用 Stable Diffusion 进行初始化,将 LLM 输出的 visual embeddings 作为 conditions 输出到 Stable Diffusion,将 Stable Diffusion 的 cross-attention 的 linear projections 维度换成适配 visual embeddings 的。d. Visual Decoder 阶段训练:只训练 U-Net,其他参数不变。NExT-GPT
NExT-GPT: Any-to-Any Multimodal LLMhttps://arxiv.org/abs/2309.05519https://next-gpt.github.io/a. 多模态编码阶段:利用用 ImageBind 的编码器,loss 是各个模态的 caption。b. 多模解码阶段:主要目的是使扩散模型与 LLM 的输出指令保持一致,最小化 LLM 的模态信号 token 表示与扩散模型(图像合成的 Stable Diffusion、视频合成的 Zeroscope 和音频合成的 AudioLDM)的条件文本表示之间的距离,确保了高度轻量级的训练。遵循和理解用户指令并生成所需多模态输出,只训练不同模态的 Linear 和 Lora,同时增加扩散的 Loss。Emu2
Generative Multimodal Models are In-Context Learnershttps://arxiv.org/abs/2312.13286https://baaivision.github.io/emu2a. Visual Encoder 使用 EVA-02-CLIP-E-plus,LLM 使用 LLaMA-33B,Visual Decoder 使用 SDXL。b. 一阶段:在图像和视频 pair 对上进行 caption 学习。c. 二阶段:Freeze Visual Encoder,使用 text classification loss and image regression loss。d. 三阶段:训练 Visual Decoder,和 Emu1 区别是训练的时候不用 LLM 输出的 visual token,直接用 Visual Encoder 的输出,还是只训练 U-Net,其他参数不变。LWM
World Model on Million-Length Video And Language With Blockwise RingAttentionhttps://arxiv.org/abs/2402.08268https://largeworldmodel.github.io/lwm/a. Visual Encoder:VQGAN 做离散化,256*256 分辨率 -> 256 tokens。b. 训练:next-token 预测,主要为了解决长序列使用了 RingAttention, Blockwise Transformers, masked sequence packing。Chameleon
Chameleon: Mixed-Modal Early-Fusion Foundation Modelshttps://arxiv.org/abs/2405.09818https://github.com/facebookresearch/chameleona. Visual Encoder:VQ-VAE 进行图像 token 离散化,512*512->1024 tokens,词表 8192。SEED
Planting a SEED of Vision in Large Language Modelhttps://arxiv.org/abs/2307.08041https://github.com/AILab-CVC/SEEDa. 训练 tokenizer:先训练 Causal Q-Former,使用 SD 的 Text Encoder 进行对比 Loss。然后进行 Visual Quantization and De-tokenization 学习,将 Genearation Embedding 和 SD 的 Text Encoder 的输入靠近,其中 ViT Encoder 和 SD Decoder 参数不参与学习。b. LLM 阶段训练:进行 next token 预测Transfusion
Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Modelhttps://arxiv.org/abs/2408.11039https://github.com/lucidrains/transfusion-pytorcha. 利用 VAE encode 和 decode 图像,Linear 换成 U-netb. 文本做自回归,图像做自编码,利用 transfomer 实现扩散,进行多步生。Show-o
Show-o: One Single Transformer to Unify Multimodal Understanding and Generationhttps://arxiv.org/abs/2408.12528https://github.com/showlab/Show-oa. 整体思路和 Transfusion 类似,区别是尝试了 discrete diffusion(比如 MaskGIT 和 Muse,其中 Muse 多了一个超分辨率模型学习高层次的语义,不能说明 continuous 比 discrete 的要好,discrete image tokenizer 没有像 clip 那样用超大规模图文对训练),图像用 MAGVIT-v2 进行离散化,256*256 分辨率->256 tokens。b. 图像生成部分利用 MaskGIT(自回归解码的方法里面,生成后面的 token 时候需要前面所有的 tokens 的信息,这个过程不能够并行,图像改成双向注意力可以并行。最近的 DeepMind 的 FLUID 消融实验 Random Order 预测 + Token 连续方式效果好一些)方式,使用 mask 方式多步生成图像,具体在是每次迭代中,模型同时预测所有 tokens,但只保留置信度最高的 tokens,剩余的 tokens 在下一次迭代中重新预测,掩码的比例逐渐减少,直到所有 tokens 在多次迭代内生成。Emu3
Emu3: Next-Token Prediction is All You Needhttps://arxiv.org/abs/2409.18869a. 图像基于SBER-MoVQGAN训练一个离散化tokenzier,512*512->4096 tokens,词表大小 32768。b. Next-Token 预测,可能是因为生成和理解相互影响,最后分成了两个模型(理解、生成)。总结
近期多模态大模型发展迅速,目前主流还是图生文和文生图分开,图生文采用 VIT+LLM 的方式,文生图采用 Duffsion 的方式。随着全模态大模型的发展,也大概能分成 token 离散化、AR+diffusion 融合和将 LLM 输出的 Visual Embeddings 作为 Stable Diffusion 的 Text Conditions 的方式。目前每种方式都各有利弊,都是还处在前期阶段,但是相信全模态大模型在未来应该是一个值得探索的方向,实现端到端的多模态训练、集成全模态输入输出和强大的多模态理解与生成能力。扫描二维码添加小助手微信
即可申请加入自然语言处理/Pytorch等技术交流群关于我们
MLNLP 社区是由国内外机器学习与自然语言处理学者联合构建的民间学术社区,目前已经发展为国内外知名的机器学习与自然语言处理社区,旨在促进机器学习,自然语言处理学术界、产业界和广大爱好者之间的进步。社区可以为相关从业者的深造、就业及研究等方面提供开放交流平台。欢迎大家关注和加入我们。