paperbot · PL 论文追踪

全部论文 待读 收藏

P2 Tensor-Aware Energy Accounting

张量感知的能量核算
ICSE Proceedings of the IEEE/ACM 46th International Conference on Software Engineering2024匹配度 3/10 · 论文关注 TensorFlow 深度学习程序的张量级能耗核算,属于 PL 与 ML 交叉的工程工具,但并不涉及静态分析、形式化验证、Agent/LLM、浮点程序分析或贝叶斯程序分析等用户核心方向;仅在与 ML 系统程序分析沾边的层面上有弱相关性。引用 5 inbox
Timur Babakol, Yu David Liu
crossref: 2026-09-12

AI 速览

一句话:提出 Smaragdine,一种张量感知的能量核算系统,可将 TensorFlow 深度学习程序的能耗按逻辑层次结构分解。

随着 AI 与深度学习的快速增长,其能耗对可持续性影响日益显著。本文提出 Smaragdine,一种面向 TensorFlow 张量式深度学习程序的能量核算系统。其核心是一种新颖的白盒能量核算方法:Smaragdine 感知深度学习程序的内部结构,称为张量感知能量核算。借助 Smaragdine,深度学习程序的能耗可被分解为与逻辑层次分解结构对齐的单元。作者将 Smaragdine 应用于理解 BERT 的能耗行为,逐层、逐张量地识别出 BERT 中能耗/功率最高的组件。此外,通过两个案例研究展示 Smaragdine 如何支持下游工具链构建:一是 BERT 超参数调优的能耗影响比较,二是 BERT 向其下一代 ALBERT 演化时的能耗行为变化。

问题
深度学习程序能耗巨大且缺乏细粒度的能耗归因手段,难以定位高能耗组件并指导优化。
方法
提出 Smaragdine,一种张量感知的白盒能量核算系统,通过感知 TensorFlow 程序的内部逻辑层次结构,将能耗分解到层和张量级别。
评估
在 BERT 上逐层逐张量分析能耗并识别高能耗/高功率组件;通过超参数调优对比和 BERT 到 ALBERT 演化的两个案例研究,验证其支持下游工具链构建的能力。
原文摘要(Abstract)

With the rapid growth of Artificial Intelligence (AI) applications supported by deep learning (DL), the energy efficiency of these applications has an increasingly large impact on sustainability. We introduce Smaragdine, a new energy accounting system for tensor-based DL programs implemented with TensorFlow. At the heart of Smaragdine is a novel white-box methodology of energy accounting: Smaragdine is aware of the internal structure of the DL program, which we call tensor-aware energy accounting. With Smaragdine, the energy consumption of a DL program can be broken down into units aligned with its logical hierarchical decomposition structure. We apply Smaragdine for understanding the energy behavior of BERT, one of the most widely used language models. Layer-by-layer and tensor-by-tensor, Smaragdine is capable of identifying the highest energy/power-consuming components of BERT. Furthermore, we conduct two case studies on how Smaragdine supports downstream toolchain building, one on the comparative energy impact of hyperparameter tuning of BERT, the other on the energy behavior evolution when BERT evolves to its next generation, ALBERT.

链接与引用

DOI 原文 · PDF(开放获取) ·

BibTeX
@inproceedings{paperbot11344,
  title = {Tensor-Aware Energy Accounting},
  author = {Timur Babakol and Yu David Liu},
  booktitle = {ICSE},
  year = {2024},
  doi = {10.1145/3597503.3639156}
}