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

通过迭代共识合成预训练模型的集成

Composing Ensembles of Pre-trained Models via Iterative Consensus

Google Brain(谷歌大脑)

Shuang Li, Yilun Du, Joshua B. Tenenbaum, Antonio Torralba, Igor Mordatch

摘要:

大型的预训练模型根据训练数据表现出不同的互补能力。语言模型(如GPT-3)能够进行文本推理,但无法理解视觉信息,而视觉模型(如DALL-E)可以生成逼真的照片,但无法理解复杂的语言描述。在这项工作中,我们提出了一个统一的框架,用于组成不同预训练模型的集合-结合每个模型的优势,以zero-shot方式解决各种多模态问题。我们使用预先训练的模型作为“生成器”或“评分器”,并通过闭环迭代共识优化来组合它们。生成器构造建议,评分器迭代地提供反馈以细化生成的结果。这种闭环通信使模型能够纠正由其他模型引起的错误,显著提高下游任务的性能,例如将小学数学问题的准确性提高7.5%,而无需任何模型微调。我们证明,通过利用每个专家模型的优势,由一个整体的评分者达成的共识优于单个评分者的反馈。结果表明,所提出的方法可以作为一个通用的框架,广泛的zero-shot多模态任务,如图像生成,视频问答,数学推理和机器人操作。项目页面:https://energy-based-model.github.io/composing-pretrained-models。

英文摘要:

Large pre-trained models exhibit distinct and complementary capabilities dependent on the data they are trained on. Language models such as GPT-3 are capable of textual reasoning but cannot understand visual information, while vision models such as DALL-E can generate photorealistic photos but fail to understand complex language descriptions. In this work, we propose a unified framework for composing ensembles of different pre-trained models -- combining the strengths of each individual model to solve various multimodal problems in a zero-shot manner. We use pre-trained models as "generators" or "scorers" and compose them via closed-loop iterative consensus optimization. The generator constructs proposals and the scorers iteratively provide feedback to refine the generated result. Such closed-loop communication enables models to correct errors caused by other models, significantly boosting performance on downstream tasks, e.g. improving accuracy on grade school math problems by 7.5%, without requiring any model finetuning. We demonstrate that consensus achieved by an ensemble of scorers outperforms the feedback of a single scorer, by leveraging the strengths of each expert model. Results show that the proposed method can be used as a general purpose framework for a wide range of zero-shot multimodal tasks, such as image generation, video question answering, mathematical reasoning, and robotic manipulation. Project page: https://energy-based-model.github.io/composing-pretrained-models.

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

↑