DSAgentBench: Can Agents Automate End-to-End Data-Science Workflows in Real Computer Environments?
Questo articolo presenta DSAgentBench, il primo benchmark progettato per valutare la capacità degli agenti IA di automatizzare flussi di lavoro di data science end-to-end all'interno di veri ambienti informatici, rivelando un significativo divario di prestazioni in cui anche i modelli più forti faticano con l'orchestrazione degli strumenti e il ragionamento multi-step.
Articolo originale sotto licenza CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Questa è una spiegazione generata dall'IA dell'articolo qui sotto. Non è stata scritta né approvata dagli autori. Per precisione tecnica, consulta l'articolo originale. Leggi il disclaimer completo
Riepilogo Tecnico: DSAgentBench
Definizione del Problema
La scienza dei dati nel mondo reale coinvolge flussi di lavoro complessi e a lungo orizzonte che spaziano dalla manipolazione dei dati, l'esplorazione, la modellazione, la visualizzazione e la validazione. Questi compiti richiedono l'uso coordinato di strumenti diversificati — inclusi notebook, IDE, terminali, browser e database — all'interno di un sistema operativo funzionante. Sebbene i recenti progressi nei modelli linguistici di grandi dimensioni (LLM) abbiano dimostrato capacità nella generazione di codice o nell'esecuzione di compiti analitici isolati, i benchmark esistenti non riescono a valutare se gli agenti possano eseguire autonomamente flussi di lavoro completi di data science end-to-end in ambienti informatici realistici.
I benchmark attuali si dividono in due categorie: quelli che valutano la generazione di codice in isolamento (ad es. DS-1000, DSEval) senza richiedere l'interazione con il sistema, e quelli che valutano il controllo generale del computer (ad es. OSWorld, WebArena) senza valutare il ragionamento analitico specifico del dominio. Di conseguenza, vi è una mancanza di framework di valutazione che testino la capacità di un agente di navigare nei file system, gestire le dipendenze, interpretare gli errori e perfezionare le analisi basandosi sugli output intermedi all'interno di un vero ambiente OS.
Metodologia
Costruzione del Benchmark (DSAgentBench)
Gli autori introducono DSAgentBench, il primo benchmark progettato per valutare i flussi di lavoro autonomi di data science all'interno di veri sistemi operativi. Il benchmark consiste in 275 compiti diversificati, scritti da esseri umani, che coprono l'intero ciclo di vita della data science.
- Formulazione del Compito: I compiti sono definiti come tuple , dove è la configurazione iniziale del sistema (dataset, struttura dei file, librerie installate), è un'istruzione in linguaggio naturale e è un valutatore Python deterministico.
- Sourcing dei Dati: I dataset sono tratti da fonti eterogenee del mondo reale, tra cui Kaggle, OpenML, GitHub, database SQLite e API web. I dati includono modalità tabulari (95,3%), immagini (3,6%) e testo (1,1%).
- Categorie di Compiti: I compiti sono organizzati in sei categorie di capacità: Acquisizione Dati, Analisi Esplorativa dei Dati (EDA), Ingegneria delle Caratteristiche (Feature Engineering), Modellazione, Valutazione/Deployment e Visualizzazione/Reporting.
- Protocollo di Valutazione: A differenza dei benchmark che misurano l'esecuzione del codice, DSAgentBench impiega valutatori deterministici che verificano la correttezza analitica, la qualità dell'output visivo e le prestazioni del modello. Un compito è considerato completato con successo solo se gli output finali dell'agente soddisfano i criteri del valutatore (punteggio ).
- Pipeline di Costruzione: Il benchmark è stato costruito attraverso un processo in tre fasi: sourcing dei dataset, progettazione collaborativa di compiti/valutatori da parte di esperti umani (con l'uso di LLM solo per il raffinamento) e verifica tramite doppio annotatore per garantire la riproducibilità e la correttezza tecnica.
Architettura dell'Ambiente
DSAgentBench estende il framework OSWorld per creare un ambiente di esecuzione realistico:
- OS: Ubuntu con Python e librerie di data science preinstallati.
- Strumenti: Visual Studio Code, Jupyter Notebook, Chrome e accesso alle API di Kaggle/OpenML.
- Spazio di Osservazione: Gli agenti ricevono uno screenshot a 1920×1080 o una modalità ibrida Screenshot + Albero di Accessibilità (A11y), che fornisce metadati strutturati dell'interfaccia utente (ruoli, bounding box, stati di interazione).
- Spazio di Azione: Gli agenti interagiscono tramite azioni GUI (click del mouse, input da tastiera) e meta-azioni (WAIT, DONE, FAIL). L'ambiente cattura le transizioni di stato dopo ogni azione.
Modelli Valutati
Gli autori hanno valutato 15 agenti open-source e closed-source, tra cui:
- Closed-source: GPT-4o, GPT-5 (e mini), O4-mini, Claude Sonnet 4/4.5/4.6, Gemini 2.5 Pro e l'Agente Computer di OpenAI.
- Open-source: UI-TARS (2B/7B), GUI-OWL-7B, OpenCUA-72B e modelli ibridi (Jedi accoppiato con GPT-4o).
Risultati Chiave
Prestazioni Complessive
Gli esperimenti rivelano un sostanziale divario di capacità tra gli attuali sistemi agentici e le richieste dei veri flussi di lavoro di data science:
- Agente più forte: Claude-4.6-Sonnet ha ottenuto la prestazione più alta con il 56,70% di successo nei compiti sotto l'impostazione Screenshot + A11y Tree.
- Altri Modelli Closed-Source: Le prestazioni sono diminuite significativamente per altri modelli, con GPT-5 al 29,81% e altri (GPT-4o, Gemini-2.5-Pro) che oscillano intorno al 20%.
- Modelli Open-Source: Tutti gli agenti open-source hanno ottenuto meno dell'1% di successo, fallendo frequentemente nell'orchestrazione degli strumenti, nel grounding dell'OS e nel ragionamento multi-step.
- Baseline Umana: Gli esperti umani hanno raggiunto un tasso di successo dell'85,09%, evidenziando il divario rimanente anche per il più forte degli agenti IA.
Analisi di Ablazione ed Errori
- Complessità del Compito: Le prestazioni degradano monotonicamente con la difficoltà. I compiti "difficili" (5+ passaggi) rimangono i più impegnativi. I flussi di lavoro multi-stadio (56,7% dei compiti) sono significativamente più difficili rispetto ai compiti a singolo stadio a causa della necessità di mantenimento dello stato e recupero dagli errori.
- Utilizzo degli Strumenti: I compiti eseguiti in Jupyter Notebook hanno performato meglio rispetto a quelli in VS Code, principalmente a causa di meno fallimenti legati al terminale e all'ambiente.
- Modalità di Osservazione: L'aggiunta di informazioni sull'albero A11y ha generalmente migliorato le prestazioni, suggerendo che i metadati strutturati dell'UI aiutino il grounding, sebbene i guadagni varino a seconda del modello.
- Modalità di Fallimento:
- Gli agenti open-source sono falliti quasi esclusivamente (97–98%) a causa di errori di grounding (incapacità di allineare le istruzioni con lo stato del desktop).
- Gli agenti closed-source più forti hanno mostrato fallimenti misti, inclusi errori del terminale, difetti nella generazione del codice e deficit di ragionamento.
- Struttura Temporale: Gli agenti open-source e quelli più deboli spesso fallivano precocemente (incapacità di aprire i terminali), mentre i modelli più forti tendevano a fallire nelle fasi successive dopo una lunga ed inefficace esplorazione.
- Sensibilità al Budget: Aumentare il budget di interazione da 15 a 50 passaggi ha prodotto solo guadagni marginali (24,54% 25,81%), indicando che i fallimenti non sono dovuti principalmente ai limiti di step, ma piuttosto a problemi fondamentali di pianificazione e ragionamento.
Significato e Rivendicazioni
Il paper posiziona DSAgentBench come una risorsa fondamentale per lo sviluppo di agenti di data science radicati, verificabili e autonomi. I suoi principali contributi sono:
- Primo Benchmark su Real-OS: È il primo benchmark che valuta i flussi di lavoro autonomi di data science all'interno di un sistema operativo funzionante, coprendo l'intero ciclo di vita dalla acquisizione dei dati alla validazione.
- Valutazione Basata sull'Esecuzione: Sposta il paradigma di valutazione dalla "correttezza del codice" alla "correttezza analitica", richiedendo agli agenti di produrre artefatti validi (visualizzazioni, modelli, report) verificati da script deterministici.
- Rivelazione del Divario: I risultati espongono un limite significativo nei sistemi agentici attuali, dimostrando che anche i modelli più forti faticano con la coordinazione degli strumenti, il ragionamento a lungo orizzonte e il grounding dell'OS richiesti per la data science reale.
- Direzione Futura: Identificando modalità di fallimento specifiche (grounding, pianificazione, orchestrazione degli strumenti), il benchmark fornisce una tabella di marcia chiara per la ricerca futura volta a costruire agenti capaci di svolgere compiti di data science nel mondo reale.
Gli autori rilasciano il benchmark all'indirizzo https://github.com/vis-nlp/DSAgentBench per facilitare ulteriori ricerche in questo dominio.
Sommerso dagli articoli nel tuo campo?
Ricevi digest giornalieri degli articoli più recenti corrispondenti alle tue parole chiave di ricerca — con riassunti tecnici, nella tua lingua.