Proceedings

Uncertainty-aware Predict-Then-Optimize Framework for Equitable Post-Disaster Power Restoration

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. AI and Social Good. Pages 9719-9727.

https://doi.org/10.24963/ijcai.2025/1080

Abstract

The increasing frequency of extreme weather events, such as hurricanes, highlights the urgent need for efficient and equitable power system restoration. Many electricity providers make restoration decisions primarily based on the volume of power restoration requests from each region. However, our data-driven analysis reveals significant disparities in request submission volume, as disadvantaged communities tend to submit fewer restoration requests. This disparity makes the current restoration solution inequitable, leaving these communities vulnerable to extended power outages. To address this, we aim to propose an equity-aware power restoration strategy that balances both restoration efficiency and equity across communities. However, achieving this goal is challenging for two reasons: the difficulty of predicting repair durations under dataset heteroscedasticity, and the tendency of reinforcement learning agents to favor low-uncertainty actions, which potentially undermine equity. To overcome these challenges, we design a predict-then-optimize framework called EPOPR with two key components: (1) Equity-Conformalized Quantile Regression for uncertainty-aware repair duration prediction, and (2) Spatial-Temporal Attentional RL that adapts to varying uncertainty levels across regions for equitable decision-making. Experimental results show that our EPOPR effectively reduces the average power outage duration by 3.60% and decreases inequity between different communities by 14.19% compared to state-of-the-art baselines.

  1. [Afsharinejad et al., 2021] Amir Hossein Afsharinejad, Chuanyi Ji, and Robert Wilcox. Large-scale data analytics for resilient recovery services from power failures. Joule, 5(9):2504-2520, 2021.
  2. [An et al., 2021] Gaon An, Seungyong Moon, Jang-Hyun Kim, and Hyun Oh Song. Uncertainty-based offline rein- forcement learning with diversified q-ensemble. Advances in neural information processing systems, 34:7436-7447, 2021.
  3. [Bertsekas, 2014] Dimitri P Bertsekas. Constrained opti- mization and Lagrange multiplier methods. Academic press, 2014.
  4. [Chung et al., 2021] Youngseog Chung, Willie Neiswanger, Ian Char, and Jeff Schneider. Beyond pinball loss: Quantile methods for calibrated uncertainty quantifica- tion. Advances in Neural Information Processing Systems, 34:10971-10984, 2021.
  5. [Ez-Zizi et al., 2023] Adnane Ez-Zizi, Simon Farrell, David Leslie, Gaurav Malhotra, and Casimir JH Ludwig. Rein- forcement learning under uncertainty: expected versus un- expected uncertainty and state versus reward uncertainty. Computational Brain & Behavior, 6(4):626-650, 2023.
  6. [Gal and Ghahramani, 2016] Yarin Gal and Zoubin Ghahra- mani. Dropout as a bayesian approximation: Representing model uncertainty in deep learning. In international con- ference on machine learning, pages 1050-1059. PMLR, 2016.
  7. [Ganz et al., 2023] Scott C Ganz, Chenghao Duan, and Chuanyi Ji. Socioeconomic vulnerability and differential impact of severe weather-induced power outages. PNAS nexus, 2(10):pgad295, 2023.
  8. [Haarnoja et al., 2018] Tuomas Haarnoja, Aurick Zhou, Pieter Abbeel, and Sergey Levine. Soft actor-critic: Off-policy maximum entropy deep reinforcement learning with a stochastic actor. In International conference on ma- chine learning, pages 1861-1870. PMLR, 2018.
  9. [Ji et al., 2016] Chuanyi Ji, Yun Wei, Henry Mei, Jorge Calzada, Matthew Carey, Steve Church, Timothy Hayes, Brian Nugent, Gregory Stella, Matthew Wallace, et al. Large-scale data analysis of power grid resilience across multiple us service regions. Nature Energy, 1(5):1-8, 2016.
  10. [Jiang et al., 2023] Lin Jiang, Shuai Wang, Baoshen Guo, Hai Wang, Desheng Zhang, and Guang Wang. Faircod: A fairness-aware concurrent dispatch system for large-scale instant delivery services. In Proceedings of the 29th ACM SIGKDD Conference on knowledge discovery and data mining, pages 4229-4238, 2023. [Kenton and Toutanova, 2019]
  11. Jacob Devlin Ming- Wei Chang Kenton and Lee Kristina Toutanova. Bert: Pre-training of deep bidirectional transformers for lan- guage understanding. In Proceedings of naacL-HLT, volume 1. Minneapolis, Minnesota, 2019.
  12. [Koenker and Bassett Jr, 1978] Roger Koenker and Gilbert Bassett Jr. Regression quantiles. Econometrica: journal of the Econometric Society, pages 33-50, 1978.
  13. [Kull et al., 2019] Meelis Kull, Miquel Perello Nieto, Markus Kängsepp, Telmo Silva Filho, Hao Song, and Peter Flach. Beyond temperature scaling: Obtaining well-calibrated multi-class probabilities with dirichlet calibration. Advances in neural information processing systems, 32, 2019.
  14. [Kumar et al., 2020] Aviral Kumar, Aurick Zhou, George Tucker, and Sergey Levine. Conservative q-learning for offline reinforcement learning. Advances in Neural Infor- mation Processing Systems, 33:1179-1191, 2020.
  15. [Lan et al., 2022] Shiwei Lan, Shuyi Li, and Babak Shah- baba. Scaling up bayesian uncertainty quantification for inverse problems using deep neural networks. SIAM/ASA Journal on Uncertainty Quantification, 10(4):1684-1713, 2022.
  16. [Ma et al., 2021] Yi Ma, Xiaotian Hao, Jianye Hao, Jiawen Lu, Xing Liu, Tong Xialiang, Mingxuan Yuan, Zhigang Li, Jie Tang, and Zhaopeng Meng. A hierarchical reinforce- ment learning based optimization framework for large- scale dynamic pickup and delivery problems. Advances in neural information processing systems, 34:23609-23620, 2021.
  17. [Meinshausen and Ridgeway, 2006] Nicolai Meinshausen and Greg Ridgeway. Quantile regression forests. Journal of machine learning research, 7(6), 2006.
  18. [NHC.gov, 2019] NHC.gov. Tropical cyclone report: Hurri- cane michael (al142018). https://www.nhc.noaa.gov/data/ tcr/AL142018 Michael.pdf, 2019. Link
  19. [Panaretos and Zemel, 2019] Victor M Panaretos and Yoav Zemel. Statistical aspects of wasserstein distances. An- nual review of statistics and its application, 6(1):405-431, 2019.
  20. [Pessach and Shmueli, 2022] Dana Pessach and Erez Shmueli. A review on fairness in machine learning. ACM Computing Surveys (CSUR), 55(3):1-44, 2022.
  21. [Rahaman and others, 2021] Rahul Rahaman et al. Uncer- tainty quantification and deep ensembles. Advances in neural information processing systems, 34:20063-20075, 2021.
  22. [Rasul et al., 2021] Kashif Rasul, Calvin Seward, Ingmar Schuster, and Roland Vollgraf. Autoregressive denois- ing diffusion models for multivariate probabilistic time se- ries forecasting. In International Conference on Machine Learning, pages 8857-8868. PMLR, 2021.
  23. [Romano et al., 2019] Yaniv Romano, Evan Patterson, and Emmanuel Candes. Conformalized quantile regression. Advances in neural information processing systems, 32, 2019.
  24. [Shafer and Vovk, 2008] Glenn Shafer and Vladimir Vovk. A tutorial on conformal prediction. Journal of Machine Learning Research, 9(3), 2008.
  25. [Takeuchi et al., 2006] Ichiro Takeuchi, Quoc V Le, Timo- thy D Sears, Alexander J Smola, and Chris Williams. Non- parametric quantile estimation. Journal of machine learn- ing research, 7(7), 2006.
  26. [Tas ¸et al., 2016] Duygu Tas ¸, Michel Gendreau, Ola Jabali, and Gilbert Laporte. The traveling salesman problem with time-dependent service times. European Journal of Oper- ational Research, 248(2):372-383, 2016. [
  27. [USA.gov, 2024] USA.gov. Hurricane helene. https://www. usa.gov/hurricane-helene, 2024. Link
  28. [Waswani et al., 2017] A Waswani, N Shazeer, N Parmar, J Uszkoreit, L Jones, A Gomez, L Kaiser, and I Polo- sukhin. Attention is all you need. In NIPS, 2017.
  29. [Wen et al., 2023] Haomin Wen, Youfang Lin, Yutong Xia, Huaiyu Wan, Qingsong Wen, Roger Zimmermann, and Yuxuan Liang. Diffstg: Probabilistic spatio-temporal graph forecasting with denoising diffusion models. In Pro- ceedings of the 31st ACM International Conference on Ad- vances in Geographic Information Systems, pages 1-12, 2023.
  30. [White, 1980] Halbert White. A heteroskedasticity- consistent covariance matrix estimator and a direct test for heteroskedasticity. Econometrica: journal of the Econometric Society, pages 817-838, 1980.
  31. [Xu and Tang, 2020] Corey Kewei Xu and Tian Tang. Clos- ing the gap or widening the divide: The impacts of technology-enabled coproduction on equity in public ser- vice delivery. Public administration review, 80(6):962- 975, 2020.
  32. [Yan et al., 2024] Hua Yan, Heng Tan, Haotian Wang, Desh- eng Zhang, and Yu Yang. Robust route planning under un- certain pickup requests for last-mile delivery. In Proceed- ings of the ACM on Web Conference 2024, pages 3022- 3030, 2024.
  33. [Zhang et al., 2021] Zizhen Zhang, Hong Liu, MengChu Zhou, and Jiahai Wang. Solving dynamic traveling sales- man problems with deep reinforcement learning. IEEE Transactions on Neural Networks and Learning Systems, 34(4):2119-2132, 2021.
  34. [Zhuang et al., 2022] Dingyi Zhuang, Shenhao Wang, Haris Koutsopoulos, and Jinhua Zhao. Uncertainty quantifi- cation of sparse travel demand prediction with spatial- temporal graph neural networks. In Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pages 4639-4647, 2022.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards