基于层次特征设计网络初始化,提升训练效果
Neural network initialization with nonlinear characteristics and information on hierarchical features
- 根据层级特征调整SWIM算法的尺度因子
- 一维回归与MNIST任务均优于传统初始化方法
- 适合关注训练效率与初始化策略的研究者
神经网络参数(如权重和偏置)的初始化对学习性能有重要影响;若设置得当,甚至可避免使用反向传播进行额外训练。例如,基于小波变换或SWIM(采样关键区域)的概念已有相关初始化算法提出。另一方面,已有研究显示训练后的神经网络存在层次特征:早期隐藏层倾向于学习粗粒度信息。本文探究在初始化中利用此类层次特征的效果,提出一种框架,通过调整SWIM算法中的尺度因子,使早期隐藏层捕捉低频成分,晚期隐藏层表示高频成分。在一维回归任务和MNIST分类任务上的数值实验表明,该方法优于传统初始化算法。本工作揭示了内在层次特征在神经网络学习中的重要性,并提出了一种有效提升训练性能的参数初始化策略。
原文摘要 · Abstract (English)
Initialization of neural network parameters, such as weights and biases, has a crucial impact on learning performance; if chosen well, we can even avoid the need for additional training with backpropagation. For example, algorithms based on the ridgelet transform or the SWIM (sampling where it matters) concept have been proposed for initialization. On the other hand, some works show hierarchical features in trained neural networks; neural networks tend to learn coarse information in the early-stage hidden layers. In this work, we investigate the effects of utilizing information on the hierarchical features in the initialization of neural networks. Hence, we propose a framework that adjusts the scale factors in the SWIM algorithm to capture low-frequency components in the early-stage hidden layers and to represent high-frequency components in the late-stage hidden layers. Numerical experiments on a one-dimensional regression task and the MNIST classification task demonstrate that the proposed method outperforms the conventional initialization algorithms. This work clarifies the importance of intrinsic hierarchical features in learning neural networks, and the finding yields an effective parameter initialization strategy that enhances their training performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。