SPIO: Ensemble and Selective Strategies via LLM-Based Multi-Agent Planning in Automated Data Science

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Seo, Wonduk, Lee, Juhyeon, Shao, Yanjun, Zhou, Qingshan, Lee, Seunghyun, Bu, Yi
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909982859984896
author Seo, Wonduk
Lee, Juhyeon
Shao, Yanjun
Zhou, Qingshan
Lee, Seunghyun
Bu, Yi
author_facet Seo, Wonduk
Lee, Juhyeon
Shao, Yanjun
Zhou, Qingshan
Lee, Seunghyun
Bu, Yi
contents Large Language Models (LLMs) have enabled dynamic reasoning in automated data analytics, yet recent multi-agent systems remain limited by rigid, single-path workflows that restrict strategic exploration and often lead to suboptimal outcomes. To overcome these limitations, we propose SPIO (Sequential Plan Integration and Optimization), a framework that replaces rigid workflows with adaptive, multi-path planning across four core modules: data preprocessing, feature engineering, model selection, and hyperparameter tuning. In each module, specialized agents generate diverse candidate strategies, which are cascaded and refined by an optimization agent. SPIO offers two operating modes: SPIO-S for selecting a single optimal pipeline, and SPIO-E for ensembling top-k pipelines to maximize robustness. Extensive evaluations on Kaggle and OpenML benchmarks show that SPIO consistently outperforms state-of-the-art baselines, achieving an average performance gain of 5.6%. By explicitly exploring and integrating multiple solution paths, SPIO delivers a more flexible, accurate, and reliable foundation for automated data science.
format Preprint
id arxiv_https___arxiv_org_abs_2503_23314
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SPIO: Ensemble and Selective Strategies via LLM-Based Multi-Agent Planning in Automated Data Science
Seo, Wonduk
Lee, Juhyeon
Shao, Yanjun
Zhou, Qingshan
Lee, Seunghyun
Bu, Yi
Artificial Intelligence
Computation and Language
Machine Learning
Multiagent Systems
Large Language Models (LLMs) have enabled dynamic reasoning in automated data analytics, yet recent multi-agent systems remain limited by rigid, single-path workflows that restrict strategic exploration and often lead to suboptimal outcomes. To overcome these limitations, we propose SPIO (Sequential Plan Integration and Optimization), a framework that replaces rigid workflows with adaptive, multi-path planning across four core modules: data preprocessing, feature engineering, model selection, and hyperparameter tuning. In each module, specialized agents generate diverse candidate strategies, which are cascaded and refined by an optimization agent. SPIO offers two operating modes: SPIO-S for selecting a single optimal pipeline, and SPIO-E for ensembling top-k pipelines to maximize robustness. Extensive evaluations on Kaggle and OpenML benchmarks show that SPIO consistently outperforms state-of-the-art baselines, achieving an average performance gain of 5.6%. By explicitly exploring and integrating multiple solution paths, SPIO delivers a more flexible, accurate, and reliable foundation for automated data science.
title SPIO: Ensemble and Selective Strategies via LLM-Based Multi-Agent Planning in Automated Data Science
topic Artificial Intelligence
Computation and Language
Machine Learning
Multiagent Systems
url https://arxiv.org/abs/2503.23314