arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.01255cs.CVcs.LG

Meta ControlNet:通过元学习增强任务适应性

Meta ControlNet: Enhancing Task Adaptation via Meta Learning

Ohio State University(俄亥俄州立大学);University of Texas at Austin(德克萨斯大学奥斯汀分校)

Junjie Yang, Jinze Zhao, Peihao Wang, Zhangyang Wang, Yingbin Liang

AI总结:

本文提出Meta ControlNet,通过元学习与层冻结设计,将控制能力学习步数从5000降至1000,实现边缘任务零样本适应,并在非边缘任务中仅需100步微调超越现有方法。

摘要:

基于扩散模型的图像合成近期引起了广泛关注。特别是使用基于图像提示的ControlNet,在canny边缘检测等图像任务中展现出强大能力,并生成与这些提示良好对齐的图像。然而,原始的ControlNet通常需要约5000步的大量训练才能在单一任务上实现理想的控制。近期的上下文学习方法提升了其适应性,但主要针对基于边缘的任务,且依赖成对样本。因此,要充分发挥ControlNet的潜力,还有两个重要的开放问题有待解决:(i) 某些任务的零样本控制,以及(ii) 非边缘任务的更快适应。本文提出了一种新颖的Meta ControlNet方法,该方法采用任务无关的元学习技术,并具有新的层冻结设计。Meta ControlNet将获得控制能力的学习步数从5000显著减少至1000。此外,Meta ControlNet在基于边缘的任务中无需任何微调即可展现直接的零样本适应性,并在Human Pose等更复杂的非边缘任务中仅需100步微调即可实现控制,超越了所有现有方法。代码可在https://github.com/JunjieYang97/Meta-ControlNet获取。

英文摘要:

Diffusion-based image synthesis has attracted extensive attention recently. In particular, ControlNet that uses image-based prompts exhibits powerful capability in image tasks such as canny edge detection and generates images well aligned with these prompts. However, vanilla ControlNet generally requires extensive training of around 5000 steps to achieve a desirable control for a single task. Recent context-learning approaches have improved its adaptability, but mainly for edge-based tasks, and rely on paired examples. Thus, two important open issues are yet to be addressed to reach the full potential of ControlNet: (i) zero-shot control for certain tasks and (ii) faster adaptation for non-edge-based tasks. In this paper, we introduce a novel Meta ControlNet method, which adopts the task-agnostic meta learning technique and features a new layer freezing design. Meta ControlNet significantly reduces learning steps to attain control ability from 5000 to 1000. Further, Meta ControlNet exhibits direct zero-shot adaptability in edge-based tasks without any finetuning, and achieves control within only 100 finetuning steps in more complex non-edge tasks such as Human Pose, outperforming all existing methods. The codes is available in https://github.com/JunjieYang97/Meta-ControlNet.

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

↑