中文

DANCE:可微分加速器与网络协同探索

机器学习 2021-02-17 v3 人工智能 硬件体系结构 机器学习

摘要

为应对深度神经网络(DNN)执行中日益增长的计算需求,近期的神经架构搜索(NAS)算法将硬件成本指标(如 GPU 延迟)纳入考量。为进一步实现快速、高效的执行,针对多种目的设计的 DNN 专用硬件加速器,其效率远超 GPU。然而,这些硬件相关指标已被证明与网络架构之间存在非线性关系。因此,针对加速器优化网络还是针对网络优化加速器,成了一个先有鸡还是先有蛋的问题。在此情形下,本工作提出 DANCE,一种面向硬件加速器与网络架构设计协同探索的可微分方法。DANCE 的核心是一个可微分评估器网络。通过将硬件评估软件建模为神经网络,加速器架构与硬件指标之间的关系变得可微分,从而允许利用反向传播进行搜索。与朴素的现有方法相比,我们的方法在显著更短的时间内完成协同探索,同时取得了更优的准确率和硬件成本指标。

关键词

引用

@article{arxiv.2009.06237,
  title  = {DANCE: Differentiable Accelerator/Network Co-Exploration},
  author = {Kanghyun Choi and Deokki Hong and Hojae Yoon and Joonsang Yu and Youngsok Kim and Jinho Lee},
  journal= {arXiv preprint arXiv:2009.06237},
  year   = {2021}
}

备注

Accepted to DAC 2021