Proceedings

LLM-Guided Monte Carlo Tree Search over Knowledge Graphs: Composing Mechanistic Explanations for Drug–Disease Pairs

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. AI and Health. Pages 6752-6760.

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

Abstract

Extracting multi-step explanations from knowledge graphs poses a combinatorial challenge requiring both heuristic guidance (as candidates proliferate with depth) and credit assignment (as path quality emerges over extended sequences). Frontier LLMs, strong on knowledge/reasoning benchmarks, offer a compelling source of such heuristics, yet their knowledge comes sans guarantees and compositional performance degrades as chains lengthen. We thus present TESSERA, a 3-part neuro-symbolic framework that uses LLMs in a circumscribed role: for local discriminative judgement rather than autonomous multi-step generation; the knowledge graph then defines the hypothesis space enforcing hard structural constraints, and MCTS coordinates the long-horizon search with principled credit assignment via backpropagation. LLMs perform dual roles as a prior policy biasing exploration and a comparative state evaluator supplying reward signals. Evaluation on drug mechanism elucidation across two complementary knowledge graphs demonstrates fidelity to curated biology while surfacing coherent alternative mechanisms, with ablations confirming discriminative contribution from both LLM components. Beyond its current application, our framework offers a general paradigm for compositional reasoning over structured knowledge.

  1. [Betz et al., 2022] Patrick Betz, Christian Meilicke, and Heiner Stuckenschmidt. Adversarial Explanations for Knowledge Graph Embeddings. volume 4, pages 2820- 2826, July 2022. IJCAI
  2. [Chou et al., 2022] Yu-Liang Chou, Catarina Moreira, Peter Bruza, Chun Ouyang, and Joaquim Jorge. Counterfac- tuals and causability in explainable artificial intelligence: Theory, algorithms, and applications. Information Fusion, 81:59-83, May 2022.
  3. [Das et al., 2018] Rajarshi Das, Shehzaad Dhuliawala, Manzil Zaheer, Luke Vilnis, Ishan Durugkar, Akshay Krishnamurthy, Alex Smola, and Andrew McCallum. Go for a Walk and Arrive at the Answer: Reasoning Over Paths in Knowledge Bases using Reinforcement Learning, December 2018. arXiv:1711.05851 [cs].
  4. [Dziri et al., 2023] Nouha Dziri, Ximing Lu, Melanie Sclar, Xiang Lorraine Li, Liwei Jiang, Bill Yuchen Lin, Pe- ter West, Chandra Bhagavatula, Ronan Le Bras, Jena D. Hwang, Soumya Sanyal, Sean Welleck, Xiang Ren, Allyson Ettinger, Zaid Harchaoui, and Yejin Choi. Faith and fate: limits of transformers on compositionality. In Proceedings of the 37th International Conference on Neu- ral Information Processing Systems, NIPS '23, pages 70293-70332, Red Hook, NY, USA, December 2023. Cur- ran Associates Inc.
  5. [d'Aquin, 2025] Mathieu d'Aquin. On the role of knowledge graphs in AI-based scientific discovery. Journal of Web Semantics, 84:100854, January 2025.
  6. [Godfrey et al., 2025] Charles Godfrey, Ping Nie, Natalia Ostapuk, David Ken, Shang Gao, and Souheil Inati. Lik- ert or Not: LLM Absolute Relevance Judgments on Fine- Grained Ordinal Scales, May 2025. arXiv:2505.19334 [cs].
  7. [Gonzalez-Cavazos et al., 2023] Adriana Carolina Gonzalez-Cavazos, Anna Tanska, Michael Mayers, Denise Carvalho-Silva, Brindha Sridharan, Patrick A. Rewers, Umasri Sankarlal, Lakshmanan Jagannathan, and Andrew I. Su. DrugMechDB: A Curated Database of Drug Mechanisms. Scientific Data, 10(1):632, September 2023. Publisher: Nature Publishing Group.
  8. [Huang et al., 2025] Lei Huang, Weijiang Yu, Weitao Ma, Weihong Zhong, Zhangyin Feng, Haotian Wang, Qian- glong Chen, Weihua Peng, Xiaocheng Feng, Bing Qin, and Ting Liu. A Survey on Hallucination in Large Lan- guage Models: Principles, Taxonomy, Challenges, and Open Questions. ACM Trans. Inf. Syst., 43(2):42:1-42:55, January 2025.
  9. [Kambhampati et al., 2024] Subbarao Kambhampati, Karthik Valmeekam, Lin Guan, Mudit Verma, Kaya Stechly, Siddhant Bhambri, Lucas Paul Saldyt, and Anil B. Murthy. Position: LLMs Can't Plan, But Can Help Planning in LLM-Modulo Frameworks. In Proceed- ings of the 41st International Conference on Machine Learning, pages 22895-22907.
  10. PMLR, July 2024. ISSN: 2640-3498.
  11. [Kambhampati, 2023] Subbarao Kambhampati. Avenging polanyi's revenge: Exploiting the approximate omni- science of llms in planning without deluding yourself in the process. Invited talk, Knowledge and Logical Reason- ing in the Era of Data-driven Learning Workshop, Inter- national Conference on Machine Learning (ICML 2023), July 2023.
  12. [Knox et al., 2024] Craig Knox, Mike Wilson, Christen M Klinger, Mark Franklin, Eponine Oler, Alex Wilson, Al- lison Pon, Jordan Cox, Na Eun (Lucy) Chin, Seth A Strawbridge, Marysol Garcia-Patino, Ray Kruger, Aad- havya Sivakumaran, Selena Sanford, Rahil Doshi, Nitya Khetarpal, Omolola Fatokun, Daphnee Doucet, Ash- ley Zubkowski, Dorsa Yahya Rayat, Hayley Jackson, Karxena Harford, Afia Anjum, Mahi Zakir, Fei Wang, Siyang Tian, Brian Lee, Jaanus Liigand, Harrison Peters, Ruo Qi (Rachel) Wang, Tue Nguyen, Denise So, Matthew Sharp, Rodolfo da Silva, Cyrella Gabriel, Joshua Scant- lebury, Marissa Jasinski, David Ackerman, Timothy Jew- ison, Tanvir Sajed, Vasuk Gautam, and David S Wishart. DrugBank 6.0: the DrugBank Knowledgebase for 2024. Nucleic Acids Research, 52(D1):D1265-D1275, January 2024.
  13. [Koo and Li, 2016] Terry K. Koo and Mae Y. Li. A Guide- line of Selecting and Reporting Intraclass Correlation Co- efficients for Reliability Research. Journal of Chiropractic Medicine, 15(2):155-163, June 2016.
  14. [Langley, 2024] Pat Langley. Integrated Systems for Com- putational Scientific Discovery. Proceedings of the AAAI Conference on Artificial Intelligence, 38(20):22598- 22606, March 2024.
  15. [Liu et al., 2021] Yushan Liu, Marcel Hildebrandt, Mitchell Joblin, Martin Ringsquandl, Rime Raissouni, and Volker Tresp. Neural Multi-hop Reasoning with Logical Rules on Biomedical Knowledge Graphs. In The Semantic Web: 18th International Conference, ESWC 2021, Virtual Event, June 6-10, 2021, Proceedings, pages 375-391, Berlin, Heidelberg, June 2021. Springer-Verlag.
  16. [Liu et al., 2024] Nelson F. Liu, Kevin Lin, John Hewitt, Ashwin Paranjape, Michele Bevilacqua, Fabio Petroni, and Percy Liang. Lost in the Middle: How Language Mod- els Use Long Contexts. Transactions of the Association for Computational Linguistics, 12:157-173, 2024.
  17. [Nunes et al., 2025] Susana Nunes, Samy Badreddine, and Catia Pesquita. Rewarding explainability in drug repurpos- ing with knowledge graphs. In James Kwok, editor, Pro- ceedings of the thirty-fourth international joint conference on artificial intelligence, IJCAI-25, pages 4624-4632. In- ternational Joint Conferences on Artificial Intelligence Or- ganization, August 2025. IJCAI
  18. [Phan et al., 2025] Long Phan, Alice Gatti, Ziwen Han, Nathaniel Li, et al. Humanity's Last Exam, February 2025. arXiv:2501.14249 [cs] version: 2.
  19. [Qin et al., 2024] Zhen Qin, Rolf Jagerman, Kai Hui, Hon- glei Zhuang, Junru Wu, Le Yan, Jiaming Shen, Tianqi Liu, Jialu Liu, Donald Metzler, Xuanhui Wang, and Michael Bendersky. Large Language Models are Effective Text Rankers with Pairwise Ranking Prompting. In Kevin Duh, Helena Gomez, and Steven Bethard, editors, Findings of the Association for Computational Linguistics: NAACL 2024, pages 1504-1518, Mexico City, Mexico, June 2024. Association for Computational Linguistics.
  20. [Rein et al., 2023] David Rein, Betty Li Hou, Asa Cooper Stickland, Jackson Petty, Richard Yuanzhe Pang, Julien Dirani, Julian Michael, and Samuel R. Bowman. GPQA: A Graduate-Level Google-Proof Q&A Benchmark, Novem- ber 2023. arXiv:2311.12022 [cs].
  21. [Rosin, 2011] Christopher D. Rosin. Multi-armed bandits with episode context. Annals of Mathematics and Artifi- cial Intelligence, 61(3):203-230, March 2011.
  22. [Rossi et al., 2022] Andrea Rossi, Donatella Firmani, Paolo Merialdo, and Tommaso Teofili. Kelpie: an explainability framework for embedding-based link prediction models. Proc. VLDB Endow., 15(12):3566-3569, August 2022.
  23. [Ruiz et al., 2021] Camilo Ruiz, Marinka Zitnik, and Jure Leskovec. Identification of disease treatment mechanisms through the multiscale interactome. Nature Communica- tions, 12(1):1796, March 2021. Publisher: Nature Pub- lishing Group.
  24. [Silver et al., 2017] David Silver, Julian Schrittwieser, Karen Simonyan, Ioannis Antonoglou, Aja Huang, Arthur Guez, Thomas Hubert, Lucas Baker, Matthew Lai, Adrian Bolton, Yutian Chen, Timothy Lillicrap, Fan Hui, Laurent Sifre, George van den Driessche, Thore Graepel, and Demis Hassabis. Mastering the game of Go without human knowledge. Nature, 550(7676):354-359, October 2017. Publisher: Nature Publishing Group.
  25. [Simon, 1966] Herbert A. Simon. Scientific discovery and the psychology of problem solving. In Robert G. Colodny, editor, Mind and Cosmos: Essays in Contemporary Sci- ence and Philosophy, pages 22-40. University of Pitts- burgh Press, 1966.
  26. [Song et al., 2025] Haerin Song, Dongmin Bang, Bonil Koo, Sun Kim, and Sangseon Lee. LLM-integrated representa- tive path selection for context-aware drug repurposing on biomedical knowledge graphs. In NeurIPS 2025 Workshop on AI Virtual Cells and Instruments: A New Era in Drug Discovery and Development, 2025.
  27. [Sosa and Altman, 2022] Daniel N. Sosa and Russ B. Alt- man. Contexts and contradictions: a roadmap for com- putational drug repurposing with knowledge inference. Briefings in Bioinformatics, 23(4):bbac268, July 2022.
  28. [Stork et al., 2023] Lise Stork, Ilaria Tiddi, René Spijker, and Annette ten Teije. Explainable Drug Repurposing in Context via Deep Reinforcement Learning: 20th Inter- national Conference on The Semantic Web, ESWC 2023. The Semantic Web, pages 3-20, 2023. Publisher: Springer Science and Business Media Deutschland GmbH.
  29. [Sun et al., 2023] Weiwei Sun, Lingyong Yan, Xinyu Ma, Shuaiqiang Wang, Pengjie Ren, Zhumin Chen, Dawei Yin, and Zhaochun Ren. Is ChatGPT Good at Search? Investi- gating Large Language Models as Re-Ranking Agents. In Houda Bouamor, Juan Pino, and Kalika Bali, editors, Pro- ceedings of the 2023 Conference on Empirical Methods in Natural Language Processing, pages 14918-14937, Sin- gapore, December 2023. Association for Computational Linguistics.
  30. [Wang et al., 2024] Yubo Wang, Xueguang Ma, Ge Zhang, Yuansheng Ni, Abhranil Chandra, Shiguang Guo, Weim- ing Ren, Aaran Arulraj, Xuan He, Ziyan Jiang, Tianle Li, Max Ku, Kai Wang, Alex Zhuang, Rongqi Fan, Xiang Yue, and Wenhu Chen. MMLU-Pro: A More Robust and Chal- lenging Multi-Task Language Understanding Benchmark, November 2024. arXiv:2406.01574 [cs].
  31. [Ying et al., 2019] Rex Ying, Dylan Bourgeois, Jiaxuan You, Marinka Zitnik, and Jure Leskovec. GNNEx- plainer: Generating Explanations for Graph Neural Net- works. Advances in neural information processing sys- tems, 32:9240-9251, December 2019.

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

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