Skip to content

amai-gsu/GreenAuto

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GreenAuto: An Automated Platform for Sustainable AI Model Design on Edge Devices

Introduction:

We present GreenAuto, an end-to-end automated platform designed for sustainable AI model exploration, generation, deployment, and evaluation. GreenAuto employs a Pareto front-based search method within an expanded neural architecture search (NAS) space, guided by gradient descent to optimize model exploration. Pre-trained kernel-level energy predictors estimate energy consumption across all models, providing a global view that directs the search toward more sustainable solutions. By automating performance measurements and iteratively refining the search process, GreenAuto demonstrates the efficient identification of sustainable AI models without the need for human intervention. please check our paper for more details: https://arxiv.org/abs/2501.14995

System Design:

GreenAuto contains three essential components: (1) energy efficiency driven model search space that includes over 900,000 DNN models tailored for energy efficiency, (2) Pareto front-based model search algorithm that, combined with gradient descent-based model sampling, efficiently identifies promising model candidates from the search space, and (3) automated energy measurement function that leverages an external power monitor to capture accurate power consumption data during model execution.

Screenshot 2025-06-01 at 1 06 57 PM

Code:

analysis:

The main function for the model search. SVGD_Pareto.py performs Pareto model search based on gradient descent.

monsoon:

Functions for power measurement using the Monsoon Power Monitor.

results:

Outputs generated from the model search.

score:

Estimated model accuracy and power consumption.

Model Search Results:

Screenshot 2025-06-01 at 1 22 25 PM

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published