cs.DBOct 12, 2026

ML Engineering Agents - A Data Management Perspective

Authors: Arnab Phani, Elias Strauss, Tim Schuster, Sebastian Schelter

Organizations: BIFOLD & TU Berlin, Germany

Abstract

Machine learning engineering (MLE) agents are emerging as a new paradigm for automating end-to-end ML pipeline development. Using large language models for code generation and predictive performance as a machine-gradable feedback signal, these agents iteratively synthesize, execute, and refine pipeline code. Yet current systems are largely evaluated on pre-built, single-table benchmarks that abstract away key data management challenges. This tutorial introduces the foundations of MLE agents and highlights research opportunities for the database community. We cover how agents combine program synthesis with iterative and evolutionary optimization, and discuss open challenges in efficient pipeline execution, data discovery and augmentation over heterogeneous sources, reliability and debugging, as well as fairness and compliance. We connect these challenges to existing data management techniques and conclude with a hands-on session using contemporary MLE agents.

Explore similar work

CardsList
  1. DataClawEval: A Benchmark for Data Engineering Agents in Real Industrial Harness

    Jul 30, 2026Debin Meng, Jiaming Yang, Zefang Zong +4LLM Agent EvaluationAutomated Evaluation

  2. DataMaster: Data-Centric Autonomous AI Research

    May 11, 2026Yaxin Du, Xiyuan Yang, Zhifan Zhou +12Autonomous ExperimentationAutonomous Research Agents

  3. Be Fair! Can Machine Learning Engineering Agents Adhere to Fairness Constraints?

    Jun 3, 2026Anna Richter, Julia Stoyanovich, Sebastian SchelterAlgorithmic FairnessAI Agent Evaluation