传统的NAS不适合小微实验室,DARTS提供了一种新的思路——利用梯度下降算法来进行模型结构搜索。
一,网络的结构用有向无环图表示。
二,引入了一个简单的搜索空间连续松弛方案,从而为体系结构及其权重的联合优化提供了一个可微学习目标
三,最后,提出了一种近似技术,使算法在计算上可行和有效
搜索空间
搜索一个计算单元作为最终体系结构的构建块。一个计算单元是由N个节点的有序序列组成的有向无环图。

上图为一个计算单元的表示(以三个节点为例)。其中,o(i,j)表示节点i,与节点j,构成点边上的operation(卷积,池化,正则化等),下一个节点的值是由输入节点计算得到的。
连续松弛与优化

为了使得搜索空间松弛(使每个节点的操作选择不绝对),将特定操作的分类选择放宽到所有可能的操作的Softmax。
其中大写的O表示所有可能的operation的集合。 α \alpha <

DARTS提出一种新型神经架构搜索方法,利用梯度下降算法优化模型结构,通过连续松弛策略使搜索过程可微,实现高效计算。采用有向无环图表示网络结构,将操作选择转换为Softmax分布,简化NAS任务为连续变量的学习。
最低0.47元/天 解锁文章
1万+

被折叠的 条评论
为什么被折叠?



