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

D-DARTS:分布式可区分体系结构搜索

D-DARTS: Distributed Differentiable Architecture Search

Université Paris-Saclay(巴黎萨克雷大学);MIT(麻省理工学院)

Alexandre Heuillet, Hedi Tabia, Hichem Arioui, Kamal Youcef-Toumi

摘要:

可区分的ARCHITERMS搜索(DARTS)是最流行的神经架构搜索(NAS)方法之一。它通过权重共享大大降低了搜索成本。然而,它也大大减少了搜索空间,从而排除了潜在的有前途的架构。在本文中,我们提出了D-DARTS,这是一种解决方案,它通过在细胞级别嵌套神经网络来解决这个问题,而不是使用权重共享来产生更多样化和专业化的架构。此外,我们引入了一种新的算法,可以从一些训练过的细胞中获得更深的架构,提高性能并节省计算时间。此外,我们还提出了一种替代的搜索空间(DARTOpti),在其中我们优化了现有的手工架构(例如,而不是从头开始。这种方法伴随着一个新的度量,该度量在我们的自定义搜索空间内测量架构之间的距离。我们的解决方案在多个计算机视觉任务上达到了具有竞争力的性能。代码和预训练模型可以在https://github.com/aheuillet/D-DARTS上访问。

英文摘要:

Differentiable ARchiTecture Search (DARTS) is one of the most trending Neural Architecture Search (NAS) methods. It drastically reduces search cost by resorting to weight-sharing. However, it also dramatically reduces the search space, thus excluding potential promising architectures. In this article, we propose D-DARTS, a solution that addresses this problem by nesting neural networks at the cell level instead of using weight-sharing to produce more diversified and specialized architectures. Moreover, we introduce a novel algorithm that can derive deeper architectures from a few trained cells, increasing performance and saving computation time. In addition, we also present an alternative search space (DARTOpti) in which we optimize existing handcrafted architectures (e.g., ResNet) rather than starting from scratch. This approach is accompanied by a novel metric that measures the distance between architectures inside our custom search space. Our solution reaches competitive performance on multiple computer vision tasks. Code and pretrained models can be accessed at https://github.com/aheuillet/D-DARTS.

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

↑