notesum.ai

Published at October 22

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

cs.LG

cs.AI

Released Date: October 22, 2024

Authors: Yizhou Chi¹, Yizhang Lin², Sirui Hong², Duyi Pan³, Yaying Fei², Guanghao Mei⁴, Bangbang Liu², Tianqi Pang⁵, Jacky Kwok⁶, Ceyao Zhang⁷, Bang Liu⁸, Chenglin Wu²

Aff.: ¹DeepWisdom, University of California, Berkeley; ²DeepWisdom; ³The Hong Kong University of Science and Technology (Guangzhou); ⁴University of California, San Diego; ⁵South China Normal University; ⁶Stanford University; ⁷The Chinese University of Hong Kong, Shenzhen; ⁸Universite de Montréal & Mila

Arxiv: https://arxiv.org/abs/2410.17238v1

Method	Wins	Losses	Top 1	Avg. NS % $\uparrow$	Avg. Best NS % $\uparrow$	Avg. Rank $\downarrow$	Avg. Best Rank $\downarrow$
AutoGluon	7	13	4	53.2	53.2	4.4	4.4
AutoSklearn	5	15	5	46.1	47.5	7.6	6.1
AIDE	5	15	2	47.1	51.8	7.8	5.3
Data Interpreter	4	16	2	47.4	50.2	8.8	6.4
\methodname	-	-	7	53.3	54.7	4.8	2.7