Combining LLMs with a Logic-Based Framework to Explain MCTS

In response to the lack of trust in Artificial Intelligence (AI) for sequential planning, we design a Computational Tree Logic-guided large language model (LLM)-based natural language explanation framework designed for the Monte Carlo Tree Search (MCTS) algorithm. MCTS is often considered challenging to interpret due to the complexity of its search trees, but our framework is flexible enough to handle a wide range of free-form post-hoc queries and knowledge-based inquiries centered around MCTS and the Markov Decision Process (MDP) of the application domain. By transforming user queries into logic and variable statements, our framework ensures that the evidence obtained from the search tree remains factually consistent with the underlying environmental dynamics and any constraints in the actual stochastic control process. We evaluate the framework rigorously through quantitative assessments, where it demonstrates strong performance in terms of accuracy and factual consistency.

Authors

• Ziyan An Vanderbilt University
• Xia Wang
• Hendrik Baier Eindhoven University of Technology
• Zirong Chen Vanderbilt University
• Abhishek Dubey Vanderbilt University
• Taylor T. Johnson Vanderbilt University
• Jonathan Sprinkle Vanderbilt University
• Ayan Mukhopadhyay William and Mary
• Meiyi Ma Vanderbilt University

Keywords

• Sequential Planning
• Explainable AI
• Large Language Model
• MCTS