SELA: Tree-Search Enhanced LLM Agents for Automated Machine Learning

Official implementation for paper SELA: Tree-Search Enhanced LLM Agents for Automated Machine Learning.

SELA is an innovative system that enhances Automated Machine Learning (AutoML) by integrating Monte Carlo Tree Search (MCTS) with LLM-based agents. Traditional AutoML methods often generate low-diversity and suboptimal code, limiting their effectiveness in model selection and ensembling. SELA addresses these challenges by representing pipeline configurations as trees, enabling agents to intelligently explore the solution space and iteratively refine their strategies based on experimental feedback.

For more details, please visit the SELA path.