arXiv:2503.17195cs.LGcs.AI2025-03NeurIPS被引 6

用树结构划分数据空间,自动生成多样且均衡的合成数据。

TreeSynth: Synthesizing Diverse Data from Scratch via Tree-Guided Subspace Partitioning

  • 基于决策树思想构建空间分割树,将数据空间递归划分为互斥原子子空间。
  • 在多个基准上合成数据,平均性能提升10%,显著优于人工数据和现有方法。
  • 可重平衡已有数据集,适合需要高质量多样性数据的模型训练场景。

模型定制需要高质量且多样化的数据集,但获取此类数据仍耗时耗力。尽管大语言模型在数据合成方面潜力巨大,但现有方法受限于初始数据少、模型偏见及低变化性提示,导致数据规模扩大时多样性不足且分布偏倚。为此,我们提出TREE SYNTH,一种受决策树启发的基于空间划分的数据合成方法。该方法构建空间划分树,将特定任务的完整数据空间(根节点)递归划分为多个互斥且完备的原子子空间(叶节点),确保属性的差异性和覆盖性,再在每个原子子空间内合成样本。这种全局划分与合成策略有效避免重复和空间坍缩,保障大规模数据合成的多样性。此外,空间划分树支持样本分配至原子子空间,可实现对现有数据集的再平衡,以获得更均衡、全面的分布。实验证明,TREE SYNTH在多个基准上持续展现更优的数据多样性、模型性能与可扩展性,平均性能提升达10%。同时,经其调整的数据集在性能上也显著提升,证明其在重分布已有数据方面的有效性。代码已开源:https://github.com/cpa2001/TreeSynth。

原文摘要 · Abstract (English)

Model customization necessitates high-quality and diverse datasets, but acquiring such data remains time-consuming and labor-intensive. Despite the great potential of large language models (LLMs) for data synthesis, current approaches are constrained by limited seed data, model biases, and low-variation prompts, resulting in limited diversity and biased distributions with the increase of data scales. To tackle this challenge, we introduce TREESYNTH, a tree-guided subspace-based data synthesis approach inspired by decision trees. It constructs a spatial partitioning tree to recursively divide a task-specific full data space (i.e., root node) into numerous atomic subspaces (i.e., leaf nodes) with mutually exclusive and exhaustive attributes to ensure both distinctiveness and comprehensiveness before synthesizing samples within each atomic subspace. This globally dividing-and-synthesizing method finally collects subspace samples into a comprehensive dataset, effectively circumventing repetition and space collapse to ensure the diversity of large-scale data synthesis. Furthermore, the spatial partitioning tree enables sample allocation into atomic subspaces, allowing the rebalancing of existing datasets for more balanced and comprehensive distributions. Empirically, extensive experiments across diverse benchmarks consistently demonstrate the superior data diversity, model performance, and robust scalability of TREESYNTH compared to both human-crafted datasets and peer data synthesis methods, with an average performance gain reaching 10%. Besides, the consistent improvements of TREESYNTH-balanced datasets highlight its efficacious application to redistribute existing datasets for more comprehensive coverage and the induced performance enhancement. The code is available at https://github.com/cpa2001/TreeSynth.

数据合成多样性树结构模型训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。