Proceedings

Rule-Bottleneck RL: Learning to Decide and Explain for Sequential Resource Allocation via LLM Agents in Public Health

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. AI and Social Good. Pages 7512-7520.

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

Abstract

Reducing preventable maternal mortality remains a global health priority. Under Sustainable Development Goal (SDG) target 3.1, the WHO emphasizes timely and equitable allocation of limited maternal health resources. Motivated by Department of Obstetrics and Gynecology at several important hospitals in Uganda and Ghana, we study the problem of sequential allocation of wearable vital sign monitoring devices among maternal mothers. While deep reinforcement learning (RL) has shown promise for sequential resource allocation, its limited interpretability hinders adoption in such high-stakes settings. In contrast, large language model (LLM) agents provide human-readable reasoning but often struggle with effective long-term decision making. To bridge this gap, we introduce Rule-Bottleneck RL (RBRL), the first LLM agent framework for resource allocation problems that jointly optimizes language-based decision policy and explainability. At each step within RBRL, an LLM first generates candidate rules---language statements capturing decision priorities tailored to the current state. RL then optimizes rule selection to maximize environmental rewards and explainability, with the LLM acting as a judge. Finally, an LLM chooses the action (optimal allocation) based on the rule. We provide conditions for RBRL performance guarantees as well as the finite-horizon evaluation gap of the learned RBRL policy. Experiments in maternal health show that RBRL outperforms baseline LLM agents and approaches the performance of deep RL, while producing clearer, policy-relevant explanations. Human evaluations further confirm improved trust and usability, demonstrating RBRL as a practical AI approach aligned with SDG target 3.1.

  1. [Bahdanau et al., 2015] Dzmitry Bahdanau, Kyunghyun Cho, and Yoshua Bengio. Neural machine translation by jointly learning to align and translate. In ICLR, 2015.
  2. [Balaji et al., 2019] Bharathan Balaji, Jordan Bell-Masterson, Enes Bilgin, Andreas Damianou, Pablo Moreno Garcia, Arpit Jain, Runfei Luo, Alvaro Maggiar, Balakrishnan Narayanaswamy, and Chun Ye. Orl: Reinforcement learn- ing benchmarks for online stochastic optimization prob- lems. arXiv preprint arXiv:1911.10641, 2019.
  3. [Bhattacharjee et al., 2024] Amrita Bhattacharjee, Raha Moraffah, Joshua Garland, and Huan Liu. Towards llm- guided causal explainability for black-box text classifiers. In AAAI 2024 Workshop on Responsible Language Models, 2024.
  4. [Boatin et al., 2021] Adeline A Boatin, Joseph Ngonzi, Blair J Wylie, Henry M Lugobe, Lisa M Bebell, God- frey Mugyenyi, Sudi Mohamed, Kenia Martinez, Nicholas Musinguzi, Christina Psaros, et al. Wireless versus rou- tine physiologic monitoring after cesarean delivery to reduce maternal morbidity and mortality in a resource- limited setting: protocol of type 2 hybrid effectiveness- implementation study. BMC Pregnancy and Childbirth, 21:1-12, 2021.
  5. [Boehmer et al., 2024] Niclas Boehmer, Yunfan Zhao, Guo- jun Xiong, Paula Rodriguez-Diaz, Paola Del Cueto Cib- rian, Joseph Ngonzi, Adeline Boatin, and Milind Tambe. Optimizing vital sign monitoring in resource-constrained maternal care: An rl-based restless bandit approach. arXiv preprint arXiv:2410.08377, 2024.
  6. [Carta et al., 2023] Thomas Carta, Clément Romac, Thomas Wolf, Sylvain Lamprier, Olivier Sigaud, and Pierre-Yves Oudeyer. Grounding large language models in interactive environments with online reinforcement learning. In ICML, 2023.
  7. [Chen et al., 2024] Haozhe Chen, Carl Vondrick, and Chengzhi Mao. Selfie: Self-interpretation of large language model embeddings. arXiv preprint arXiv:2403.10949, 2024.
  8. [Chevalier-Boisvert et al., 2018] Maxime Chevalier- Boisvert, Dzmitry Bahdanau, Salem Lahlou, Lucas Willems, Chitwan Saharia, Thien Huu Nguyen, and Yoshua Bengio. Babyai: A platform to study the sample efficiency of grounded language learning. arXiv preprint arXiv:1810.08272, 2018.
  9. [Considine et al., 2025] Ellen M Considine, Rachel C Neth- ery, Gregory A Wellenius, Francesca Dominici, and Mauri- cio Tec. Optimizing heat alert issuance with reinforcement learning. In AAAI, 2025.
  10. [Du et al., 2023] Yuqing Du, Olivia Watkins, Zihan Wang, Cédric Colas, Trevor Darrell, Pieter Abbeel, Abhishek Gupta, and Jacob Andreas. Guiding pretraining in rein- forcement learning with large language models. In ICML, pages 8657-8677, 2023.
  11. [Furuta et al., 2024] Hiroki Furuta, Yutaka Matsuo, Aleksan- dra Faust, and Izzeddin Gur. Exposing limitations of lan- guage model agents in sequential-task compositions on the web. In ICLR 2024 Workshop on Large Language Model (LLM) Agents, 2024.
  12. [Gu et al., 2024] Jiawei Gu, Xuhui Jiang, Zhichao Shi, Hexi- ang Tan, Xuehao Zhai, Chengjin Xu, Wei Li, Yinghan Shen, Shengjie Ma, Honghao Liu, et al. A survey on llm-as-a- judge. arXiv preprint arXiv:2411.15594, 2024.
  13. [Guo et al., 2021] Wenbo Guo, Xian Wu, Usmann Khan, and Xinyu Xing. Edge: Explaining deep reinforcement learn- ing policies. Advances in Neural Information Processing Systems, 34:12222-12236, 2021.
  14. [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 Proceedings of the 35th Interna- tional Conference on Machine Learning, pages 1856-1865. PMLR, 2018.
  15. [Hu and Sadigh, 2023] Hengyuan Hu and Dorsa Sadigh. Lan- guage instructed reinforcement learning for human-ai coor- dination. In ICML, 2023.
  16. [Hu et al., 2021] Edward J Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, and Weizhu Chen. Lora: Low-rank adaptation of large language models. arXiv preprint arXiv:2106.09685, 2021.
  17. [Jacovi and Goldberg, 2020] Alon Jacovi and Yoav Goldberg. Towards faithfully interpretable nlp systems: How should we define and evaluate faithfulness? arXiv preprint arXiv:2004.03685, 2020.
  18. [Johnson et al., 2016] Alistair EW Johnson, Tom J Pollard, Lu Shen, Li-wei H Lehman, Mengling Feng, Moham- mad Ghassemi, Benjamin Moody, Peter Szolovits, Leo Anthony Celi, and Roger G Mark. Mimic-iii, a freely ac- cessible critical care database. Scientific data, 3(1):1-9, 2016.
  19. [Kunz and Kuhlmann, 2024] Jenny Kunz and Marco Kuhlmann. Properties and challenges of llm-generated explanations. arXiv preprint arXiv:2402.10532, 2024.
  20. [Meta AI, 2024] Meta AI. Introducing llama 3.1: Our most capable models to date. AI at Meta, 2024.
  21. [Milani et al., 2024] Stephanie Milani, Nicholay Topin, Manuela Veloso, and Fei Fang. Explainable reinforce- ment learning: A survey and comparative review. ACM Computing Surveys, 56(7):1-36, 2024.
  22. [OpenAI, 2024] OpenAI. Gpt-4o mini: Advancing cost- efficient intelligence. OpenAI Blog, 2024.
  23. [Parcalabescu and Frank, 2023] Letitia Parcalabescu and Anette Frank. On measuring faithfulness or self- consistency of natural language explanations. arXiv preprint arXiv:2311.07466, 2023.
  24. [Peng et al., 2022] Xiangyu Peng, Mark Riedl, and Prithviraj Ammanabrolu. Inherently explainable reinforcement learn- ing in natural language. In Alice H. Oh, Alekh Agarwal, Danielle Belgrave, and Kyunghyun Cho, editors, NeurIPS, 2022.
  25. [Rashid et al., 2024] Ahmad Rashid, Ruotian Wu, Julia Grosse, Agustinus Kristiadi, and Pascal Poupart. A critical look at tokenwise reward-guided text generation. arXiv preprint arXiv:2406.07780, 2024.
  26. [Schulman et al., 2017] John Schulman, Filip Wolski, Pra- fulla Dhariwal, Alec Radford, and Oleg Klimov. Prox- imal policy optimization algorithms. arXiv preprint arXiv:1707.06347, 2017. [SDG, 2026] SDG. Sdg target 3.1: Maternal mortal- ity. https://www.who.int/data/gho/data/themes/topics/ sdg-target-3-1-maternal-mortality, 2026. Accessed: Jan 2026. Link
  27. [Shao et al., 2024] Zhihong Shao, Peiyi Wang, Qihao Zhu, Runxin Xu, Junxiao Song, Xiao Bi, Haowei Zhang, Mingchuan Zhang, YK Li, Y Wu, et al. Deepseekmath: Pushing the limits of mathematical reasoning in open lan- guage models. arXiv preprint arXiv:2402.03300, 2024.
  28. [Shen et al., 2024] Yiran Shen, Aditya Emmanuel Arokiaraj John, and Brandon Fain. Explainable rewards in RLHF using LLM-as-a-judge, 2024.
  29. [Silva et al., 2020] Andrew Silva, Matthew Gombolay, Tay- lor Killian, Ivan Jimenez, and Sung-Hyun Son. Optimiza- tion methods for interpretable differentiable decision trees applied to reinforcement learning. In International confer- ence on artificial intelligence and statistics, pages 1855- 1865. PMLR, 2020.
  30. [Srivastava et al., 2024] Megha Srivastava, Cédric Colas, Dorsa Sadigh, and Jacob Andreas. Policy learning with a language bottleneck. In RLC Workshop on Training Agents with Foundation Models, 2024.
  31. [Sumers et al., 2024] Theodore Sumers, Shunyu Yao, Karthik Narasimhan, and Thomas Griffiths. Cognitive architectures for language agents. TMLR, 2024.
  32. [Szot et al., 2023] Andrew Szot, Max Schwarzer, Harsh Agrawal, Bogdan Mazoure, Rin Metcalf, Walter Talbott, Natalie Mackraz, R Devon Hjelm, and Alexander T To- shev. Large language models as generalizable policies for embodied tasks. In ICLR, 2023.
  33. [Talaat, 2022] Fatma M Talaat. Effective deep q-networks (edqn) strategy for resource allocation based on optimized reinforcement learning algorithm. Multimedia Tools and Applications, 81(28):39945-39961, 2022.
  34. [Towers et al., 2024] Mark Towers, Ariel Kwiatkowski, Jor- dan K Terry, John U Balis, Gianluca De Cola, Tristan Deleu, Manuel Goulão, Andreas Kallinteris, Markus Krim- mel, Arjun KG, et al. Gymnasium: A standard interface for reinforcement learning environments. arXiv preprint arXiv:2407.17032, 2024.
  35. [Vaswani et al., 2017] Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Kukasz Kaiser, and Illia Polosukhin. Attention is all you need. In Advances in Neural Information Processing Sys- tems, volume 30, 2017.
  36. [Wei et al., 2022] Jason Wei, Xuezhi Wang, Dale Schuur- mans, Maarten Bosma, Fei Xia, Ed Chi, Quoc V Le, Denny Zhou, et al. Chain-of-thought prompting elicits reasoning in large language models. Advances in neural information processing systems, 35:24824-24837, 2022.
  37. [Wen et al., 2024] Muning Wen, Ziyu Wan, Jun Wang, Weinan Zhang, and Ying Wen. Reinforcing LLM agents via policy optimization with action decomposition. In The Thirty-eighth Annual Conference on Neural Information Processing Systems, 2024.
  38. [Whittle, 1988] Peter Whittle. Restless bandits: Activity allo- cation in a changing world. Journal of applied probability, 25(A):287-298, 1988.
  39. [Wolf et al., 2020] Thomas Wolf, Lysandre Debut, Victor Sanh, Julien Chaumond, Clement Delangue, Anthony Moi, Pierric Cistac, Tim Rault, Rémi Louf, Morgan Funtowicz, Joe Davison, Sam Shleifer, Patrick von Platen, Clara Ma, Yacine Jernite, Julien Plu, Canwen Xu, Teven Le Scao, Sylvain Gugger, Mariama Drame, Quentin Lhoest, and Alexander M. Rush. Transformers: State-of-the-art natural language processing. In Proceedings of the 2020 Confer- ence on Empirical Methods in Natural Language Process- ing: System Demonstrations, pages 38-45, Online, October 2020. Association for Computational Linguistics.
  40. [Yao et al., 2023] Shunyu Yao, Jeffrey Zhao, Dian Yu, Nan Du, Izhak Shafran, Karthik R Narasimhan, and Yuan Cao. React: Synergizing reasoning and acting in language mod- els. In ICLR, 2023.
  41. [Yu et al., 2021] Chao Yu, Jiming Liu, Shamim Nemati, and Guosheng Yin. Reinforcement learning in healthcare: A survey. ACM Computing Surveys (CSUR), 55(1):1-36, 2021.
  42. [Zhai et al., 2024] Yuexiang Zhai, Hao Bai, Zipeng Lin, Jiayi Pan, Shengbang Tong, Yifei Zhou, Alane Suhr, Saining Xie, Yann LeCun, Yi Ma, et al. Fine-tuning large vision- language models as decision-making agents via reinforce- ment learning. arXiv preprint arXiv:2405.10292, 2024.
  43. [Zhang et al., 2023] Xijia Zhang, Yue Guo, Simon Stepput- tis, Katia Sycara, and Joseph Campbell. Understanding your agent: Leveraging large language models for behavior explanation. arXiv preprint arXiv:2311.18062, 2023.

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

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