Search papers, labs, and topics across Lattice.
This paper introduces In-Context Whitening (ICW), a novel approach for adapting machine-learned surrogates for the AC power flow (ACPF) problem to changes in network topology without relying on gradients or weights. By leveraging statistical properties of the output space, ICW effectively re-estimates the necessary corrections from a limited number of solved cases on new topologies, achieving significant accuracy improvements. The method demonstrates a reduction in overall error by up to 28 times and a 30-fold decrease in worst-bus power-balance mismatch across various test systems, while also being substantially faster than traditional gradient-based adaptation methods.
ICW achieves up to 28 times improvement in accuracy for power flow surrogates while adapting 21 to 34 times faster than conventional methods.
Machine-learned surrogates for the AC power flow (ACPF) problem amortize the cost of repeated solves on a fixed network, but lose one to two orders of magnitude of accuracy when a line outage changes the topology. This degradation is an operator shift. The altered admittance matrix changes the input-to-output map, so identical inputs yield a different output distribution. Existing methods correct this with target-topology data and per-topology gradient steps. We ask whether the correction can instead be made statistical and gradient-free. We propose In-Context Whitening (ICW), which trains an ACPF surrogate in an output space whitened by the base topology's first two moments, and adapts it to an unseen N-1 or N-2 topology by re-estimating that whitening from a few hundred solved cases on the new topology. This adaptation is gradient-free, weight-free, and architecture-agnostic. We prove that among affine whiteners the unique choice that preserves the coordinate-wise semantics of the physical output vector is ZCA whitening, so within efficient invertible corrections, two moments are sufficient. Across the IEEE 30-, 118-, and 300-bus systems under N-1 and N-2 contingencies, ICW reduces overall error by 6$\times$ to 28$\times$ over frozen surrogates (up to 54$\times$ per-quantity under N-2) and cuts worst-bus power-balance mismatch by up to 30$\times$, with consistent gains across three backbones. At deployment scale it matches or beats gradient-based adaptation in accuracy while adapting 21$\times$ to 34$\times$ faster, with a cost that parallelizes on commodity CPU cores rather than requiring one GPU per contingency.
