arXiv:2607.15901: DSWorld 세계 모델, 데이터 사이언스 에이전트의 훈련과 추론을 최대 14배 가속

DSWorld란 무엇이며 데이터 사이언스 에이전트에 왜 세계 모델이 필요한가?

세계 모델은 행동이 실제로 실행되기 전에 그 결과를 내부적으로 시뮬레이션하는 구성 요소로, 에이전트가 매번 실제 결과를 기다릴 필요를 없애줍니다. Zherui Yang, Fan Liu, Hao Liu는 데이터 사이언스 에이전트—데이터 분석, 모델 훈련, 결과 평가를 위한 코드를 자동으로 작성하고 실행하는 시스템—를 위해 특별히 설계된 세계 모델인 DSWorld를 도입했습니다. 이들이 해결하려는 문제는 연산 비효율성입니다. 시도되는 모든 작업은 보통 실제 실행을 필요로 하며, 이는 모델 훈련이나 대규모 데이터셋 처리 같은 자원 집약적 작업에서 특히 두드러집니다.

DSWorld는 구조화된 상태, 라우팅, 시뮬레이션을 결합한다

이 아키텍처는 세 가지 요소를 결합합니다. 작업의 현재 맥락을 간결하게 기술하는 구조화된 상태, 실제 시뮬레이션을 호출할 가치가 있을 때와 빠른 추정이 더 유리할 때를 판단하는 지능형 라우팅, 그리고 실제로 실행하지 않고도 자원 집약적 작업의 결과를 예측하는 LLM 기반 시뮬레이터입니다. 이 시스템은 또한 “Reflective World Model Optimization”이라 불리는 기법—세계 모델 자체의 예측 오류로부터 학습하는, 오류를 인식하는 강화학습(RL) 접근 방식—및 훈련의 기반이 되는 8,000개의 기록된 궤적 데이터셋과 결합되어 있습니다.

DSWorld: 훈련 14배, 추론 3~6배 가속

결과는 상당한 가속을 보여줍니다. DSWorld는 세계 모델이 없는 접근 방식에 비해 RL 에이전트 훈련을 약 14배 가속하고, 탐색 기반 추론—에이전트가 최선의 행동을 선택하기 전에 여러 가능한 행동을 탐색하는 과정—을 3~6배 가속합니다. 또한 이 시스템은 상태 전이 예측—특정 작업 이후 작업 상태가 어떻게 변화하는지 예측하는 것—과제에서 테스트된 가장 강력한 LLM 베이스라인보다 35.6% 더 나은 성능을 보이면서도, 전체 데이터 사이언스 작업에서 경쟁력 있는 성능을 유지합니다.

가속화가 에이전트 작업의 경제성을 바꾼다

14배의 훈련 가속과 3~6배의 추론 가속이 결합되면, 데이터 사이언스 에이전트는 동일한 시간과 연산 예산 안에서 훨씬 더 많은 전략을 시도할 수 있게 됩니다. 하이퍼파라미터 탐색이나 모델 선택처럼 보통 비용이 많이 드는 반복적 코드 실행이 필요한 작업의 경우, 이는 곧바로 비용 절감과 자동화된 데이터 사이언스 파이프라인 개발에서의 더 빠른 반복으로 이어집니다.