Rajput, Saurabhsingh;
Kechagia, Maria;
Sarro, Federica;
Sharma, Tushar;
(2024)
Greenlight: Highlighting TensorFlow APIs Energy Footprint.
In:
2024 IEEE/ACM 21st International Conference on Mining Software Repositories (MSR).
IEEE /ACM: Lisbon, Portugal.
Preview |
Text
MSR2024_Greenlight.pdf - Accepted Version Download (722kB) | Preview |
Abstract
Deep learning (dl) models are being widely deployed in real-world applications, but their usage remains computationally intensive and energy-hungry. While prior work has examined model-level energy usage, the energy footprint of the dl frameworks, such as TensorFlow and PyTorch, used to train and build these models, has not been thoroughly studied. We present Greenlight, a largescale dataset containing fine-grained energy profiling information of 1284 TensorFlow api calls. We developed a command line tool called CodeGreen to curate such a dataset. CodeGreen is based on our previously proposed framework FECoM, which employs static analysis and code instrumentation to isolate invocations of TensorFlow operations and measure their energy consumption precisely. By executing api calls on representative workloads and measuring the consumed energy, we construct detailed energy profiles for the apis. Several factors, such as input data size and the type of operation, significantly impact energy footprints. Greenlight provides a ground-truth dataset capturing energy consumption along with relevant factors such as input parameter size to take the first step towards optimization of energy-intensive TensorFlow code. The Greenlight dataset opens up new research directions such as predicting api energy consumption, automated optimization, modeling efficiency trade-offs, and empirical studies into energy-aware dl system design
Type: | Proceedings paper |
---|---|
Title: | Greenlight: Highlighting TensorFlow APIs Energy Footprint |
Event: | MSR 2024 Data and Tools Showcase Track |
Location: | Lisbon, Portugal |
Dates: | 15 Apr 2024 - 16 Apr 2024 |
ISBN-13: | 979-8-3503-6398-2 |
Open access status: | An open access version is available from UCL Discovery |
DOI: | 10.1145/3643991.3644894 |
Publisher version: | https://doi.org/10.1145/3643991.3644894 |
Language: | English |
Additional information: | This version is the author accepted manuscript. For information on re-use, please refer to the publisher’s terms and conditions. |
Keywords: | Energy measurement, Green Artificial Intelligence, Fine-grained energy measurement |
UCL classification: | UCL UCL > Provost and Vice Provost Offices > UCL BEAMS UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Computer Science |
URI: | https://discovery.ucl.ac.uk/id/eprint/10192368 |




Archive Staff Only
![]() |
View Item |