arXiv:2502.17157cs.CV2025-02NeurIPS被引 41

用极少数据训练出能完成多种视觉任务的通用模型,性能媲美专用模型。

DICEPTION: A Generalist Diffusion Model for Visual Perceptual Tasks

  • 复用预训练扩散模型,保留先验知识实现多任务泛化。
  • 仅用60万标注图像即达SAM-vit-h同等效果,数据量为1%。
  • 新任务微调只需50张图,参数更新不到1%,高效实用。

本文旨在开发一个在计算资源与训练数据有限条件下仍具鲁棒性的通用感知模型。我们利用在数十亿图像上预训练的文本到图像扩散模型,成功构建了DICEPTION——一种视觉通用模型。大量实验表明,DICEPTION能有效应对多种感知任务,性能甚至可媲美当前最优的单任务专用模型。具体而言,仅使用其0.06%的数据(如60万对像素级标注图像,相较SAM-vit-h的10亿像素),即可达到相近表现。通过系统性实验验证,将单一扩散模型适配至多任务的关键在于最大限度保留预训练模型的先验知识。因此,DICEPTION的训练成本远低于从头训练的传统模型。此外,适应新任务极为高效,仅需50张图像和约1%的参数微调即可。最后,我们发现适度应用无分类器引导可提升深度与法向估计性能;同时,像素对齐训练显著增强了模型保持细节的能力。DICEPTION为基于扩散模型的视觉通用模型发展提供了重要启示。代码与模型:https://github.com/aim-uofa/Diception

原文摘要 · Abstract (English)

This paper's primary objective is to develop a robust generalist perception model capable of addressing multiple tasks under constraints of computational resources and limited training data. We leverage text-to-image diffusion models pre-trained on billions of images and successfully introduce our DICEPTION, a visual generalist model. Exhaustive evaluations demonstrate that DICEPTION effectively tackles diverse perception tasks, even achieving performance comparable to SOTA single-task specialist models. Specifically, we achieve results on par with SAM-vit-h using only 0.06% of their data (e.g., 600K vs.\ 1B pixel-level annotated images). We designed comprehensive experiments on architectures and input paradigms, demonstrating that the key to successfully re-purposing a single diffusion model for multiple perception tasks lies in maximizing the preservation of the pre-trained model's prior knowledge. Consequently, DICEPTION can be trained with substantially lower computational costs than conventional models requiring training from scratch. Furthermore, adapting DICEPTION to novel tasks is highly efficient, necessitating fine-tuning on as few as 50 images and approximately 1% of its parameters. Finally, we demonstrate that a subtle application of classifier-free guidance can improve the model's performance on depth and normal estimation. We also show that pixel-aligned training, as is characteristic of perception tasks, significantly enhances the model's ability to preserve fine details. DICEPTION offers valuable insights and presents a promising direction for the development of advanced diffusion-based visual generalist models. Code and Model: https://github.com/aim-uofa/Diception

扩散模型通用模型少样本学习视觉感知

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。