论文阅读笔记2——DARTS：Differentiable Architecture Search可微分架构搜索（二）（论文翻译学习）

加油加油！如果你感觉你现在很累，那么恭喜你，你现在正在走上坡路！让我们一起加油！

欢迎关注我的讲解视频，让我们一起学习：

Bilibili主页: https://space.bilibili.com/481802918

DARTS：Differentiable Architecture Search（二）

Differentiable Architecture Search

2.2 持续放松和优化

让$\mathcal{O}$为一组候选运算（例如：卷积，最大池化或者零操作），其中每个运算代表要应用于x⁽ⁱ⁾的某个函数o。为了使搜索空间连续，我们将特定操作的分类选择放宽到所有可能操作的softmax:

$${\bar{o}}^{(i,j)}(x)=\sum _{o\in \mathcal{O}}\frac{\exp \left({\alpha }_o^{(i,j)}\right)}{{\sum }_{o^{\prime }\in \mathcal{O}}\exp \left({\alpha }_{o^{\prime }}^{(i,j)}\right)}o(x)\text{\hspace{1em}(2)}$$
其中一对节点 (i,j) 的运算混合权重由维度为 |$\mathcal{O}$| 的向量 α^(i,j) 参数化。然后架构搜索的任务就简化为学习一组连续变量α={α^(i,j)}，如图1所示。在搜索结束时，可以通过用最可能（likely）的操作替换每个混合操作${\bar{o}}^{(i,j)}$来获得离散体系结构。例如，$o^{(i,j)}=argma{x}_{o\in \mathcal{O}}{\alpha }_o^{(i,j)}$。在下文中，我们将α称为编码架构。

放松（搜索空间）之后，我们的目标是共同学习所有混合操作中的体系结构α和权重w（例如卷积滤波器的权重）。类似于使用强化学习或演算（evolution）的架构搜索，它们将验证集性能作为奖励或适用性的指标，DARTS旨在通过使用梯度下降来优化验证损失。

声明${\mathcal{L}}_{train}$和${\mathcal{L}}_{val}$来分别表示训练损失和验证损失。这两种损失不仅由架构α决定，而且由网络中的权重w决定。架构搜索的目标就是找到使验证损失${\mathcal{L}}_{val}(w\ast ,{\alpha }^{\ast })$最小的${\alpha }^{\ast }$，其中与架构相关的权重w^*通过最小化训练损失获得 $w^{\ast }=argmi{n}_w{\mathcal{L}}_{train}(w,{\alpha }^{\ast })$。

这暗示了一个双层优化问题，其中α为上层变量，w为下层变量：

$$\underset{\alpha }{\min }{\mathcal{L}}_{val}(w^{\ast }(\alpha ),\alpha )\text{\hspace{1em}(3)}$$
$$s.t.w^{\ast }(\alpha )=argmi{n}_w{\mathcal{L}}_{train}(w,\alpha )\text{\hspace{1em}(4)}$$

嵌套公式也出现在基于梯度的超参数优化中，这在某种意义上是相关的，即体系结构α可以看作是一种特殊类型的超参数，尽管它的维数大大高于标量值的超参数，例如学习率，并且(嵌套公式)很难优化。

$$注：\nabla 求梯度$$
2.3 近似架构梯度

由于昂贵的内部优化，准确评估架构梯度可能会令人望而却步。因此，我们提出了一种简单的近似方案，如下所示：
$$\begin{gathered} {\nabla }_{\alpha }{\mathcal{L}}_{val}(w^{\ast }(\alpha ),\alpha ) \\ \approx {\nabla }_{\alpha }{\mathcal{L}}_{val}(w-\xi {\nabla }_w{\mathcal{L}}_{train}(w,\alpha ),\alpha )\text{\hspace{1em}(6)} \end{gathered}$$
其中w表示算法保持的当前权重，ξ是内部优化步骤的学习率。这个想法是通过仅使用一个训练步骤就对w进行自适应来近似$w^{\ast }(\alpha )$，而不是通过训练直到收敛来完全解决内部优化问题（等式4）。 相关技术已经用于元学习中模型迁移，基于梯度的超参数调整和展开式的生成对抗网络GAN。注意，如果w对于内部优化来说是局部最优的，则方程6将简化为${\nabla }_{\alpha }{\mathcal{L}}_{val}(w,\alpha )$，因此${\nabla }_{\alpha }{\mathcal{L}}_{train}(w,\alpha )=0$。

算法1中概述了这种迭代过程。虽然我们目前尚不了解优化算法的收敛性保证，但实际上，通过选择合适的${\xi }^1$可以达到固定点。我们还注意到，当为权重优化启用动量（momentum）时，方程式6中的一步式展开学习目标会相应修改，我们所有的分析仍然适用。将链式规则应用于近似架构梯度（方程式6）可得出：
$${\nabla }_{\alpha }{\mathcal{L}}_{val}(w^{\prime },\alpha )-\xi {\nabla }_{\alpha ,w}^2{\mathcal{L}}_{train}(w,\alpha ){\nabla }_{w^{\prime }}{\mathcal{L}}_{val}(w^{\prime },\alpha )\text{\hspace{1em}(7)}$$
其中$w^{\prime }=w-\xi {\nabla }_w{\mathcal{L}}_{train}(w,\alpha )$表示单步前进模型的权重。上面的表达式在第二项中包含昂贵的矩阵向量乘积。幸运的是，使用有限差分近似（ finite difference approximation）可以大大降低复杂度。使用$ϵ$为小标量2，并且有$w^{\pm }=w\pm ϵ{\nabla }_{w^{\prime }}{\mathcal{L}}_{val}(w^{\prime },\alpha )$。然后：
$${\nabla }_{\alpha ,w}^2{\mathcal{L}}_{train}(w,\alpha ){\nabla }_{w^{\prime }}(L{)}_{val}(w^{\prime },\alpha )\approx \frac{{\nabla }_{\alpha }{\mathcal{L}}_{train}(w^{+},\alpha )-{\nabla }_{\alpha }{\mathcal{L}}_{train}(w^{-},\alpha )}{2ϵ}\text{\hspace{1em}(8)}$$
计算有限差分仅需要权重两次前向传播，而α则需要两次后向传播，并且复杂度从O(|α|| w |)降低为O(|α| + | w |)。

一阶近似: 当ξ= 0时，等式7中的二阶导数将消失。在这种情况下，架构梯由${\nabla }_{\alpha }{\mathcal{L}}_{val}(w,\alpha )$给出，对应于通过假设当前$w$与$w^{\ast }(\alpha )$相同来优化验证损失的简单启发式算法。根据我们在表1和表2中的实验结果，这会导致一些速度提升，但从经验上讲会降低性能。在下文中，我们将ξ= 0的情况称为一阶近似，而将ξ> 0的梯度公式称为二阶近似。

2.4 推导离散架构

为了形成离散体系结构中的每个节点，我们保留从所有先前节点收集的所有非零候选操作中排在前k位的最强操作（来自不同的节点）。一项操作的强度定义为$\frac{exp({\alpha }_o^{(i,j)})}{{\sum }_{o^{\prime }\in \mathcal{O}}exp({\alpha }_{o^{\prime }}^{(i,j)})}$.为了使我们的派生架构与现有工作具有可比性，我们对卷积单元使用k = 2，对递归单元使用k = 1。

以上操作将零运算排除在外出于两个原因。首先，我们需要每个节点正好有k个非零输入边，以便与现有模型进行公平比较。其次，归零操作的强度是不确定的，因为增加归零操作的对数仅会影响所得节点表示的规模，并且由于存在批次归一化而不会影响最终分类结果。

3 实验与结果

我会以视频的形式与大家分享讲解:)
Bilibili主页: https://space.bilibili.com/481802918