GENCO: One Neural Solver for Three Grid Problems

Paper: 2608.09921 Authors: Alban Puech, Matteo Mazzonelli, Tamara R. Govindasamy, Héctor Maeso-García, Thomas Tolhurst, Javad Bayazi, Ali Moeini, Naomi Simumba, Celia Cintas Categories: cs.AI

Note up front: I’m working from the abstract and the surrounding literature, not the full text. Where I describe internals below, I flag what is stated versus what I’m reconstructing.

The Gap

Neural power system solvers have been a busy corner of ML-for-science for about five years, and the literature is oddly shaped: lots of papers, almost no reusable artifacts.

Where prior work sits:

• Power flow (PF) learners — PowerFlowNet, physics-informed GNNs, unrolled Newton variants. These typically train one model per grid, or per grid family, and are evaluated on mean-squared error against a Newton-Raphson reference. MSE is a proxy; nobody dispatches a grid on MSE.
• OPF learners — DeepOPF and descendants, DC3-style constraint-completion, primal-dual learning. Fast, but feasibility is the perennial problem: a predicted operating point that violates line limits is not a solution, it’s a suggestion.
• State estimation (SE) — largely separate literature, mostly deep-unrolled weighted least squares (WLS) or robust-estimator surrogates.
• Production practice — DC approximations (DC-PF, DC-OPF), which are fast and linear but throw away voltage magnitudes and reactive power. For anything voltage-security-related, that’s the wrong tool.

So there are three structural gaps. First, task fragmentation: three model families for three problems that share the same physics and the same network. Second, no common substrate: no standard synthetic data pipeline, no shared grid representation, so cross-paper numbers are barely comparable. Third, the wrong success metric: MSE-against-solver instead of physical residuals, feasibility, and optimality gap.

GENCO’s bet is that all three collapse if you fix the representation.

  PROBLEM
  [ PF, OPF, SE solved by 3 separate model families ]
  [ no shared data pipeline; MSE-vs-solver metrics  ]
  [ production falls back to DC: no |V|, no Q       ]
        |
        v
  ASSUMPTION
  [ all three tasks = same grid graph + same AC equations ]
  [ they differ only in: which values are known,          ]
  [ which are unknown, and what objective is imposed      ]
        |
        v
  METHOD
  [ GENCO: geometric (graph) encoder over a shared        ]
  [ network representation + corrective iteration driven  ]
  [ by physics residuals; task selected by known/unknown  ]
  [ mask and objective term                               ]
  [ + GridFM Development Framework (data gen + training)  ]
  [ + released PF/OPF datasets, millions of scenarios     ]
        |
        v
  EVIDENCE
  [ PFDelta, OPFData benchmarks; Hydro-Quebec SCADA       ]
  [ PF:  ~30x vs Newton-Raphson, ~2x DC-PF runtime,       ]
  [      full AC state at DC-level P-balance residuals    ]
  [ OPF: ~85x vs IPOPT, better than DC-OPF on feasibility ]
  [ SE:  degrades gracefully where WLS fails to converge  ]
        |
        v
  CONCLUSION
  [ one solver + one framework lowers the entry barrier;  ]
  [ framed as a step toward "Grid Foundation Models"      ]

The Increment

One sentence: Before, you picked a neural solver per task and rebuilt the data pipeline yourself; after, there is one architecture that fills in whichever grid quantities are missing, plus the framework and datasets to reproduce and extend it.

Core Mechanism

The stated design has two load-bearing pieces: a shared network representation and a corrective loop (the C in GENCO). The grid becomes a heterogeneous graph — buses as nodes, branches as edges, generators and loads as attached features — and every quantity of interest (voltage magnitude, voltage angle, active and reactive injections, setpoints) lives on that graph as a per-node or per-edge slot. Crucially, each slot carries a companion flag: *known or to-be-determined. The topology of the graph is data, not architecture, which is what lets one trained model span grids of different sizes.

With that representation, the three tasks stop being three problems. PF: generator setpoints and loads are known, bus voltages are unknown. OPF: loads are known, setpoints are unknown, and a cost objective plus operating limits are imposed. SE: a noisy, partially redundant set of measurements is known, and the true state is unknown. Same graph, same AC equations, different pattern of filled and empty slots.

The “corrective optimizer” part is where this differs from plain regression. Rather than a single forward pass mapping inputs to outputs, the model appears to run an iterative refinement: produce a candidate state, evaluate the actual AC power-flow residuals of that candidate, feed those residuals back in as features, and emit a correction. That is a learned analogue of a Newton step — except the update direction is amortized across the training distribution instead of computed from a fresh Jacobian factorization. It explains both the physics story (residuals are measured, not hoped for) and the speed story (no repeated sparse linear solves), and it explains why the SE results are robust: an iterative corrector has no convergence criterion to fail, so it always hands back something. Whether that something is trustworthy is a separate question, and I’ll come back to it.

    grid case (any topology, any size)
            |
            v
  +---------------------------------------------+
  |  SHARED NETWORK REPRESENTATION              |
  |  nodes: buses  [ |V| th P Q ] + known-mask  |
  |  edges: lines  [ R X B ratio limits ]       |
  |  task tag: PF / OPF / SE                    |
  +---------------------------------------------+
            |
            v
  +---------------------------------------------+
  |  GEOMETRIC ENCODER (message passing)        |
  |  neighbors exchange state; permutation      |
  |  equivariant, so size is not baked in       |
  +---------------------------------------------+
            |
            v
      candidate state x_0
            |
   .--------+--------------------------.
   |        v                          |
   |  [ AC RESIDUAL CHECK ]            |  k corrective
   |  g(x) = P/Q mismatch, limits,     |  passes
   |         measurement misfit        |
   |        |                          |
   |        v                          |
   |  [ CORRECTOR: dx = f(x, g(x)) ]   |
   |        |                          |
   |        +--> x_\{i+1\} = x_i + dx  |
   '-----------------------------------'
            |
            v
   full AC state  ( |V|, theta, P, Q )
   + objective / feasibility reporting

The metaphor: think of a ship’s navigator working a paper chart.

The chart is the shared network representation — one standardized set of symbols that works for a harbor or an ocean, which is why the same navigator handles both. The blanks on the chart are the task: sometimes you know your engine settings and heading and want your position (that’s PF); sometimes you know the destination and the fuel budget and want the engine settings (OPF); sometimes you have three wobbly instrument readings and want to know where you actually are (SE). Same chart, same laws of motion, different blanks.

Dead reckoning is the neural forward pass: a fast, experienced guess that is usually close because the navigator has crossed this water a thousand times. The sextant fix is the AC residual check: it doesn’t tell you the answer, it tells you how wrong you currently are. The course correction is the corrector network, and repeating fix-then-correct a few times is the iterative loop.

Newton-Raphson, in this metaphor, is the navigator who refuses to use experience and recomputes the whole position from spherical trigonometry tables at every step — exact, slow, and occasionally unable to close the calculation at all. DC-PF is the navigator who assumes the Earth is flat and ignores altitude: fast, fine in a small bay, and structurally incapable of telling you your vertical position (voltage magnitude and reactive power). GENCO’s claim is that experienced dead reckoning plus sextant fixes gets you the full three-dimensional position at nearly flat-Earth speed.

Key Concepts

• Same equations, different blanks. The AC power flow equations relate four quantities at every bus: voltage magnitude, voltage angle, active power, reactive power. Two are given, two are solved for — but *which two is a modeling choice, not physics. A generator bus fixes voltage and active power; a load bus fixes active and reactive power. Once you internalize this, PF, OPF, and SE look less like three algorithms and more like three fill-in-the-blank templates over one system of equations. That reframing is the paper’s actual conceptual move; the architecture follows from it.

• The DC shortcut and what it costs. Grid operators run linearized DC power flow constantly because it’s a single linear solve. The linearization assumes voltage magnitudes are all near 1.0 per-unit and angle differences are small, which lets you drop reactive power entirely. That’s a fine approximation for asking “how much power flows down this line,” and useless for asking “will this bus’s voltage collapse.” The abstract’s most concrete claim is on exactly this axis: recover the full AC state (magnitudes and reactive power) while matching DC-PF’s active-power-balance residuals at roughly 2x DC’s runtime. If that holds, it’s a genuinely attractive operating point, because the alternative today is “cheap and blind” or “complete and slow.”

• Residual as input, not just as loss. Most physics-informed ML adds a residual penalty to the training loss and hopes the network internalizes the constraint. Feeding the residual back in as an *input feature at inference time is different in kind: the network gets to see how wrong it currently is on this specific case and respond. That’s the difference between a student who studied the answer key and a student who checks their work. It’s the transferable trick in this paper.

Framework Shift

  Before (mainstream approach):          After (this paper):

  PF task                                    one grid graph
   [ GNN_pf ] -- trained on grid A            [ buses + edges + mask ]
   loss: MSE vs Newton                                |
                                                      v
  OPF task                                   [ geometric encoder ]
   [ DeepOPF_x ] -- grid A, fixed size                |
   loss: MSE + penalty                                v
                                             [ candidate state ]
  SE task                                            |  ^
   [ unrolled WLS net ]                              v  |
   loss: measurement fit                    [ residual check ]
                                                     |
  each: own data pipeline                            v
  each: own grid encoding                    PF / OPF / SE answer
  each: own metric                          ( task = which slots
  cross-paper comparison: hard                are empty + objective )

                                            + GridFM framework
                                            + released datasets
                                            + physics-based metrics

One sentence: from three task-specific regressors judged against a reference solver to one topology-agnostic corrector judged against the physics, the core shift is treating the task as metadata on a shared representation rather than as a separate model.

Expert Assessment

Problem choice. Real gap, and a slightly unglamorous one. The neural-solver literature genuinely does suffer from fragmented representations and non-comparable benchmarks, and the people who could use these tools — power system engineers — are not the people who can stand up a PyG training pipeline. Releasing a low-code framework plus million-scenario datasets addresses that directly, and artifacts like these tend to outlive the model they shipped with. The “step toward Grid Foundation Models” framing is where I’d push back: nothing in the abstract demonstrates the properties that make foundation models interesting — few-shot transfer to unseen task types, scaling behavior, emergent capability. Multi-task supervised learning on a shared graph encoding is a good idea, but it’s a good idea from 2019, not a foundation model. The framing will help the paper get read and will annoy careful reviewers.

Method maturity. The unification insight is clean and correct, and the residual-in-the-loop corrector is the right architecture for the problem — it’s the piece that makes the physics claims credible rather than aspirational. But the components are individually familiar: graph encoders for grids, learned corrective iteration, mask-based task conditioning. The contribution is composition and packaging, executed well. On overlooked simpler alternatives: a warm-started Newton-Raphson, where the neural net supplies only the initial guess and a classical solver does the iterations, gets you exactness plus most of the speedup and is far easier to certify. I’d want the paper to argue explicitly against that baseline, because it’s what a utility’s engineering team will propose in the first meeting.

Experimental integrity. Evaluating on PFDelta and OPFData is the right choice, and the Hydro-Québec SCADA evaluation is a real differentiator — most papers in this space never touch measured data, and real SCADA brings bad data, topology errors, and parameter drift that synthetic sets don’t. Two things I’d scrutinize. First, all speedup claims (30x over Newton-Raphson, 85x over IPOPT) are almost certainly GPU-batched neural inference against single-case CPU solvers. That’s a legitimate deployment comparison and a misleading algorithmic one; the honest version reports per-case latency at batch size one alongside throughput, and states hardware for both sides. Second, “improving feasibility over DC-OPF” is a low bar — DC-OPF solutions are frequently AC-infeasible by construction. The number that matters is feasibility versus IPOPT’s AC-optimal solution, and the optimality gap distribution’s tail, not its mean. Third, the SE claim cuts both ways: “always returns a high-quality estimate even when WLS fails to converge” is a real operational advantage, but WLS non-convergence is also a *signal — often that measurements are corrupted or the assumed topology is wrong. A solver that never announces failure needs a calibrated confidence output, or it’s replacing a loud failure with a quiet one. I’d want to see bad-data detection performance before deploying this.

Writing quality. The abstract is well-organized and unusually specific about numbers, which I appreciate. The predictable soft spots in papers shaped like this one: the architecture section will underspecify how the task mask interacts with the encoder and how many corrective iterations are used (and whether that count is tuned per task, which would complicate the runtime claims); and the limitations discussion will be thin. Rewriting a proper failure-mode section — where does GENCO produce confidently wrong states, what happens on topologies far outside the training distribution, what happens under N-1 contingencies — would do more for the paper’s credibility than another benchmark table. Papers that ship frameworks are also chronically vague about the framework’s scope; a page on what GridFM does *not cover would build trust.

Verdict: weak accept — the unification insight is sound, the artifacts are genuinely useful to the community, and the SCADA evaluation is rare and welcome; the “foundation model” framing is unearned and the speedup comparisons need hardware-parity accounting before I’d cite the numbers.

Takeaways

Concrete things worth stealing:

1. Task-as-mask conditioning. If you have several problems over the same underlying system that differ only in which variables are given and which are sought, don’t build several models — build one that reads a known/unknown mask. This generalizes far beyond grids: structural mechanics with different boundary conditions, chemical process simulation in design-mode versus operating-mode, any inverse/forward pair over shared equations.

2. Feed the residual in, not just into the loss. Physics-informed losses train the model to be right on average over the training distribution. Passing the constraint violation back in as an inference-time feature lets the model be right on *this instance. The cost is a few extra forward passes; the benefit is that your error signal is measured rather than assumed. This is the single most portable idea in the paper.

3. Report the physics residual, not the MSE against a solver. If your neural surrogate is evaluated by how closely it matches a classical solver’s output, you have capped your ambition at imitation and you’ve hidden whether your errors are physically harmful. Measure the quantity the domain actually cares about.

4. The adoption strategy is part of the contribution. A low-code framework plus released datasets is how you get domain engineers, not ML researchers, to use your work. If your target user cannot reproduce your pipeline in an afternoon, your paper’s impact is bounded by citations.

5. A solver that always answers needs a confidence signal. Treat “more robust than the classical method because it never fails to converge” as an incomplete claim until paired with calibrated uncertainty or anomaly detection. Silent failure is worse than loud failure in operational settings.