arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.08151cs.LGcs.AIcs.NE
收藏

测量深度神经网络的能耗与效率:实证分析与设计建议

Measuring the Energy Consumption and Efficiency of Deep Neural Networks: An Empirical Analysis and Design Recommendations

National Renewable Energy Laboratory(国家可再生能源实验室);Carnegie Mellon University(卡内基梅隆大学)

Charles Edison Tripp, Jordan Perr-Sauer, Jamil Gafur, Amabarish Nag, Avi Purkayastha, Sagi Zisman, Erik A. Bensen

AI总结:

本研究通过BUTTER-E数据集实证分析全连接神经网络的能耗,提出考虑缓存效应的能量模型,并揭示能效与网络设计的非线性关系,倡导综合设计以实现绿色AI。

摘要:

针对大规模神经网络能耗不断上升的所谓“红AI”趋势,本研究利用节点级功率计测量了训练各种全连接神经网络架构的实际能耗。我们引入了BUTTER-E数据集,这是对BUTTER经验深度学习数据集的扩展,包含63,527次独立实验运行的能耗和性能数据,涵盖30,582种不同配置:13个数据集、20种规模(可训练参数数量)、8种网络“形状”和14种深度,并在CPU和GPU硬件上使用节点级功率计收集数据。该数据集揭示了数据集大小、网络结构与能耗之间的复杂关系,并强调了缓存效应的影响。我们提出了一种简单而有效的能量模型,该模型考虑了网络规模、计算和内存层次结构。我们的分析还揭示了一种令人惊讶的、由硬件介导的能效与网络设计之间的非线性关系,挑战了减少参数数量或FLOPs是实现更高能效的最佳方式的假设。强调需要开发考虑缓存的算法,我们建议采用结合网络、算法和硬件设计的综合方法来实现能效。这项工作为可持续计算和绿色AI领域做出了贡献,为创建更节能的神经网络和促进可持续AI提供了实用指导。

英文摘要:

Addressing the so-called ``Red-AI'' trend of rising energy consumption by large-scale neural networks, this study investigates the actual energy consumption, as measured by node-level watt-meters, of training various fully connected neural network architectures. We introduce the BUTTER-E dataset, an augmentation to the BUTTER Empirical Deep Learning dataset, containing energy consumption and performance data from 63,527 individual experimental runs spanning 30,582 distinct configurations: 13 datasets, 20 sizes (number of trainable parameters), 8 network ``shapes'', and 14 depths on both CPU and GPU hardware collected using node-level watt-meters. This dataset reveals the complex relationship between dataset size, network structure, and energy use, and highlights the impact of cache effects. We propose a straightforward and effective energy model that accounts for network size, computing, and memory hierarchy. Our analysis also uncovers a surprising, hardware-mediated non-linear relationship between energy efficiency and network design, challenging the assumption that reducing the number of parameters or FLOPs is the best way to achieve greater energy efficiency. Highlighting the need for cache-considerate algorithm development, we suggest a combined approach to energy efficient network, algorithm, and hardware design. This work contributes to the fields of sustainable computing and Green AI, offering practical guidance for creating more energy-efficient neural networks and promoting sustainable AI.

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

↑