arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.14525cs.CV
收藏

神经结构搜索中的层间转换

Inter-layer Transition in Neural Architecture Search

Northwestern Polytechnical University(西北工业大学);The University of Sydney(悉尼大学)

Benteng Ma, Jing Zhang, Yong Xia, Dacheng Tao

摘要:

差分神经结构搜索(Differential Neural Architecture Search,NAS)方法将网络结构表示为一个重复的代理有向无环图(DAG),并以差分方式交替优化网络权值和结构权值。然而,现有方法对每个边缘上的架构权重进行建模(即,网络中的一层)作为统计独立变量,忽略DAG中由其有向拓扑连接引起的边之间的依赖性。在本文中,我们提出了一种新的层间转换NAS方法,首次尝试调查这种依赖。它将体系结构优化转化为一个连续的决策过程,其中连接边的体系结构权重之间的依赖关系被显式建模。具体地,根据边缘的前趋边缘是否在同一单元中,将边缘划分为内部组和外部组。外边缘的结构权值独立优化,内边缘的结构权值基于其前一个边缘的结构权值和可学习的转移矩阵,以一种注意的概率转移方式依次导出。五个基准测试的实验证实了建模层间依赖的价值,并证明了所提出的方法优于国家的最先进的方法。

英文摘要:

Differential Neural Architecture Search (NAS) methods represent the network architecture as a repetitive proxy directed acyclic graph (DAG) and optimize the network weights and architecture weights alternatively in a differential manner. However, existing methods model the architecture weights on each edge (i.e., a layer in the network) as statistically independent variables, ignoring the dependency between edges in DAG induced by their directed topological connections. In this paper, we make the first attempt to investigate such dependency by proposing a novel Inter-layer Transition NAS method. It casts the architecture optimization into a sequential decision process where the dependency between the architecture weights of connected edges is explicitly modeled. Specifically, edges are divided into inner and outer groups according to whether or not their predecessor edges are in the same cell. While the architecture weights of outer edges are optimized independently, those of inner edges are derived sequentially based on the architecture weights of their predecessor edges and the learnable transition matrices in an attentive probability transition manner. Experiments on five benchmarks confirm the value of modeling inter-layer dependency and demonstrate the proposed method outperforms state-of-the-art methods.

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

↑