用核压缩技术加速监督学习,训练推理快两倍
Supervised Kernel Thinning
- 将核薄化算法用于回归任务,构建新型核薄化估计器
- 训练与推理速度提升近两倍,且比随机采样更准确
- 适合大规模核方法应用,尤其数据量大时
Dwivedi & Mackey(2024)提出的核薄化(Kernel Thinning, KT)算法能以远优于独立同分布采样的方式压缩任意点集。通过生成高保真、规模显著小于原始数据的压缩核心集,KT可显著加速蒙特卡洛积分、不确定性量化和非参数假设检验等无监督任务,同时保持极低的统计误差。本文将该算法推广至监督学习场景,结合经典算法——Nadaraya-Watson(NW)回归或核平滑,以及核岭回归(KRR),实现训练与推理时间的二次加速。我们证明,在每种设定下,使用KT进行分布压缩等价于构建特定核函数,并提出核薄化NW与核薄化KRR估计器。理论分析表明,基于KT的回归估计器在计算效率上显著优于全数据方法,且在统计效率上优于独立同分布采样。此外,我们还给出了一个全新的乘法误差保证。仿真与真实数据实验验证了设计的有效性。
原文摘要 · Abstract (English)
The kernel thinning algorithm of Dwivedi & Mackey (2024) provides a better-than-i.i.d. compression of a generic set of points. By generating high-fidelity coresets of size significantly smaller than the input points, KT is known to speed up unsupervised tasks like Monte Carlo integration, uncertainty quantification, and non-parametric hypothesis testing, with minimal loss in statistical accuracy. In this work, we generalize the KT algorithm to speed up supervised learning problems involving kernel methods. Specifically, we combine two classical algorithms--Nadaraya-Watson (NW) regression or kernel smoothing, and kernel ridge regression (KRR)--with KT to provide a quadratic speed-up in both training and inference times. We show how distribution compression with KT in each setting reduces to constructing an appropriate kernel, and introduce the Kernel-Thinned NW and Kernel-Thinned KRR estimators. We prove that KT-based regression estimators enjoy significantly superior computational efficiency over the full-data estimators and improved statistical efficiency over i.i.d. subsampling of the training data. En route, we also provide a novel multiplicative error guarantee for compressing with KT. We validate our design choices with both simulations and real data experiments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。