arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.07593cs.LGcs.NEstat.ML
收藏

Optimal Transport Kernels for Sequential and Parallel Neural Architecture Search

Amazon(亚马逊);RIKEN AIP(理化学研究所先进智能项目中心);Kyoto University(京都大学);University of Oxford(牛津大学)

Vu Nguyen, Tam Le, Makoto Yamada, Michael A Osborne

英文摘要:

Neural architecture search (NAS) automates the design of deep neural networks. One of the main challenges in searching complex and non-continuous architectures is to compare the similarity of networks that the conventional Euclidean metric may fail to capture. Optimal transport (OT) is resilient to such complex structure by considering the minimal cost for transporting a network into another. However, the OT is generally not negative definite which may limit its ability to build the positive-definite kernels required in many kernel-dependent frameworks. Building upon tree-Wasserstein (TW), which is a negative definite variant of OT, we develop a novel discrepancy for neural architectures, and demonstrate it within a Gaussian process surrogate model for the sequential NAS settings. Furthermore, we derive a novel parallel NAS, using quality k-determinantal point process on the GP posterior, to select diverse and high-performing architectures from a discrete set of candidates. Empirically, we demonstrate that our TW-based approaches outperform other baselines in both sequential and parallel NAS.

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

↑