通过参数重参化提升神经网络对高频多尺度问题的建模能力
RepNN: Tackling spectral bias in deep neural networks via parameter reparameterization

- 重参化首层权重与偏置,精准控制初始斜率与分段点分布
- 在1维和4维函数逼近、PDE求解等任务中显著提升高频特征捕捉精度
- 适合需处理振荡或多尺度数据的科学计算场景,如地震波建模
深度神经网络(DNN)在科学计算中表现卓越,但在捕捉振荡和多尺度行为时常受谱偏差限制。本文通过分析浅层ReLU网络拟合高频率函数的失败,揭示两个关键因素:初始斜率尺度和网络诱导的分段点分布。基于此,提出RepNN,一种针对高频率与多尺度问题设计的重参化网络模型,其激活函数可为ReLU或tanh。核心思想是重参化第一隐藏层的权重与偏置,以有效控制初始斜率尺度并提供合适的初始分段点分布;同时将重参化参数设为可训练,使DNN在训练中实现自适应频率缩放。此外,推导了重参化DNN输出与斜率幅度的定量估计,用于指导初始化。数值实验包括一维与四维多尺度函数逼近、结合物理信息神经网络(PINNs)的前向与反向偏微分方程问题,以及基于真实数据的地震问题算子学习,结果表明:与原始DNN相比,RepNN在仅略增计算成本下显著提升了对高度振荡特征的预测精度。这些结果表明,RepNN是一种有效且灵活的解决方案,可克服谱偏差并应用于多尺度问题。
原文摘要 · Abstract (English)
Deep neural networks (DNNs) have achieved remarkable success in scientific computing, yet they often suffer from spectral bias in capturing oscillatory and multiscale behaviors. In this study, we investigate this limitation by examining the failure of shallow ReLU neural networks in fitting high-frequency functions. This observation identifies two important factors in resolving rapid oscillations: the initial slope scale and the distribution of partition points induced by the networks. Motivated by this analysis, we propose RepNN, a reparameterized neural network model with activation ReLU or tanh designed for high-frequency and multiscale problems. The key idea is to reparameterize the weights and biases in the first hidden layer, which enables effective control of the initial slope scale and provides an appropriate distribution of the initial partition points. Furthermore, treating the reparameterized weights and biases as trainable parameters allows the DNN to achieve adaptive frequency scaling during training. In addition, we derive quantitative estimates for the output and slope magnitudes of the reparameterized DNN to guide the initialization of the proposed method. Numerical experiments, including multiscale one- and four-dimensional function approximations, forward and inverse PDE problems in combination with physics-informed neural networks (PINNs), and operator learning for an earthquake problem using real data, demonstrate that RepNN improves the predicted accuracy of vanilla DNNs in capturing highly oscillatory features with slightly additional computational cost. These results indicate that RepNN provides an effective and flexible approach for overcoming spectral bias and applying DNNs to multiscale problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。