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

4M-21:面向数十种任务与模态的任意到任意视觉模型

4M-21: An Any-to-Any Vision Model for Tens of Tasks and Modalities

Swiss Federal Institute of Technology Lausanne (EPFL)(洛桑联邦理工学院);Apple(苹果公司)

Roman Bachmann, Oğuzhan Fatih Kar, David Mizrahi, Ali Garjani, Mingfei Gao, David Griffiths, Jiaming Hu, Afshin Dehghan, Amir Zamir

AI总结:

本文提出4M-21,通过在数十种模态上进行离散分词与协同训练,构建了一个能处理至少比现有模型多3倍任务/模态的任意到任意视觉基础模型,并在不损失性能的前提下实现了三亿参数规模的训练与开源。

摘要:

当前的多模态和多任务基础模型(如4M或UnifiedIO)展现出有前景的结果,但在实际应用中,它们开箱即用地接受多样化输入并执行多样化任务的能力,受到其训练时所使用的(通常相当少的)模态和任务数量的限制。在本文中,我们通过在一个单一模型上训练数十种高度多样化的模态,并在大规模多模态数据集和文本语料库上进行协同训练,来扩展这些模型的能力。这包括在若干语义和几何模态、来自近期最先进模型(如DINOv2和ImageBind)的特征图、专家模型(如SAM和4DHumans)的伪标签,以及一系列允许以新颖方式与模型交互并引导生成的新模态(例如图像元数据或调色板)上进行训练。这一过程中的一个关键步骤是对各种模态执行离散分词,无论它们是类图像数据、神经网络特征图、向量、结构化数据(如实例分割或人体姿态),还是可以表示为文本的数据。通过这种方式,我们扩展了多模态模型的开箱即用能力,并特别展示了训练一个模型以解决至少比现有模型多3倍的任务/模态,且性能不下降的可能性。这使得更细粒度和可控的多模态生成能力成为可能,并使我们能够研究将在多样化数据和目标上训练的模型蒸馏到一个统一模型中的过程。我们成功地将训练扩展到使用数十种模态和不同数据集的三亿参数模型。所得模型和训练代码已在4m.epfl.ch开源。

英文摘要:

Current multimodal and multitask foundation models like 4M or UnifiedIO show promising results, but in practice their out-of-the-box abilities to accept diverse inputs and perform diverse tasks are limited by the (usually rather small) number of modalities and tasks they are trained on. In this paper, we expand upon the capabilities of them by training a single model on tens of highly diverse modalities and by performing co-training on large-scale multimodal datasets and text corpora. This includes training on several semantic and geometric modalities, feature maps from recent state of the art models like DINOv2 and ImageBind, pseudo labels of specialist models like SAM and 4DHumans, and a range of new modalities that allow for novel ways to interact with the model and steer the generation, for example image metadata or color palettes. A crucial step in this process is performing discrete tokenization on various modalities, whether they are image-like, neural network feature maps, vectors, structured data like instance segmentation or human poses, or data that can be represented as text. Through this, we expand on the out-of-the-box capabilities of multimodal models and specifically show the possibility of training one model to solve at least 3x more tasks/modalities than existing ones and doing so without a loss in performance. This enables more fine-grained and controllable multimodal generation capabilities and allows us to study the distillation of models trained on diverse data and objectives into a unified model. We successfully scale the training to a three billion parameter model using tens of modalities and different datasets. The resulting models and training code are open sourced at 4m.epfl.ch.

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

↑