GPU-Accelerated Verification of Machine Learning Models for Power Systems

TL;DR

This paper introduces GPU-accelerated methods for verifying large-scale machine learning models in power systems, enabling simultaneous verification of multiple problems and direct encoding of power flow constraints, significantly improving speed and flexibility.

Contribution

It presents novel transformations and dualization techniques that allow simultaneous verification and direct constraint encoding, enhancing existing GPU-accelerated verification tools for power system applications.

Findings

Achieved over 100x speedup in verification tasks.

Enabled verification of multiple problems simultaneously.

Improved flexibility in verifying power system models.

Abstract

Computational tools for rigorously verifying the performance of large-scale machine learning (ML) models have progressed significantly in recent years. The most successful solvers employ highly specialized, GPU-accelerated branch and bound routines. Such tools are crucial for the successful deployment of machine learning applications in safety-critical systems, such as power systems. Despite their successes, however, barriers prevent out-of-the-box application of these routines to power system problems. This paper addresses this issue in two key ways. First, for the first time to our knowledge, we enable the simultaneous verification of multiple verification problems (e.g., checking for the violation of all line flow constraints simultaneously and not by solving individual verification problems). For that, we introduce an exact transformation that converts the "worst-case" violation across a set of potential violations to a series of ReLU-based layers that augment the original neural network. This allows verifiers to interpret them directly. Second, power system ML models often must be verified to satisfy power flow constraints. We propose a dualization procedure which encodes linear equality and inequality constraints directly into the verification problem; and in a manner which is mathematically consistent with the specialized verification tools. To demonstrate these innovations, we verify problems associated with data-driven security constrained DC-OPF solvers. We build and test our first set of innovations using the $\alpha,\beta$-CROWN solver, and we benchmark against Gurobi 10.0. Our contributions achieve a speedup that can exceed 100x and allow higher degrees of verification flexibility.