Large Language Models for Constructing and Optimizing Machine Learning Workflows: A Survey

📅 2024-11-11
🏛️ arXiv.org
📈 Citations: 0
✨ Influential: 0
📄 PDF
🤖 AI Summary
This work addresses the intelligent evolution of AutoML by investigating how large language models (LLMs) can optimize the end-to-end machine learning (ML) pipeline. Method: We propose a four-dimensional capability framework—language understanding, reasoning, interaction, and generation—to systematically characterize LLM-driven ML workflow paradigms; integrate prompt engineering, instruction tuning, chain-of-thought reasoning, tool-augmented LLMs, and multi-stage orchestration; and synthesize over 50 state-of-the-art techniques. Contribution/Results: Empirical evaluation demonstrates that LLMs substantially lower modeling barriers, enhance cross-task generalization, and improve human-AI collaboration efficiency—achieving semantic modeling and human-in-the-loop breakthroughs in data preprocessing, feature engineering, model selection, hyperparameter optimization, and workflow evaluation. However, critical challenges remain regarding reliability, interpretability, and computational overhead.

Technology Category

Machine Learning: Auto ML and Hyperparameter TuningNatural Language Processing: (Large) Language ModelsHumans and AI: Human-in-the-loop Machine Learning

Application Category

User Modeling, Personalization and Recommendation: Large Language Models (LLM) for user modeling and recommendationSemantics and Knowledge: Data modeling to support human-machine intelligence, including LLMs agents, intelligent system behavior, explanations, and user-friendly interactionsSearch and Retrieval-Augmented AI: Search Tool Learning with LLM: Teaching LLMs to invoke search and make use of retrieved information
📝 Abstract
Building effective machine learning (ML) workflows to address complex tasks is a primary focus of the Automatic ML (AutoML) community and a critical step toward achieving artificial general intelligence (AGI). Recently, the integration of Large Language Models (LLMs) into ML workflows has shown great potential for automating and enhancing various stages of the ML pipeline. This survey provides a comprehensive and up-to-date review of recent advancements in using LLMs to construct and optimize ML workflows, focusing on key components encompassing data and feature engineering, model selection and hyperparameter optimization, and workflow evaluation. We discuss both the advantages and limitations of LLM-driven approaches, emphasizing their capacity to streamline and enhance ML workflow modeling process through language understanding, reasoning, interaction, and generation. Finally, we highlight open challenges and propose future research directions to advance the effective application of LLMs in ML workflows.
Problem

Research questions and friction points this paper is trying to address.

Large Language Models
Automated Machine Learning
Data Processing and Model Selection
Innovation

Methods, ideas, or system contributions that make the work stand out.

Large Language Models
Machine Learning Integration
AutoML Efficiency Enhancement
Shanghai Jiao Tong University | Stanford University
Y
Yang Gu
Shanghai Jiao Tong University, Shanghai, China
H
Hengyu You
Shanghai Jiao Tong University, Shanghai, China
J
Jian Cao
Shanghai Jiao Tong University, Shanghai, China
M
Muran Yu
Stanford University, Stanford, California, USA