用少量合成数据逼近真实数据训练效果,理论保证高效可靠。
Algorithmic Guarantees for Distilling Supervised and Offline RL Datasets
- 通过匹配随机回归器的损失,无须训练直接生成合成数据。
- 只需约 d² 量级的回归器即可保证线性模型性能接近原数据。
- 首次在离线强化学习中利用奖励与下一状态信息,无需策略优化。
给定训练数据集,数据蒸馏的目标是生成一个合成数据集,使得在后者上训练的模型表现与在原始数据上训练的模型相当。本文提出一种高效的监督学习数据蒸馏算法,针对 ℝᵈ 空间中的回归任务,通过匹配固定随机采样回归器在原始数据与合成数据上的损失,无需模型训练即可完成。首个关键贡献是证明:仅需 𝔛̃(d²) 个采样回归器,即可使任意有界线性模型在合成数据上的均方误差(MSE)几乎等于其在原始数据上的误差;因此在合成数据上优化的模型能接近原始数据上的最优性能。同时证明了 Ω(d²) 的下界,表明分析紧致性。第二个贡献是将该方法扩展至离线强化学习数据蒸馏,通过匹配贝尔曼损失而非以往的行为克隆目标,首次利用离线数据中的奖励与下一状态信息,无需策略模型优化。合成数据上的贝尔曼损失与原始数据相近,因此在合成数据上优化的动作价值函数所导出的策略,性能也接近原始数据优化的结果。实验验证了理论保证并观察到性能提升。
原文摘要 · Abstract (English)
Given a training dataset, the goal of dataset distillation is to derive a synthetic dataset such that models trained on the latter perform as well as those trained on the training dataset. In this work, we develop and analyze an efficient dataset distillation algorithm for supervised learning, specifically regression in $\mathbb{R}^d$, based on matching the losses on the training and synthetic datasets with respect to a fixed set of randomly sampled regressors without any model training. Our first key contribution is a novel performance guarantee proving that our algorithm needs only $\tilde{O}(d^2)$ sampled regressors to derive a synthetic dataset on which the MSE loss of any bounded linear model is nearly the same as its MSE loss on the given training data. In particular, the model optimized on the synthetic data has close to minimum loss on the training data, thus performing nearly as well as the model optimized on the latter. Complementing this, we also prove a matching lower bound of $Ω(d^2)$ for the number of sampled regressors showing the tightness of our analysis. Our second contribution is to extend our algorithm to offline RL dataset distillation by matching the Bellman loss, unlike previous works which used a behavioral cloning objective. This is the first such method which leverages both, the rewards and the next state information, available in offline RL datasets, without any policy model optimization. Our algorithm generates a synthetic dataset whose Bellman loss with respect to any linear action-value predictor is close to the latter's Bellman loss on the offline RL training dataset. Therefore, a policy associated with an action-value predictor optimized on the synthetic dataset performs nearly as well as that derived from the one optimized on the training data. We conduct experiments to validate our theoretical guarantees and observe performance gains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。