arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.00503cs.CV
收藏

迈向基于扩散的稠密视觉任务通用模型

Toward a Diffusion-Based Generalist for Dense Vision Tasks

Max Planck Institute for Informatics(马克斯·普朗克信息学研究所);Google(谷歌公司);Google DeepMind(谷歌DeepMind);ETH Zurich(苏黎世联邦理工学院);TU Munich(慕尼黑工业大学)

Yue Fan, Yongqin Xian, Xiaohua Zhai, Alexander Kolesnikov, Muhammad Ferjad Naeem, Bernt Schiele, Federico Tombari

AI总结:

本文提出在像素空间进行扩散并微调预训练文本到图像扩散模型,将多种稠密预测任务统一为条件图像生成,在四类任务上取得与其他视觉通用模型相当的性能。

摘要:

构建能够同时解决多种计算机视觉任务的通用模型是一个引人关注的方向。近期研究表明,图像本身可以作为通用视觉感知的自然接口,并展示了令人鼓舞的结果。本文探索基于扩散的视觉通用模型,将不同类型的稠密预测任务统一为条件图像生成,并为此重新利用预训练的扩散模型。然而,直接应用现成的潜在扩散模型会导致量化问题。因此,我们提出在像素空间中进行扩散,并提供了一种针对稠密视觉任务微调预训练文本到图像扩散模型的方案。在实验中,我们在四种不同类型的任务上评估了所提方法,并展示了与其他视觉通用模型相比具有竞争力的性能。

英文摘要:

Building generalized models that can solve many computer vision tasks simultaneously is an intriguing direction. Recent works have shown image itself can be used as a natural interface for general-purpose visual perception and demonstrated inspiring results. In this paper, we explore diffusion-based vision generalists, where we unify different types of dense prediction tasks as conditional image generation and re-purpose pre-trained diffusion models for it. However, directly applying off-the-shelf latent diffusion models leads to a quantization issue. Thus, we propose to perform diffusion in pixel space and provide a recipe for finetuning pre-trained text-to-image diffusion models for dense vision tasks. In experiments, we evaluate our method on four different types of tasks and show competitive performance to the other vision generalists.

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

↑