提出多尺度自适应理论,统一解释神经网络特征学习的两种经典视角。
From Kernels to Features: A Multi-Scale Adaptive Theory of Feature Learning
- 基于统计力学推导跨尺度网络输出的解析表达式。
- 发现线性网络中核自适应可简化为有效核缩放,但非线性网络需更复杂建模。
- 揭示特征学习中方向性变化的深层机制,适合研究理论深度的学者。
神经网络的特征学习对其表达能力与归纳偏置至关重要,催生了多种理论方法。部分方法通过训练后核尺度从初始化的变化来描述网络行为,其泛化能力接近高斯过程;另一些方法则强调训练使核适应数据,涉及核的方向性调整。这两种观点的关系与优势长期未解。本文提出多尺度自适应特征学习的理论框架,结合统计力学方法,推导出适用于不同尺度范式及两者之间的网络输出统计解析表达式。系统展开网络概率分布发现,均值场尺度仅需鞍点近似,而标准尺度需额外修正项。令人惊讶的是,在线性网络中预测均值输出时,核自适应可等价于有效核缩放;但对于线性和非线性网络,多尺度自适应方法能捕捉方向性特征学习效应,提供远超单纯核缩放的深刻洞见。
原文摘要 · Abstract (English)
Feature learning in neural networks is crucial for their expressive power and inductive biases, motivating various theoretical approaches. Some approaches describe network behavior after training through a change in kernel scale from initialization, resulting in a generalization power comparable to a Gaussian process. Conversely, in other approaches training results in the adaptation of the kernel to the data, involving directional changes to the kernel. The relationship and respective strengths of these two views have so far remained unresolved. This work presents a theoretical framework of multi-scale adaptive feature learning bridging these two views. Using methods from statistical mechanics, we derive analytical expressions for network output statistics which are valid across scaling regimes and in the continuum between them. A systematic expansion of the network's probability distribution reveals that mean-field scaling requires only a saddle-point approximation, while standard scaling necessitates additional correction terms. Remarkably, we find across regimes that kernel adaptation can be reduced to an effective kernel rescaling when predicting the mean network output in the special case of a linear network. However, for linear and non-linear networks, the multi-scale adaptive approach captures directional feature learning effects, providing richer insights than what could be recovered from a rescaling of the kernel alone.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。