arXiv:2607.15901: DSWorld-Weltmodell beschleunigt Data-Science-Agenten um bis zu 14× beim Training und bei der Inferenz

Was ist DSWorld, und warum brauchen Data-Science-Agenten ein Weltmodell?

Ein Weltmodell ist eine Komponente, die intern die Ergebnisse von Aktionen simuliert, bevor sie tatsächlich ausgeführt werden, statt dass der Agent jedes Mal auf das reale Ergebnis einer Operation wartet. Zherui Yang, Fan Liu und Hao Liu führen DSWorld ein, ein Weltmodell, das speziell für Data-Science-Agenten entwickelt wurde — Systeme, die automatisch Code für die Datenanalyse, das Training von Modellen und die Auswertung von Ergebnissen schreiben und ausführen. Das gelöste Problem ist rechnerische Ineffizienz: Jede ausprobierte Operation würde sonst eine tatsächliche Ausführung erfordern, oft bei ressourcenintensiven Aufgaben wie dem Training von Modellen oder der Verarbeitung großer Datensätze.

DSWorld kombiniert strukturierten Zustand, Routing und Simulation

Die Architektur kombiniert drei Elemente: einen strukturierten Zustand, der den aktuellen Aufgabenkontext knapp beschreibt, intelligentes Routing, das entscheidet, wann sich eine echte Simulation gegenüber einer schnellen Schätzung lohnt, und einen LLM-basierten Simulator, der das Ergebnis ressourcenintensiver Operationen vorhersagt, ohne sie tatsächlich auszuführen. Das System ist zusätzlich mit der Technik „Reflective World Model Optimization” gekoppelt — einem fehlerbewussten Reinforcement-Learning-Ansatz (RL), der aus den eigenen Vorhersagefehlern lernt — sowie einem Datensatz von 8.000 aufgezeichneten Trajektorien, der als Trainingsgrundlage dient.

DSWorld: 14× schnelleres Training und 3-6× schnellere Inferenz

Die Ergebnisse zeigen erhebliche Beschleunigungen: DSWorld erzielt ein etwa 14× schnelleres Training von RL-Agenten im Vergleich zu Ansätzen ohne Weltmodell sowie eine 3- bis 6-fach schnellere suchbasierte Inferenz, bei der der Agent mehrere mögliche Aktionen erkundet, bevor er die beste auswählt. Zudem übertrifft das System die stärkste getestete LLM-Baseline um 35,6 % bei Aufgaben zur Vorhersage von Zustandsübergängen — also der Vorhersage, wie sich der Zustand einer Aufgabe nach einer bestimmten Operation verändert —, während es bei den allgemeinen Data-Science-Aufgaben konkurrenzfähig bleibt.

Beschleunigungen verändern die Ökonomie der Agentenarbeit

Die Kombination aus 14× schnellerem Training und 3-6× schnellerer Inferenz bedeutet, dass Data-Science-Agenten innerhalb desselben Zeit- und Rechenbudgets deutlich mehr Strategien ausprobieren können. Bei Aufgaben, die sonst teure, wiederholte Codeausführung erfordern — etwa Hyperparametersuche oder Modellauswahl —, übersetzt sich das direkt in niedrigere Kosten und schnellere Iterationen bei der Entwicklung automatisierter Data-Science-Pipelines.