Maximal Load Shedding Verification for Neural Network Models of AC Line Switching
Samuel Chevalier, Duncan Starkenburg, Robert Parker, Noah Rhodes
TL;DR
The paper addresses the risk of deploying NN-based line-switching decisions in AC grids by formulating a bilevel attacker–defender verification problem that maximizes load shedding under NN-guided topology changes. It solves the inner AC-redispatch problem via a SOC relaxation and dualization to obtain a single-level, tractable bound, implemented in MathOptAI.jl, and couples this with an outer NN-input optimization. The key contributions are (i) a bilevel verification framework for maximal load shedding, (ii) a dualized SOC-OPF approach yielding guaranteed lower bounds, and (iii) a scalable pipeline combining NN proxies with conic and AC-redispatch restorations that scales to networks with millions of NN parameters. The findings indicate the method provides meaningful lower bounds and can be solved efficiently, supporting safer deployment of ML-driven OPS decisions in practice, with future work to tighten bounds and explore upper-bound guarantees.
Abstract
Solving for globally optimal line switching decisions in AC transmission grids can be intractability slow. Machine learning (ML) models, meanwhile, can be trained to predict near-optimal decisions at a fraction of the speed. Verifying the performance and impact of these ML models on network operation, however, is a critically important step prior to their actual deployment. In this paper, we train a Neural Network (NN) to solve the optimal power shutoff line switching problem. To assess the worst-case load shedding induced by this model, we propose a bilevel attacker-defender verification approach that finds the NN line switching decisions that cause the highest quantity of network load shedding. Solving this problem to global optimality is challenging (due to AC power flow and NN nonconvexities), so our approach exploits a convex relaxation of the AC physics, combined with a local NN search, to find a guaranteed lower bound on worst--case load shedding. These under-approximation bounds are solved via MathOptAI.jl. We benchmark against a random sampling approach, and we find that our optimization-based approach always finds larger load shedding. Test results are collected on multiple PGLib test cases and on trained NN models which contain more than 10 million model parameters.
