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
1.4k Dec 29, 2022
154 Jan 02, 2023
0 Jul 13, 2021
309 Dec 12, 2022
320 Dec 28, 2022
76 Dec 08, 2022
38 Nov 25, 2022
111 Dec 21, 2022
1 Oct 03, 2022
109 Nov 22, 2022
7 Jan 03, 2023
7.2k Dec 29, 2022
171 Dec 05, 2022
60 Dec 31, 2022
789 Dec 29, 2022
2 Dec 20, 2021
433 Dec 31, 2022
123 Jan 01, 2023
11 Aug 22, 2022
3 Dec 18, 2021