arXiv:2411.19468cs.LG2024-11被引 1

让随机特征模型的激活函数可学习,提升表达能力与效率

Learning Expressive Random Feature Models via Parametrized Activations

  • 用基函数加权构造可学习的激活函数,扩展函数表示空间
  • 使用RBF或样条基时,性能显著优于传统随机特征模型
  • 在计算效率上,RBF基比样条基快三倍,适合实际应用

随机特征(RF)方法是一种强大的核近似技术,但通常采用固定激活函数,限制了其在不同任务中的适应性。为此,我们提出可学习激活函数的随机特征模型(RFLAF),将激活函数参数化为基函数的加权和,如径向基函数(RBF)、样条函数、多项式等。以RBF为例,我们从单个RBF出发,拓展至多个RBF的情形,证明引入可学习激活组件能显著扩大函数表示空间。理论分析给出了达到低过拟合风险所需的样本数与随机特征数。实验对比了三种基函数:RBF、样条和多项式;结果表明,使用RBF和样条的RFLAF均优于其他随机特征模型,且RBF在计算效率上比样条高3倍。进一步解冻第一层参数并重训练,验证了可学习激活组件在典型两层神经网络中的表达优势。本工作深化了对现代神经网络中可学习激活机制的理解。

原文摘要 · Abstract (English)

The random feature (RF) method is a powerful kernel approximation technique, but it typically uses fixed activation functions, limiting its adaptability across diverse tasks. To overcome this limitation, we introduce the Random Feature Model with Learnable Activation Functions (RFLAF), a novel statistical model that parameterizes activation functions as weighted sums of basis functions within the random feature framework. Examples of basis functions include radial basis functions (RBFs), spline functions, polynomials, and so forth. For theoretical results, we consider RBFs as representative basis functions. We start with a single RBF as the activation, and then extend the results to multiple RBFs, demonstrating that RF models with a learnable activation component substantially expand the represented function space. We provide estimates on the required number of samples and random features to achieve low excess risk. In our experiments, we test RFLAF with three types of bases: radial basis functions, spline functions and polynomials. Experimental results show that RFLAFs with RBFs and splines consistently outperform other RF models, where RBFs are three times more computationally efficient than splines. We then unfreeze the first-layer parameters and retrain the models, validating the expressivity advantage of learnable activation components on regular two-layer neural networks. Our work provides a deeper understanding of learnable activation components within modern neural network architectures.

随机特征可学习激活表达能力RBF

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。