用扩散 Transformer 统一生成20+视觉任务,无需微调即可跨任务泛化。
LaVin-DiT: Large Vision Diffusion Transformer
- 引入时空变分自编码器将图像转为连续潜空间。
- 设计联合扩散 Transformer 逐步生成视觉输出,提升效率与质量。
- 支持上下文学习,测试时仅需任务提示即可适应新任务。
本文提出大规模视觉扩散 Transformer(LaVin-DiT),一个可扩展的统一基础模型,用于在生成框架下解决超过20项计算机视觉任务。不同于直接沿用自然语言处理架构的大型视觉模型,后者依赖低效的自回归方法并破坏视觉数据的空间关系,LaVin-DiT通过三项创新优化生成性能:首先,采用时空变分自编码器将高维视觉数据编码至连续潜空间;其次,设计联合扩散变压器,逐步生成视觉输出;第三,实现上下文学习进行统一多任务训练,输入-目标对作为任务上下文,引导扩散变压器在潜空间中对齐输出以匹配特定任务。推理时,使用任务特定上下文集和测试数据作为查询,实现无需微调的跨任务泛化。模型参数规模从0.1B扩展至3.4B,基于大量视觉数据训练,在多种视觉任务上达到领先性能。该工作为大型视觉基础模型提供了新路径,凸显扩散变压器的巨大潜力。代码与模型已公开。
原文摘要 · Abstract (English)
This paper presents the Large Vision Diffusion Transformer (LaVin-DiT), a scalable and unified foundation model designed to tackle over 20 computer vision tasks in a generative framework. Unlike existing large vision models directly adapted from natural language processing architectures, which rely on less efficient autoregressive techniques and disrupt spatial relationships essential for vision data, LaVin-DiT introduces key innovations to optimize generative performance for vision tasks. First, to address the high dimensionality of visual data, we incorporate a spatial-temporal variational autoencoder that encodes data into a continuous latent space. Second, for generative modeling, we develop a joint diffusion transformer that progressively produces vision outputs. Third, for unified multi-task training, in-context learning is implemented. Input-target pairs serve as task context, which guides the diffusion transformer to align outputs with specific tasks within the latent space. During inference, a task-specific context set and test data as queries allow LaVin-DiT to generalize across tasks without fine-tuning. Trained on extensive vision datasets, the model is scaled from 0.1B to 3.4B parameters, demonstrating substantial scalability and state-of-the-art performance across diverse vision tasks. This work introduces a novel pathway for large vision foundation models, underscoring the promising potential of diffusion transformers. The code and models are available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。