中文

ApproxDARTS:基于近似乘法器的可微神经架构搜索

机器学习 2025-10-23 v1

摘要

将近似计算原理集成到硬件感知深度神经网络(DNN)的设计中,使得 DNN 的实现展现出良好的输出质量以及高度优化的硬件参数,如低延迟或低推理能耗。在本工作中,我们提出了 ApproxDARTS,一种神经架构搜索(NAS)方法,使流行的可微神经架构搜索方法 DARTS 能够利用近似乘法器,从而降低生成神经网络的功耗。我们在 CIFAR-10 数据集上表明,ApproxDARTS 能够在不到 1010 个 GPU 小时内完成完整的架构搜索,并生成包含近似乘法器的具有竞争力的卷积神经网络(CNN),其中近似乘法器位于卷积层中。例如,与使用原生 3232 位浮点乘法器的 CNN 相比,ApproxDARTS 创建的 CNN 在推理阶段的算术运算中能耗降低了 (a) 53.84%53.84\%;与使用精确 88 位定点乘法器的 CNN 相比,能耗降低了 (b) 5.97%5.97\%,且在这两种情况下精度下降均可忽略不计。此外,ApproxDARTS 比一种类似但基于进化算法的方法 EvoApproxNAS 快 2.3×2.3\times

关键词

引用

@article{arxiv.2404.08002,
  title  = {ApproxDARTS: Differentiable Neural Architecture Search with Approximate Multipliers},
  author = {Michal Pinos and Lukas Sekanina and Vojtech Mrazek},
  journal= {arXiv preprint arXiv:2404.08002},
  year   = {2025}
}

备注

To appear at the IEEE World Congress on Computational Intelligence (IEEE WCCI 2024) at Yokohama, Japan, 30 June - 5 July 2024