I-MCTS: Enhancing Agentic AutoML via Introspective Monte Carlo Tree Search

Zujie Liang, Feng Wei, Wujiang Xu, Yuxi Qian, Lin Chen, Xinhui Wu


Abstract
Recent advancements in large language models (LLMs) have shown remarkable potential in automating machine learning tasks. However, existing LLM-based agents often struggle with low diversity and suboptimal code generation. While recent work (CITATION) has introduced Monte Carlo Tree Search (MCTS) to address these issues, limitations persist in the quality and diversity of thoughts generated, as well as in the scalar value feedback mechanisms used for node selection. In this study, we introduce Introspective Monte Carlo Tree Search (I-MCTS), a novel approach that iteratively expands tree nodes through an introspective process that meticulously analyzes solutions and results from parent and sibling nodes. This facilitates a continuous refinement of the node in the search tree, thereby enhancing the overall decision-making process. Furthermore, we integrate a Large Language Model (LLM)-based value model to facilitate direct evaluation of each node’s solution prior to conducting comprehensive computational rollouts. A hybrid rewarding mechanism is implemented to seamlessly transition the Q-value from estimated score to actual performance scores. Applied to the various ML tasks, our approach demonstrates a 4% absolute improvement in performance compared to the strong open-source AutoML agents, showcasing its effectiveness in enhancing agentic AutoML systems. Resource available at https://github.com/jokieleung/I-MCTS
Anthology ID:
2026.findings-eacl.11
Volume:
Findings of the Association for Computational Linguistics: EACL 2026
Month:
March
Year:
2026
Address:
Rabat, Morocco
Editors:
Vera Demberg, Kentaro Inui, Lluís Marquez
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
189–210
Language:
URL:
https://preview.aclanthology.org/ingest-eacl/2026.findings-eacl.11/
DOI:
Bibkey:
Cite (ACL):
Zujie Liang, Feng Wei, Wujiang Xu, Yuxi Qian, Lin Chen, and Xinhui Wu. 2026. I-MCTS: Enhancing Agentic AutoML via Introspective Monte Carlo Tree Search. In Findings of the Association for Computational Linguistics: EACL 2026, pages 189–210, Rabat, Morocco. Association for Computational Linguistics.
Cite (Informal):
I-MCTS: Enhancing Agentic AutoML via Introspective Monte Carlo Tree Search (Liang et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-eacl/2026.findings-eacl.11.pdf
Checklist:
 2026.findings-eacl.11.checklist.pdf