mlsanityData to 14 Oct 2026

SAGE-Loop: Reliable Closed-Loop LLM-Driven AutoML with Trial-and-Correction and Adaptive Ensembling

arXiv 2609.12455 ↗First posted 11 Sep 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Automated machine learning (AutoML) is reshaping data-driven science and industrial practice, and as large language models are introduced into AutoML, pipeline reliability becomes as important as automation efficiency. However, existing AutoML still struggles to realize instant feedback and adaptive optimization during execution, so once a run drifts into a suboptimal or failed state, it lacks a process-level correction mechanism. The fundamental pathology lies in its one-way pipeline: intermediate failures are typically terminated or bypassed, while fixed paradigms often strengthen model generation but leave ensemble decisions static, weakening both execution reliability and the controlled use of structural diversity. This indicates that LLM-driven AutoML needs a closed-loop ability for trial-correction-improvement together with evidence-based use of model diversity. To this end, we propose SAGE-Loop, a reliable closed-loop, self-adaptive, LLM-driven AutoML framework that performs multi-round generation and validation for trial-and-repair, and adaptively selects ensemble strategies in both supervised and unsupervised tasks, thereby unifying how to generate with how to use models. Across 20 public datasets, SAGE-Loop consistently improves performance and stability on classification, regression, and clustering tasks. Additional results further show its ability to recover from execution failures and maintain robust pipeline behavior.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 24 of its 50 references that are in the map. First posted on arXiv 11 Sep 2026.

Builds on

24 of 50 references.

References · 24

  1. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  2. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
  3. 2023Sparks of Artificial General Intelligence: Early experiments with GPT-4cited by 2,592
  4. 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
  5. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  6. 2019AutoML: A Survey of the State-of-the-Artcited by 390
  7. 2020AutoGluon-Tabular: Robust and Accurate AutoML for Structured Datacited by 383
  8. 2024OpenCodeInterpreter: Integrating Code Generation with Execution and Refinementcited by 247
  9. 2020Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learningcited by 174
  10. 2024Large Language Models(LLMs) on Tabular Data: Prediction, Generation, and Understanding -- A Surveycited by 159
  11. 2023Large Language Models for Automated Data Science: Introducing CAAFE for Context-Aware Automated Feature Engineeringcited by 127
  12. 2024DS-Agent: Automated Data Science by Empowering Large Language Models with Case-Based Reasoningcited by 125
12 more
  1. 2019FLAML: A Fast and Lightweight AutoML Librarycited by 106
  2. 2023AutoML-GPT: Automatic Machine Learning with GPTcited by 81
  3. 2022AMLB: an AutoML Benchmarkcited by 55
  4. 2022Automatic Code Documentation Generation Using GPT-3cited by 37
  5. 2023AutoML in the Age of Large Language Models: Current Challenges, Future Opportunities and Riskscited by 35
  6. 2020Fast, Accurate, and Simple Models for Tabular Data via Augmented Distillationcited by 27
  7. 2021AlphaD3M: Machine Learning Pipeline Synthesiscited by 24
  8. 2024Position: A Call to Action for a Human-Centered AutoML Paradigmcited by 14
  9. 2024Large Language Models Synergize with Automated Machine Learningcited by 9
  10. 2023Q(D)O-ES: Population-based Quality (Diversity) Optimisation for Post Hoc Ensemble Selection in AutoMLcited by 8
  11. 2020Ensemble Squared: A Meta AutoML Systemcited by 3
  12. 2023Integration Of Evolutionary Automated Machine Learning With Structural Sensitivity Analysis For Composite Pipelinescited by 1

Built on by

Not cited yet.