Proceedings

Probabilistic Verification of Recurrent Neural Networks for Single and Multi-Agent Reinforcement Learning

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. Main Track. Pages 262-270.

https://doi.org/10.24963/ijcai.2026/30

Abstract

History-dependent policies induced by recurrent neural networks (RNNs) rely on latent hidden state dynamics, making verification in partially observable reinforcement learning (RL) challenging. Existing RNN verification tools typically rely on restrictive modeling assumptions or coarse over-approximations of the hidden state space, which can lead to overly conservative or inconclusive results. We propose RNN Probabilistic Verification (RNN-ProVe), a probabilistic framework that estimates the likelihood of undesired behaviors in RNN-based policies. RNN-ProVe uses policy-driven sampling to approximate the set of hidden states that are feasible under a trained policy, and derives statistical error bounds to produce bounded-error, high-confidence estimates of behavioral violations. Experiments on partially observable single-agent and cooperative multi-agent tasks show that RNN-ProVe yields more quantitative, feasibility-aware probabilistic guarantees than existing tools, while scaling to recurrent and multi-agent settings.

  1. [Akintunde et al., 2019] Michael E Akintunde, Andreea Kevorchian, Alessio Lomuscio, and Edoardo Pirovano. Verification of rnn-based neural agent-environment sys- tems. In AAAI Conference on Artificial Intelligence, pages 6006-6013, 2019.
  2. [Amato, 2024] Christopher Amato. An introduction to cen- tralized training for decentralized execution in coopera- tive multi-agent reinforcement learning. arXiv preprint arXiv:2409.03052, 2024.
  3. [Åström, 1965] Karl Johan Åström. Optimal control of markov processes with incomplete state information i. Journal of Mathematical Analysis and Applications, 10:174-205, 1965.
  4. [Aydeniz et al., 2024] Ayhan Alp Aydeniz, Enrico March- esini, Christopher Amato, and Kagan Tumer. Entropy seeking constrained multiagent reinforcement learning. In International Conference on Autonomous Agents and Mul- tiagent Systems (AAMAS), page 2141-2143, 2024.
  5. [Aydeniz et al., 2025] Ayhan Alp Aydeniz, Enrico March- esini, Robert Loftin, Christopher Amato, and Kagan Tumer. Safe entropic agents under team constraints. In International Conference on Autonomous Agents and Mul- tiagent Systems (AAMAS), page 2411-2413, 2025.
  6. [Bunel et al., 2020] Rudy Bunel, Jingyue Lu, Ilker Turkaslan, Philip HS Torr, Pushmeet Kohli, and M Pawan Kumar. Branch and bound for piecewise linear neural network verification. Journal of Machine Learning Research, 21(42):1-39, 2020.
  7. [Carr et al., 2021] Steven Carr, Nils Jansen, and Ufuk Topcu. Verifiable rnn-based policies for pomdps under temporal logic constraints. In International Conference on Interna- tional Joint Conferences on Artificial Intelligence, pages 4121-4127, 2021. IJCAI
  8. [Cho et al., 2014] Kyunghyun Cho, Bart Van Merriënboer, Dzmitry Bahdanau, and Yoshua Bengio. On the proper- ties of neural machine translation: Encoder-decoder ap- proaches. arXiv preprint arXiv:1409.1259, 2014.
  9. [Du et al., 2021] Tianyu Du, Shouling Ji, Lujia Shen, Yao Zhang, Jinfeng Li, Jie Shi, Chengfang Fang, Jianwei Yin, Raheem Beyah, and Ting Wang. Cert-rnn: Towards certi- fying the robustness of recurrent neural networks. In ACM SIGSAC Conference on Computer and Communications Security, pages 15-19, 2021.
  10. [Everett et al., 2021] Michael Everett, Golnaz Habibi, Chuangchuang Sun, and Jonathan P How. Reachabil- ity analysis of neural feedback loops. IEEE Access, 9:163938-163953, 2021.
  11. [Hausknecht and Stone, 2017] Matthew Hausknecht and Pe- ter Stone. Deep recurrent q-learning for partially observ- able mdps. arXiv preprint arXiv:1507.06527, 2017.
  12. [Hochreiter and Schmidhuber, 1997] Sepp Hochreiter and Jürgen Schmidhuber. Long short-term memory. Neural computation, 9(8):1735-1780, 1997.
  13. [Hoeffding, 1963] Wassily Hoeffding. Probability inequali- ties for sums of bounded random variables. Journal of the American statistical association, 58(301):13-30, 1963.
  14. [Hornik et al., 1989] Kurt Hornik, Maxwell Stinchcombe, and Halbert White. Multilayer feedforward networks are universal approximators. Neural networks, 2(5):359-366, 1989.
  15. [Jacoby et al., 2020] Yuval Jacoby, Clark Barrett, and Guy Katz. Verifying recurrent neural networks using invari- ant inference. In International Symposium on Automated Technology for Verification and Analysis, pages 57-74. Springer, 2020.
  16. [Kaelbling et al., 1996] Leslie Pack Kaelbling, Michael L Littman, and Andrew W Moore. Reinforcement learn- ing: A survey. Journal of Artificial Intelligence Research, 4:237-285, 1996.
  17. [Katz et al., 2017] Guy Katz, Clark Barrett, David L Dill, Kyle Julian, and Mykel J Kochenderfer. Reluplex: An efficient smt solver for verifying deep neural networks. In International Conference on Computer Aided Verification, pages 97-117, 2017.
  18. [Ko et al., 2019] Ching-Yun Ko, Zhaoyang Lyu, Lily Weng, Luca Daniel, Ngai Wong, and Dahua Lin. Popqorn: Quan- tifying robustness of recurrent neural networks. In Inter- national Conference on Machine Learning, pages 3468- 3477, 2019.
  19. [Kotha et al., 2024] Suhas Kotha, Christopher Brix, J Zico Kolter, Krishnamurthy Dvijotham, and Huan Zhang. Prov- ably bounding neural network preimages. Advances in Neural Information Processing Systems, 2024.
  20. [Liu et al., 2021] Changliu Liu, Tomer Arnon, Christopher Lazarus, Christopher Strong, Clark Barrett, Mykel J Kochenderfer, et al. Algorithms for verifying deep neu- ral networks. Foundations and Trends® in Optimization, 4(3-4):244-404, 2021.
  21. [Marchesini et al., 2023] Enrico Marchesini, Luca Marzari, Alessandro Farinelli, and Christopher Amato. Safe deep reinforcement learning by verifying task-level proper- ties. In Proceedings of the 2023 International Conference on Autonomous Agents and Multiagent Systems, AAMAS 2023, pages 1466-1475. ACM, 2023.
  22. [Marchesini et al., 2025a] Enrico Marchesini, Andrea Bais- ero, Rupali Bhati, and Christopher Amato. On stateful value factorization in multi-agent reinforcement learning. In International Conference on Autonomous Agents and Multiagent Systems (AAMAS), page 1445-1453, 2025.
  23. [Marchesini et al., 2025b] Enrico Marchesini, Benjamin Donnot, Constance Crozier, Ian Dytham, Christian Merz, Lars Schewe, Nico Westerbeck, Cathy Wu, Antoine Marot, and Priya L Donti. Rl2grid: Benchmarking reinforcement learning in power grid operations. arXiv preprint arXiv:2503.23101, 2025.
  24. [Marchesini et al., 2026] Enrico Marchesini, Eva Bogus- lawski, Alessandro Leite, Christopher Amato, Matthieu DUSSARTRE, Marc Schoenauer, Benjamin Donnot, and Priya L. Donti. MARL2grid-TR: A multi-agent RL bench- mark in power grid operations. In The Fourteenth Interna- tional Conference on Learning Representations, 2026.
  25. [Marzari et al., 2023a] Luca Marzari, Davide Corsi, Ferdi- nando Cicalese, and Alessandro Farinelli. The #dnn- verification problem: Counting unsafe inputs for deep neu- ral networks. In International Joint Conference on Artifi- cial Intelligence, IJCAI, pages 217-224, 2023.
  26. [Marzari et al., 2023b] Luca Marzari, Enrico Marchesini, and Alessandro Farinelli. Online safety property collec- tion and refinement for safe deep reinforcement learning in mapless navigation. In IEEE International Conference on Robotics and Automation, (ICRA), pages 7133-7139. IEEE, 2023.
  27. [Marzari et al., 2024] Luca Marzari, Davide Corsi, Enrico Marchesini, Alessandro Farinelli, and Ferdinando Ci- calese. Enumerating safe regions in deep neural networks with provable probabilistic guarantees. In Conference on Artificial Intelligence (AAAI), pages 21387-21394, 2024.
  28. [Marzari et al., 2025a] Luca Marzari, Ferdinando Cicalese, Alessandro Farinelli, Christopher Amato, and Enrico Marchesini. Verifying online safety properties for safe deep reinforcement learning. ACM Trans. Intell. Syst. Technol., September 2025.
  29. [Marzari et al., 2025b] Luca Marzari, Priya L. Donti, Changliu Liu, and Enrico Marchesini. Improving policy optimization via ϱ-retrain. In Proceedings of the 24th International Conference on Autonomous Agents and Multiagent Systems, AAMAS 2025, pages 1464-1472. International Foundation for Autonomous Agents and Multiagent Systems / ACM, 2025.
  30. [Marzari et al., 2025c] Luca Marzari, Isabella Mastroeni, and Alessandro Farinelli. Advancing neural network veri- fication through hierarchical safety abstract interpretation. In ECAI 2025 -28th European Conference on Artificial Intelligence, pages 1736-1743, 2025.
  31. [Marzari et al., 2025d] Luca Marzari, Francesco Trotti, En- rico Marchesini, and Alessandro Farinelli. Designing con- trol barrier function via probabilistic enumeration for safe reinforcement learning navigation. IEEE Robotics and Au- tomation Letters, 10(10):9630-9637, 2025.
  32. [Marzari et al., 2026a] Luca Marzari, Manuele Bicego, Fer- dinando Cicalese, and Alessandro Farinelli. On the prob- abilistic learnability of compact neural network preimage bounds. In Proceedings of the AAAI Conference on Artifi- cial Intelligence, volume 40, pages 35707-35714, 2026.
  33. [Marzari et al., 2026b] Luca Marzari, Changliu Liu, Priya L Donti, and Enrico Marchesini. ϖ-retraining reinforcement learning algorithms. Autonomous Agents and Multi-Agent Systems, 40(1):24, 2026.
  34. [Mohammadinejad et al., 2021] Sara Mohammadinejad, Brandon Paulsen, Jyotirmoy V Deshmukh, and Chao Wang. Diffrnn: differential verification of recurrent neural networks. In International Conference on Formal Modeling and Analysis of Timed Systems, pages 117-134.
  35. [Oliehoek and Amato, 2016] Frans A Oliehoek and Christo- pher Amato. A concise introduction to decentralized POMDPs. Springer, 2016.
  36. [Papoudakis et al., 2021] Georgios Papoudakis, Filippos Christianos, Lukas Schäfer, and Stefano V Albrecht. Benchmarking multi-agent deep reinforcement learning algorithms in cooperative tasks. In Conference on Neural Information Processing Systems Datasets and Benchmarks Track, 2021.
  37. [Tran et al., 2023] Hoang-Dung Tran, Taylor T Johnson, et al. Reachability analysis of recurrent neural networks using star sets. In ACM International Conference on Hy- brid Systems: Computation and Control, 2023.
  38. [Valiant, 1979] Leslie G Valiant. The complexity of com- puting the permanent. Theoretical Computer Science, 8(2):189-201, 1979.
  39. [Wang et al., 2021] Shiqi Wang, Huan Zhang, Kaidi Xu, Xue Lin, Suman Jana, Cho-Jui Hsieh, and J Zico Kolter. Beta-crown: Efficient bound propagation with per-neuron split constraints for neural network robustness verifica- tion. Advances in Neural Information Processing Systems, 34:29909-29921, 2021.
  40. [Wei et al., 2025] Tianhao Wei, Hanjiang Hu, Luca Marzari, Kai S. Yun, Peizhi Niu, Xusheng Luo, and Changliu Liu. Modelverification.jl: A comprehensive toolbox for for- mally verifying deep neural networks. In Computer Aided Verification -37th International Conference, CAV 2025, volume 15932 of Lecture Notes in Computer Science, pages 395-408. Springer, 2025.
  41. [Xiao et al., 2020] Yuchen Xiao, Joshua Hoffman, and Christopher Amato. Macro-action-based deep multi-agent reinforcement learning. In Conference on Robot Learning, 2020.
  42. [Xu et al., 2021] Kaidi Xu, Huan Zhang, Shiqi Wang, Yihan Wang, Suman Jana, Xue Lin, and Cho-Jui Hsieh. Fast and Complete: Enabling complete neural network verifica- tion with rapid and massively parallel incomplete verifiers. In International Conference on Learning Representations, 2021.
  43. [Zhang et al., 2024] Xiyue Zhang, Benjie Wang, and Marta Kwiatkowska. Provable preimage under-approximation for neural networks. In International Conference on Tools and Algorithms for the Construction and Analysis of Sys- tems, pages 3-23. Springer, 2024.

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

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