用结构信息量指导数据选择与生成,提升模型跨域泛化能力。
Epiplexity Guided Data Selection and Generation for Out-of-Distribution Generalization

- 基于结构信息量(epiplexity)动态调整数据采样权重
- 合成数据生成时以提升epiplexity为奖励目标,优化生成分布
- 在零样本和微调任务中验证了高结构数据的强迁移性
现代系统需在训练未涵盖的任务间实现迁移。什么样的数据有助于这种新场景下的泛化?一种假设是:具有更多结构信息的数据可能包含可复用的通用模块。最近提出的epiplexity度量了计算受限学习者从数据中提取的结构信息量,为此提供了分析依据。本文将epiplexity作为在线训练信号,用于数据选择与合成数据生成。在数据选择上,拟合自然数据域的训练损失曲线,预测随训练词元数变化的预期epiplexity增益,并据此自适应调整各数据域的采样权重。在合成数据生成中,定义生成器奖励为学习者在历史生成数据缓冲区上的epiplexity变化量,使用REINFORCE策略梯度引导生成器趋向最大化epiplexity的分布。两种方法均表明,更高epiplexity能显著提升零样本和微调任务的下游性能,支持‘结构信息丰富的数据产生可跨领域迁移的表征’这一核心假设。
原文摘要 · Abstract (English)
Modern systems are increasingly expected to transfer across tasks not specified during training. What data facilitates generalization in these new, unanticipated settings? One hypothesis is that data with more structural information could contain shared circuits and subprograms that could be recycled in a wider array of downstream settings. Epiplexity, a recently proposed measure of the structural information a compute-bounded learner can extract from data, provides a mechanism to reason about this relationship. In this paper, we show how to operationalize epiplexity as an online training signal for data selection and synthetic data generation. For selection, we fit scaling laws to the training loss curves of natural data domains to predict the expected epiplexity gain as a function of training tokens, and use this signal to adaptively determine the sampling weights over domains during training. For synthetic data generation, we define a generator's reward as the change in learner epiplexity over a buffer of previously generated data and use REINFORCE policy gradients to guide the generator toward an epiplexity-maximizing distribution. In both cases, higher epiplexity predicts improved downstream performance on zero-shot and fine-tuning based tasks, supporting the hypothesis that data rich in structural information yield representations that transfer across domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。