Collaborative Evolving Strategy for Automatic Data-Centric Development

TL;DR

This paper introduces AD^2, an autonomous data-centric AI development framework leveraging LLMs, with a novel collaborative evolution strategy that improves data scheduling and implementation through practical feedback.

Contribution

It pioneers the automatic data-centric development task and proposes Co-STEER, an LLM-based agent that collaboratively evolves scheduling and implementation skills.

Findings

Co-STEER significantly improves data development efficiency.

The collaborative evolution strategy enhances scheduling and implementation accuracy.

Experimental results validate the effectiveness of the proposed approach.

Abstract

Artificial Intelligence (AI) significantly influences many fields, largely thanks to the vast amounts of high-quality data for machine learning models. The emphasis is now on a data-centric AI strategy, prioritizing data development over model design progress. Automating this process is crucial. In this paper, we serve as the first work to introduce the automatic data-centric development (AD^2) task and outline its core challenges, which require domain-experts-like task scheduling and implementation capability, largely unexplored by previous work. By leveraging the strong complex problem-solving capabilities of large language models (LLMs), we propose an LLM-based autonomous agent, equipped with a strategy named Collaborative Knowledge-STudying-Enhanced Evolution by Retrieval (Co-STEER), to simultaneously address all the challenges. Specifically, our proposed Co-STEER agent enriches its domain knowledge through our proposed evolving strategy and develops both its scheduling and implementation skills by accumulating and retrieving domain-specific practical experience. With an improved schedule, the capability for implementation accelerates. Simultaneously, as implementation feedback becomes more thorough, the scheduling accuracy increases. These two capabilities evolve together through practical feedback, enabling a collaborative evolution process. Extensive experimental results demonstrate that our Co-STEER agent breaks new ground in AD^2 research, possesses strong evolvable schedule and implementation ability, and demonstrates the significant effectiveness of its components. Our Co-STEER paves the way for AD^2 advancements.