小哲的博客

本文介绍了 ControlNet，这是一种端到端的神经网络架构，用于学习大型预训练文本到图像扩散模型（在我们的实现中为 Stable Diffusion）的条件控制。 **ControlNet 通过锁定大型预训练模型的参数并复制其编码层，保留了该大型模型的质量和能力。这种架构将大型预训练模型视为学习各种条件控制的强大主干网络。**可训练的副本和原始锁定的模型通过零卷积层连接，权重初始化为零，以便在训练过程中逐渐增长。这种架构确保在训练开始时不会向大型扩散模型的深层特征添加有害噪声，并保护可训练副本中的大型预

DDIM论文提出了一种使过程非马尔可夫的方法（如右图所示），允许跳过去噪过程中的步骤，而不需要在当前状态之前访问所有过去的状态。DDIM最好的部分是，它们可以在训练模型后应用，因此DDPM模型可以很容易地转换为DDIM，而无需重新训练新模型。

在这篇文章中，我们将手把手展示如何使用Hugging Face的diffusers包通过文本生成图像。

为了更好的理解扩散模型，我们尝试从零开始搭建它。从一个简单的扩散模型开始，理解其不同部分的工作院里，并对比它们与更复杂的结构之间的不同。首先，我们将回答四个问题：①什么是退化过程（如何向数据添加噪声？），②什么是UNet模型以及③如何从零开始实现一个简单的UNet模型，④如何进行扩散模型的训练以及相关的采样理论；然后，我们将介绍UNet模型的一种改进方法、以及当前流行的DDPM噪声特点、并给出训练目标的差异以及调节时间步和采样方法。

本文介绍一种给扩散模型添加额外条件信息的方法。具体地，将在MNIST数据集上训练一个以类别为条件的扩散模型。并且可以在推理阶段指定想要生成的是哪个数字。

`微调`（Fine-tuning）指的是在预先训练好的模型上进行进一步训练，以适应特定任务或领域的过程。这个过程在机器学习和深度学习领域中常常被使用。**通过微调，可以使用先前在大规模数据集上训练好的模型来学习新数据的特定特征，以完成特定的任务**。微调的关键在于利用已经学到的通用特征，通过在特定任务数据集上进行少量训练，使模型适应新的任务。`引导`（Guidance）通常指的是**在某种领域或任务中，给予模型特定的指导、提示或信息，以帮助模型更好地学习和完成特定任务**。这可能包括提供额外的信息、特定

Diffusers是用于生成图像，音频甚至分子3D结构的最先进的扩散模型的首选库。无论是寻找简单的推理解决方案还是训练自己的扩散模型，Diffusers都是一个支持两者的模块化工具箱。该library 的设计侧重于性能，简单的简单性和对抽象的可定制性。

Stable Diffusion是一个强大的文本条件隐式扩散模型（text-conditioned latent diffusion model），它具有根据文字描述生成精美图片的能力。它不仅是一个完全开源的模型（代码，数据，模型全部开源），而且是它的参数量只有`1B`左右，大部分人可以在普通的显卡上进行推理甚至精调模型。毫不夸张的说，Stable Diffusion的出现和开源对AIGC的火热和发展是有巨大推动作用的，因为它让更多的人能快地上手AI作画。本文将基于Hugging Face的diffuser