PowerGridworld: A Framework for Multi-Agent Reinforcement Learning in Power Systems
Authors: David Biagioni, Xiangyu Zhang, Dylan Wald, Deepthi Vaidhynathan, Rhoit Chintala, Jennifer King, Ahmed S. Zamzam
Corresponding author: David Biagioni
All authors are with the National Renewable Energy Laboratory (NREL).
Description
PowerGridworld provides users with a lightweight, modular, and customizable framework for creating power-systems-focused, multi-agent Gym environments that readily integrate with existing training frameworks for reinforcement learning (RL). Although many frameworks exist for training multi-agent RL (MARL) policies, none can rapidly prototype and develop the environments themselves, especially in the context of heterogeneous (composite, multidevice) power systems where power flow solutions are required to define grid-level variables and costs. PowerGridworld is an opensource software package that helps to fill this gap. To highlight PowerGridworld’s key features, we include two case studies and demonstrate learning MARL policies using both OpenAI’s multi-agent deep deterministic policy gradient (MADDPG) and RLLib’s proximal policy optimization (PPO) algorithms. In both cases, at least some subset of agents incorporates elements of the power flow solution at each time step as part of their reward (negative cost) structures.
Please refer to our preprint on arXiv for more details. Data and run scripts used to generate figures in the preprint are available in the paper directory.
Basic installation instructions
Env setup:
conda create -n gridworld python=3.8 -y
conda activate gridworld
git clone [email protected]:NREL/PowerGridworld.git
cd PowerGridWorld
pip install -e .
pip install -r requirements.txt
Run the pytests to sanity check:
pytest tests/
pytests --nbmake examples/envs
Examples
Examples of running various environments and MARL training algorithms can be found in examples.
Funding Acknowledgement
This work was authored by the National Renewable Energy Laboratory (NREL), operated by Alliance for Sustainable Energy, LLC, for the U.S. Department of Energy (DOE) under Contract No. DE-AC36-08GO28308. This work was supported by the Laboratory Directed Research and Development (LDRD) Program at NREL.
Citation
If citing this work, please use the following:
@article{biagioni2021powergridworld,
title={PowerGridworld: A Framework for Multi-Agent Reinforcement Learning in Power Systems},
author={Biagioni, David and Zhang, Xiangyu and Wald, Dylan and Vaidhynathan, Deepthi and Chintala, Rohit and King, Jennifer and Zamzam, Ahmed S},
journal={arXiv preprint arXiv:2111.05969},
year={2021}
}
1 Dec 17, 2021
56 Nov 15, 2022
8 Sep 30, 2022
1.1k Jan 2, 2023
669 Dec 29, 2022
1 May 15, 2022
0 Jul 26, 2022
183 Dec 27, 2022
8 Nov 7, 2022
35 Oct 26, 2022
0 Mar 08, 2022
24 Dec 17, 2022
2 Dec 04, 2021
11 Nov 14, 2022
8 Dec 27, 2022
25 Dec 13, 2022
7 Oct 27, 2022
1 Oct 08, 2021
17 Jan 03, 2023
1 Dec 28, 2021
734 Jan 03, 2023
13 Mar 05, 2022
46 Dec 19, 2022
427 Jan 03, 2023
22 Dec 12, 2022
109 Dec 29, 2022
184 Dec 12, 2022
4 Sep 08, 2022
437 Jan 03, 2023