search
HomeTechnology peripheralsAIStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

Recently, classifier-free guided diffusion models have been very effective in high-resolution image generation and have been widely used in large-scale diffusion frameworks, including DALL-E 2, GLIDE and Imagen.

However, a drawback of classifier-free guided diffusion models is that they are computationally expensive at inference time. Because they require evaluating two diffusion models—a class-conditional model and an unconditional model—hundreds of times.

In order to solve this problem, scholars from Stanford University and Google Brain proposed to use a two-step distillation method to improve the sampling efficiency of the classifier-free guided diffusion model.

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

##Paper address: https://arxiv.org/abs/2210.03142

How to refine the classifier-free guided diffusion model into a fast sampling model?

First, for a pre-trained classifier-free guidance model, the researchers first learned a single model to match the combined output of the conditional model and the unconditional model.

The researchers then gradually distilled this model into a diffusion model with fewer sampling steps.

It can be seen that on ImageNet 64x64 and CIFAR-10, this method is able to generate images that are visually comparable to the original model.

With only 4 sampling steps, FID/IS scores comparable to those of the original model can be obtained, while the sampling speed is as high as 256 times.

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

It can be seen that by changing the guidance weight w, the model distilled by the researcher can make a trade-off between sample diversity and quality. And with just one sampling step, visually pleasing results are achieved.

Background of the Diffusion Model

With samples x from the data distribution

, the noise scheduling functionStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!The researchers trained a model with parameters θ by minimizing the weighted mean square error Diffusion modelStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!. Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

where Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! is the signal-to-noise ratio, Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! and Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! is a pre-specified weighting function.

Once you have trained a diffusion modelStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, you can use the discrete-time DDIM sampler to sample from the model.

Specifically, the DDIM sampler starts from z1 ∼ N (0,I) and is updated as follows

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

Where, N is the total number of sampling steps. Using Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, the final sample is generated.

Classifier-free guidance is an effective method that can significantly improve the sample quality of conditional diffusion models and has been widely used including GLIDE, DALL·E 2 and Imagen.

It introduces a guidance weight parameterStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! to measure the quality and diversity of the sample. To generate samples, classifier-free guidance uses Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! as the prediction model at each update step to evaluate the conditional diffusion model Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! and the jointly trained Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!.

Sampling using classifier-free guidance is generally expensive since two diffusion models need to be evaluated for each sampling update.

In order to solve this problem, the researchers used progressive distillation, which is a method to increase the sampling speed of the diffusion model through repeated distillation.

Previously, this method could not be used directly to guide the distillation of the model, nor could it be used on samplers other than the deterministic DDIM sampler. In this paper, the researchers solved these problems.

Distillation of the guided diffusion model without a classifier

Their approach is to distill the guided diffusion model without a classifier.

For a trained teacher-led modelStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, they took two steps.

In the first step, the researcher introduced a continuous-time student modelStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, which has learnable parameters η1, to match the output of the teacher model at any time step t ∈ [0, 1]. After specifying a range of instructional intensitiesStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! that they were interested in, they used the following objectives to optimize the student model.

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

inStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!.

In order to combine the guidance weight w, the researcher introduced the w conditional model, where w serves as the input of the student model. To better capture the features, they applied Fourier embedding w and then incorporated it into the backbone of the diffusion model using the time-stepping method used by Kingma et al.

Since initialization plays a key role in performance, when the researchers initialized the student model, they used the same parameters as the teacher condition model (except for the newly introduced parameters related to w-conditioning).

The second step, the researcher imagined a discrete time step scenario, and gradually changed the learning model from The first step Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! is distilled into a student model Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! with learnable parameters η2 and fewer steps.

Among them, N represents the number of sampling steps. For Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! and Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, the researcher starts to train the student model, Let it match the output of the teacher model's two-step DDIM sampling with one step (eg: from t/N to t - 0.5/N, from t - 0.5/N to t - 1/N).

After distilling the 2N steps in the teacher model into N steps in the student model, we can use the new N-step student model as the new teacher model, and then repeat the same The process of distilling the teacher model into an N/2-step student model. At each step, the researchers initialize the chemical model using the parameters of the teacher model.

N-step deterministic and random sampling

Once the modelStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! is trained, for Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, the researcher can perform sampling through DDIM update rules. The researcher noticed that for the distillation modelStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, this sampling process is deterministic given the initializationStanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!.

In addition, researchers can also conduct N-step random sampling. Use a deterministic sampling step twice the original step size (i.e., the same as the N/2-step deterministic sampler), and then take a random step back (i.e., perturb it with noise) using the original step size.

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!, when t > 1/N, the following update rules can be used——

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

in,Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!.

When t=1/N, the researcher uses the deterministic update formula to derive Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times! from Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!.

It is worth noting that we note that performing random sampling requires evaluating the model at slightly different time steps compared to the deterministic sampler, and requires edge cases Small modifications to the training algorithm.

Other distillation methods

There is also a method that directly applies progressive distillation to the bootstrap model, That is, following the structure of the teacher model, the student model is directly distilled into a jointly trained conditional and unconditional model. After researchers tried it, they found that this method was not effective.

Experiments and Conclusions

Model experiments were conducted on two standard data sets: ImageNet (64*64) and CIFAR 10.

Different ranges of the guidance weight w were explored in the experiment, and it was observed that all ranges were comparable, so [wmin, wmax] = [0, 4] was used for the experiment. The first- and second-step models are trained using signal-to-noise loss.

Baseline standards include DDPM ancestral sampling and DDIM sampling.

To better understand how to incorporate the guidance weight w, a model trained with a fixed w value is used as a reference.

In order to make a fair comparison, the experiment uses the same pre-trained teacher model for all methods. Using the U-Net (Ronneberger et al., 2015) architecture as the baseline, and using the same U-Net backbone, a structure with w embedded in it is introduced as a two-step student model.

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

The above picture shows the performance of all methods on ImageNet 64x64. where D and S represent deterministic and stochastic samplers respectively.

In the experiment, the model training conditional on the guidance interval w∈[0, 4] was equivalent to the model training with w as a fixed value. When there are fewer steps, our method significantly outperforms the DDIM baseline performance, and basically reaches the performance level of the teacher model at 8 to 16 steps.

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

ImageNet 64x64 sampling quality evaluated by FID and IS scores

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

CIFAR-10 sampling quality evaluated by FID and IS scores

We also distill the encoding process of the teacher model, And conducted experiments on style transfer. Specifically, to perform style transfer between two domains A and B, images from domain A are encoded using a diffusion model trained on domain A, and then decoded using a diffusion model trained on domain B.

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!

Since the encoding process can be understood as an inverted sampling process of DDIM, we distilled both the encoder and decoder with classifier-free guidance and compared it with the DDIM encoder and decoder, as above As shown in the figure. We also explore the performance impact of changes to the boot strength w.

In summary, we propose a distillation method for guided diffusion models, and a random sampler to sample from the distilled model. Empirically, our method achieves visually high-experience sampling in only one step, and obtains FID/IS scores comparable to those of teachers in only 8 to 16 steps.

The above is the detailed content of Stanford/Google Brain: Double distillation, guided diffusion model sampling speed up 256 times!. For more information, please follow other related articles on the PHP Chinese website!

Statement
This article is reproduced at:51CTO.COM. If there is any infringement, please contact admin@php.cn delete
从VAE到扩散模型:一文解读以文生图新范式从VAE到扩散模型:一文解读以文生图新范式Apr 08, 2023 pm 08:41 PM

1 前言在发布DALL·E的15个月后,OpenAI在今年春天带了续作DALL·E 2,以其更加惊艳的效果和丰富的可玩性迅速占领了各大AI社区的头条。近年来,随着生成对抗网络(GAN)、变分自编码器(VAE)、扩散模型(Diffusion models)的出现,深度学习已向世人展现其强大的图像生成能力;加上GPT-3、BERT等NLP模型的成功,人类正逐步打破文本和图像的信息界限。在DALL·E 2中,只需输入简单的文本(prompt),它就可以生成多张1024*1024的高清图像。这些图像甚至

找不到中文语音预训练模型?中文版 Wav2vec 2.0和HuBERT来了找不到中文语音预训练模型?中文版 Wav2vec 2.0和HuBERT来了Apr 08, 2023 pm 06:21 PM

Wav2vec 2.0 [1],HuBERT [2] 和 WavLM [3] 等语音预训练模型,通过在多达上万小时的无标注语音数据(如 Libri-light )上的自监督学习,显著提升了自动语音识别(Automatic Speech Recognition, ASR),语音合成(Text-to-speech, TTS)和语音转换(Voice Conversation,VC)等语音下游任务的性能。然而这些模型都没有公开的中文版本,不便于应用在中文语音研究场景。 WenetSpeech [4] 是

普林斯顿陈丹琦:如何让「大模型」变小普林斯顿陈丹琦:如何让「大模型」变小Apr 08, 2023 pm 04:01 PM

“Making large models smaller”这是很多语言模型研究人员的学术追求,针对大模型昂贵的环境和训练成本,陈丹琦在智源大会青源学术年会上做了题为“Making large models smaller”的特邀报告。报告中重点提及了基于记忆增强的TRIME算法和基于粗细粒度联合剪枝和逐层蒸馏的CofiPruning算法。前者能够在不改变模型结构的基础上兼顾语言模型困惑度和检索速度方面的优势;而后者可以在保证下游任务准确度的同时实现更快的处理速度,具有更小的模型结构。陈丹琦 普

解锁CNN和Transformer正确结合方法,字节跳动提出有效的下一代视觉Transformer解锁CNN和Transformer正确结合方法,字节跳动提出有效的下一代视觉TransformerApr 09, 2023 pm 02:01 PM

由于复杂的注意力机制和模型设计,大多数现有的视觉 Transformer(ViT)在现实的工业部署场景中不能像卷积神经网络(CNN)那样高效地执行。这就带来了一个问题:视觉神经网络能否像 CNN 一样快速推断并像 ViT 一样强大?近期一些工作试图设计 CNN-Transformer 混合架构来解决这个问题,但这些工作的整体性能远不能令人满意。基于此,来自字节跳动的研究者提出了一种能在现实工业场景中有效部署的下一代视觉 Transformer——Next-ViT。从延迟 / 准确性权衡的角度看,

Stable Diffusion XL 现已推出—有什么新功能,你知道吗?Stable Diffusion XL 现已推出—有什么新功能,你知道吗?Apr 07, 2023 pm 11:21 PM

3月27号,Stability AI的创始人兼首席执行官Emad Mostaque在一条推文中宣布,Stable Diffusion XL 现已可用于公开测试。以下是一些事项:“XL”不是这个新的AI模型的官方名称。一旦发布稳定性AI公司的官方公告,名称将会更改。与先前版本相比,图像质量有所提高与先前版本相比,图像生成速度大大加快。示例图像让我们看看新旧AI模型在结果上的差异。Prompt: Luxury sports car with aerodynamic curves, shot in a

五年后AI所需算力超100万倍!十二家机构联合发表88页长文:「智能计算」是解药五年后AI所需算力超100万倍!十二家机构联合发表88页长文:「智能计算」是解药Apr 09, 2023 pm 07:01 PM

人工智能就是一个「拼财力」的行业,如果没有高性能计算设备,别说开发基础模型,就连微调模型都做不到。但如果只靠拼硬件,单靠当前计算性能的发展速度,迟早有一天无法满足日益膨胀的需求,所以还需要配套的软件来协调统筹计算能力,这时候就需要用到「智能计算」技术。最近,来自之江实验室、中国工程院、国防科技大学、浙江大学等多达十二个国内外研究机构共同发表了一篇论文,首次对智能计算领域进行了全面的调研,涵盖了理论基础、智能与计算的技术融合、重要应用、挑战和未来前景。论文链接:​https://spj.scien

​什么是Transformer机器学习模型?​什么是Transformer机器学习模型?Apr 08, 2023 pm 06:31 PM

译者 | 李睿审校 | 孙淑娟​近年来, Transformer 机器学习模型已经成为深度学习和深度神经网络技术进步的主要亮点之一。它主要用于自然语言处理中的高级应用。谷歌正在使用它来增强其搜索引擎结果。OpenAI 使用 Transformer 创建了著名的 GPT-2和 GPT-3模型。自从2017年首次亮相以来,Transformer 架构不断发展并扩展到多种不同的变体,从语言任务扩展到其他领域。它们已被用于时间序列预测。它们是 DeepMind 的蛋白质结构预测模型 AlphaFold

AI模型告诉你,为啥巴西最可能在今年夺冠!曾精准预测前两届冠军AI模型告诉你,为啥巴西最可能在今年夺冠!曾精准预测前两届冠军Apr 09, 2023 pm 01:51 PM

说起2010年南非世界杯的最大网红,一定非「章鱼保罗」莫属!这只位于德国海洋生物中心的神奇章鱼,不仅成功预测了德国队全部七场比赛的结果,还顺利地选出了最终的总冠军西班牙队。不幸的是,保罗已经永远地离开了我们,但它的「遗产」却在人们预测足球比赛结果的尝试中持续存在。在艾伦图灵研究所(The Alan Turing Institute),随着2022年卡塔尔世界杯的持续进行,三位研究员Nick Barlow、Jack Roberts和Ryan Chan决定用一种AI算法预测今年的冠军归属。预测模型图

See all articles

Hot AI Tools

Undresser.AI Undress

Undresser.AI Undress

AI-powered app for creating realistic nude photos

AI Clothes Remover

AI Clothes Remover

Online AI tool for removing clothes from photos.

Undress AI Tool

Undress AI Tool

Undress images for free

Clothoff.io

Clothoff.io

AI clothes remover

AI Hentai Generator

AI Hentai Generator

Generate AI Hentai for free.

Hot Tools

mPDF

mPDF

mPDF is a PHP library that can generate PDF files from UTF-8 encoded HTML. The original author, Ian Back, wrote mPDF to output PDF files "on the fly" from his website and handle different languages. It is slower than original scripts like HTML2FPDF and produces larger files when using Unicode fonts, but supports CSS styles etc. and has a lot of enhancements. Supports almost all languages, including RTL (Arabic and Hebrew) and CJK (Chinese, Japanese and Korean). Supports nested block-level elements (such as P, DIV),

Notepad++7.3.1

Notepad++7.3.1

Easy-to-use and free code editor

SAP NetWeaver Server Adapter for Eclipse

SAP NetWeaver Server Adapter for Eclipse

Integrate Eclipse with SAP NetWeaver application server.

VSCode Windows 64-bit Download

VSCode Windows 64-bit Download

A free and powerful IDE editor launched by Microsoft

DVWA

DVWA

Damn Vulnerable Web App (DVWA) is a PHP/MySQL web application that is very vulnerable. Its main goals are to be an aid for security professionals to test their skills and tools in a legal environment, to help web developers better understand the process of securing web applications, and to help teachers/students teach/learn in a classroom environment Web application security. The goal of DVWA is to practice some of the most common web vulnerabilities through a simple and straightforward interface, with varying degrees of difficulty. Please note that this software