Energy efficiency of Python machine learning frameworks

• Main Author: Ajel, Salwa
• Publication Date: 2023
• Other Authors: Ribeiro, Francisco, Ejbali, Ridha, Saraiva, João
• Language: eng
• Source: Repositórios Científicos de Acesso Aberto de Portugal (RCAAP)
• Download full: https://hdl.handle.net/1822/90293
• topic: DeepLearning; Energy-Efficient; Execution time; Keras; Machine Learning; Memory usage; Pytorch; Tensorflow; Engenharia e Tecnologia::Engenharia Eletrotécnica, Eletrónica e Informática

Although machine learning (ML) is a field that has been the subject of research for decades, a large number of applications with high computational power have recently emerged. Usually, we only focus on solving machine learning problems without considering how much energy has been consumed by the different frameworks used for such applications. This study aims to provide a comparison among four widely used frameworks such as Tensorflow, Keras, Pytorch, and Scikit-learn in terms of many aspects, including energy efficiency, memory usage, execution time, and accuracy. We monitor the performance of such frameworks using different well-known machine learning benchmark problems. Our results show interesting findings, such as slower and faster frameworks consuming less or more energy, higher or lower memory usage, etc. We show how to use our results to provide machine learning developers with information to decide which framework to use for their applications when energy efficiency is a concern.