将可微分核岭回归融入深度学习,实现无需训练的高效模型增强。
Differentiable Kernel Ridge Regression for Deep Learning Pipelines

- 提出稀疏核(SK):可微、局部化、延迟训练的核岭回归变体。
- 在多个任务中,性能媲美训练过的神经读出,但训练量显著减少。
- 支持零训练迁移、非线性探查和混合核-神经模型,灵活适配各类场景。
深度神经网络主导现代机器学习,而替代函数逼近器在大规模应用中仍被低估。本文重新审视核方法作为标准深度学习流水线的即插即用组件。提出可微分、局部化、懒惰型的稀疏核(SK),其训练延迟至推理阶段,退化为求解小型局部系统。我们将SK集成到PyTorch中作为模块化层,保持端到端可训练性,并揭示其包含三类参数——特征表示、目标值与评估点——每类均可固定或学习。这一分解拓展了实践者的设计空间,支持零训练迁移、非线性探查及混合核-神经模型。在卷积网络、视觉变压器与强化学习中,基于SK的模块发挥双重作用:部分场景下,性能媲美训练神经读出,但训练成本更低;另一些场景中,作为附加组件提升现有模型性能。结果表明,一旦核方法实现可扩展与可微分,即可无缝集成于深度学习,而非作为独立范式。
原文摘要 · Abstract (English)
Deep neural networks dominate modern machine learning, while alternative function approximators remain comparatively underexplored at scale. In this work, we revisit kernel methods as drop-in components for standard deep learning pipelines. We introduce \emph{Sparse Kernels} (SKs), a differentiable, localized, and lazy variant of kernel ridge regression (KRR) that defers training to inference time and reduces to the solution of small local systems. We integrate SKs into PyTorch as modular layers that preserve end-to-end trainability, and we show that they expose three distinct sets of parameters -- feature representations, target values, and evaluation points -- each of which can be fixed or learned. This decomposition broadens the design space available to practitioners, enabling, in particular, training-free transfer, nonlinear probing, and hybrid kernel-neural models. Across convolutional networks, vision transformers, and reinforcement learning, SK-based modules serve two complementary roles: in some settings, they match the performance of trained neural readouts with substantially less training; in others, they augment existing models and improve their performance when used as additional components. Our results suggest that kernel methods, once made scalable and differentiable, can be readily integrated with deep learning rather than treated as a separate paradigm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。