中文

能量感知的DNN图优化

机器学习 2026-01-27 v2 机器学习

摘要

与现有针对推理性能的深度神经网络(DNN)图优化工作不同,我们探索用于功耗和资源受限机器学习设备的能量感知与节能的DNN图优化。我们提出一种方法,允许用户优化DNN图的能量消耗或在能量与推理性能之间取得平衡。该方法高效地搜索等价图空间,并识别出执行代价最小的图及相应算法。我们实现了该方法并在基于GPU的机器上用多个DNN模型进行了评估。结果表明,我们的方法实现了显著的节能,即在性能影响可忽略的情况下节能24%。

关键词

引用

@article{arxiv.2005.05837,
  title  = {Energy-Aware DNN Graph Optimization},
  author = {Yu Wang and Rong Ge and Shuang Qiu},
  journal= {arXiv preprint arXiv:2005.05837},
  year   = {2026}
}

备注

Accepted paper at Resource-Constrained Machine Learning (ReCoML) Workshop of MLSys 2020 Conference, Austin, TX, USA, 2020