用多项式设计神经网络块,提升精度且减少参数量
Poly-MgNet: Polynomial Building Blocks in Multigrid-Inspired ResNets
- 基于多网格理论中的多项式平滑器设计新网络块
- 使用实虚根构造二次多项式,精度优于原MgNet和ResNet
- 参数更少,适合追求轻量化与高精度的模型应用
ResNets与多网格(Multigrid, MG)方法在结构上存在相似性,如卷积和池化等基本单元。2016年,He等人指出二者共用构建模块。多网格方法常用于求解偏微分方程产生的大型稀疏线性系统,其核心依赖于平滑与残差限制/降维两个概念。基于此,He与Xu提出了将多网格机制融入残差网络设计的MgNet框架。本文提出一种受多网格理论中多项式平滑器启发的新神经网络构建模块,从多网格视角自然扩展了原有框架,形成Poly-MgNet,并同时减少模型参数量。我们系统研究了初始系数选择、多项式阶数、激活函数位置及批量归一化的配置影响。结果表明,基于实数与虚数根构建的二次多项式块能显著提升模型精度。此外,该方法在精度与参数数量之间实现更优权衡,优于标准ResNet以及特定配置的MgNet。
原文摘要 · Abstract (English)
The structural analogies of ResNets and Multigrid (MG) methods such as common building blocks like convolutions and poolings where already pointed out by He et al.\ in 2016. Multigrid methods are used in the context of scientific computing for solving large sparse linear systems arising from partial differential equations. MG methods particularly rely on two main concepts: smoothing and residual restriction / coarsening. Exploiting these analogies, He and Xu developed the MgNet framework, which integrates MG schemes into the design of ResNets. In this work, we introduce a novel neural network building block inspired by polynomial smoothers from MG theory. Our polynomial block from an MG perspective naturally extends the MgNet framework to Poly-Mgnet and at the same time reduces the number of weights in MgNet. We present a comprehensive study of our polynomial block, analyzing the choice of initial coefficients, the polynomial degree, the placement of activation functions, as well as of batch normalizations. Our results demonstrate that constructing (quadratic) polynomial building blocks based on real and imaginary polynomial roots enhances Poly-MgNet's capacity in terms of accuracy. Furthermore, our approach achieves an improved trade-off of model accuracy and number of weights compared to ResNet as well as compared to specific configurations of MgNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。