English

Newton's Lantern: A Reinforcement Learning Framework for Finetuning AC Power Flow Warm Start Models

Machine Learning 2026-05-13 v1 Artificial Intelligence Systems and Control Systems and Control

Abstract

Neural warm starts can sharply reduce the number of Newton-Raphson iterations required to solve the AC power flow problem, but existing supervised approaches generalize poorly on heavily loaded instances near voltage collapse. We prove a lower bound on the Newton-Raphson iteration count that depends on the direction of the warm start error rather than on its magnitude, and show as a corollary that the bound becomes vacuous as the smallest singular value of the power-flow Jacobian shrinks, identifying the failure mode of supervised regression near the saddle-node bifurcation. Motivated by this analysis, we introduce Newton's Lantern, a finetuning pipeline that combines group relative policy optimization with a learned reward model trained on perturbations of the base model's predictions, using the iteration count itself as the supervisory signal. Across IEEE 118-bus, GOC 500-bus, and GOC 2000-bus benchmarks, Newton's Lantern is the only method that converges on every test snapshot while attaining the smallest mean iteration count.

Keywords

Cite

@article{arxiv.2605.11102,
  title  = {Newton's Lantern: A Reinforcement Learning Framework for Finetuning AC Power Flow Warm Start Models},
  author = {Shourya Bose and Helgi Hilmarsson and Dhruv Suri},
  journal= {arXiv preprint arXiv:2605.11102},
  year   = {2026}
}
R2 v1 2026-07-22T07:05:40.973Z