arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2511.09488cs.LG
收藏

AutoSynth:通过蒙特卡洛树搜索实现高质量合成数据集生成的自动化工作流优化

AutoSynth: Automated Workflow Optimization for High-Quality Synthetic Dataset Generation via Monte Carlo Tree Search

Shanghai Innavation Institute(上海创新研究院);Shanghai Institute of AI for Education(上海人工智能教育研究院);Department of Education Information Technology(教育信息技术部);School of Computer Science and Technology(计算机科学与技术学院)

Shuzhen Bi, Chang Song, Siyu Song, Jinze Lv, Jian Chen, Xinyun Wang, Aimin Zhou, Hao Hao

AI总结:

针对LLM特定任务SFT需高质量数据集但人工整理昂贵、现有合成数据工作流自动化方法存在冷启动问题的挑战,提出AutoSynth框架,通过蒙特卡洛树搜索与无数据集混合奖励实现无参考数据集的工作流优化,在主观教育任务中减少90%以上人工并生成数据训练模型显著优于基线。

摘要:

针对特定任务的大型语言模型(LLMs)监督微调(SFT)需要高质量数据集,但人工整理成本过高。合成数据生成具有可扩展性,但其有效性依赖于复杂的多阶段工作流,需整合提示工程和模型编排。现有自动化工作流方法存在冷启动问题:它们需要带标签的数据集进行奖励建模,这在无客观真值的主观、开放式任务中尤其棘手。我们提出AutoSynth框架,通过将问题重构为蒙特卡洛树搜索(由新型无数据集混合奖励引导),在无需参考数据集的情况下实现工作流发现与优化。该奖励通过两个LLM-as-judge组件实现元学习:一个利用动态生成的任务特定指标评估样本质量,另一个评估工作流代码和提示质量。在主观教育任务上的实验显示,尽管专家设计的工作流获得更高的人类偏好率(96-99%胜率 vs. AutoSynth的40-51%),但基于AutoSynth生成数据训练的模型显著优于基线(40-51% vs. 2-5%),并在某些指标上匹配或超过专家工作流,表明发现了超出人类直觉的质量维度。这些结果是在将人类工作量从5-7小时减少到仅30分钟(减少>90%)的情况下实现的。AutoSynth解决了数据中心AI中的冷启动问题,为主观LLM任务提供了可扩展、经济高效的方法。代码:https://github.com/bisz9918-maker/AutoSynth。

英文摘要:

Supervised fine-tuning (SFT) of large language models (LLMs) for specialized tasks requires high-quality datasets, but manual curation is prohibitively expensive. Synthetic data generation offers scalability, but its effectiveness relies on complex, multi-stage workflows, integrating prompt engineering and model orchestration. Existing automated workflow methods face a cold start problem: they require labeled datasets for reward modeling, which is especially problematic for subjective, open-ended tasks with no objective ground truth. We introduce AutoSynth, a framework that automates workflow discovery and optimization without reference datasets by reframing the problem as a Monte Carlo Tree Search guided by a novel dataset-free hybrid reward. This reward enables meta-learning through two LLM-as-judge components: one evaluates sample quality using dynamically generated task-specific metrics, and another assesses workflow code and prompt quality. Experiments on subjective educational tasks show that while expert-designed workflows achieve higher human preference rates (96-99% win rates vs. AutoSynth's 40-51%), models trained on AutoSynth-generated data dramatically outperform baselines (40-51% vs. 2-5%) and match or surpass expert workflows on certain metrics, suggesting discovery of quality dimensions beyond human intuition. These results are achieved while reducing human effort from 5-7 hours to just 30 minutes (>90% reduction). AutoSynth tackles the cold start issue in data-centric AI, offering a scalable, cost-effective method for subjective LLM tasks. Code: https://github.com/bisz9918-maker/AutoSynth.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑