# BOOST-RPF: Boosted Sequential Trees for Radial Power Flow

arXiv: https://arxiv.org/abs/2603.21977 · HTML: https://mlsanity.com/paper/2603.21977/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 11 of its 30 references that are in the map.
- First posted on arXiv 23 Mar 2026.

## Builds on (most cited first, 11 of 11)

- [XGBoost: A Scalable Tree Boosting System](https://mlsanity.com/paper/1603.02754/) (2016)
- [Physics-Guided Deep Neural Networks for Power Flow Analysis](https://mlsanity.com/paper/2002.00097/) (2020)
- [PowerFlowNet: Power Flow Approximation Using Message Passing Graph Neural Networks](https://mlsanity.com/paper/2311.03415/) (2023)
- [Power Flow Balancing with Decentralized Graph Neural Networks](https://mlsanity.com/paper/2111.02169/) (2021)
- [PowerGraph: A power grid benchmark dataset for graph neural networks](https://mlsanity.com/paper/2402.02827/) (2024)
- [Fast Calculation of Probabilistic Power Flow: A Model-based Deep Learning Approach](https://mlsanity.com/paper/1906.06017/) (2019)
- [Data-Driven Power Flow Linearization: A Regression Approach](https://mlsanity.com/paper/1710.10621/) (2017)
- [Solving AC Power Flow with Graph Neural Networks under Realistic Constraints](https://mlsanity.com/paper/2204.07000/) (2022)
- [Load-Flow in Multiphase Distribution Networks: Existence, Uniqueness, Non-Singularity and Linear Models](https://mlsanity.com/paper/1702.03310/) (2017)
- [Assessing the Optimality of LinDist3Flow for Optimal Tap Selection of Step Voltage Regulators in Unbalanced Distribution Networks](https://mlsanity.com/paper/2210.08550/) (2022)
- [Physics-Informed Inductive Biases for Voltage Prediction in Distribution Grids](https://mlsanity.com/paper/2509.25158/) (2025)

## Abstract (arXiv)

Accurate power flow analysis is critical for modern distribution systems, yet classical solvers face scalability issues, and current machine learning models often struggle with generalization. We introduce BOOST-RPF, a novel method that reformulates voltage prediction from a global graph regression task into a sequential path-based learning problem. By decomposing radial networks into root-to-leaf paths, we leverage gradient-boosted decision trees (XGBoost) to model local voltage-drop regularities. We evaluate three architectural variants: Absolute Voltage, Parent Residual, and Physics-Informed Residual. This approach aligns the model architecture with the recursive physics of power flow, ensuring size-agnostic application and superior out-of-distribution robustness. Benchmarked against the Kerber Dorfnetz grid and the ENGAGE suite, BOOST-RPF achieves state-of-the-art results with its Parent Residual variant which consistently outperforms both analytical and neural baselines in standard accuracy and generalization tasks. While global Multi-Layer Perceptrons (MLPs) and Graph Neural Networks (GNNs) often suffer from performance degradation under topological shifts, BOOST-RPF maintains high precision across unseen feeders. Furthermore, the framework displays linear O(N) computational scaling and significantly increased sample efficiency through per-edge supervision, offering a scalable and generalizable alternative for real-time distribution system operator (DSO) applications.
