不改模型和训练方式,用线性化方法显著提升神经网络预测精度。
A universal linearized subspace refinement framework for neural networks
- 在固定网络参数下构建雅可比线性残差模型,求解子空间最优解。
- 相比标准梯度训练,误差降低一个数量级,尤其在复杂损失结构中表现突出。
- 适用于函数拟合、物理信息学习等场景,适合追求精度的科研与工程应用。
神经网络主要依赖梯度方法训练,但其最终预测精度常远低于模型表达能力所能达到的水平。本文提出线性化子空间精炼(LSR)框架,利用固定训练状态下的雅可比诱导线性残差模型,在该子空间内求解简化最小二乘问题,得到子空间最优解,从而获得精度显著提升的线性预测器,无需修改网络架构、损失函数或训练流程。在监督函数逼近、数据驱动算子学习及物理信息算子微调任务中,我们发现梯度训练往往无法达到这一可实现精度,即使局部线性化后问题为凸。这表明,损失导致的数值病态性而非非凸性或模型表达能力,是主要瓶颈。相比之下,单次LSR系统揭示了梯度训练未充分挖掘的精度潜力,常实现数量级误差下降。对于具有复合损失结构的算子约束问题,我们进一步提出迭代LSR,交替执行单次LSR与监督非线性对齐,将病态残差最小化转化为数值稳定的拟合步骤,实现加速收敛与更高精度。通过在固定线性化点处连接非线性神经表示与降阶线性求解器,LSR为监督学习、算子学习与科学计算提供了一个数值稳健且广泛适用的精炼框架。
原文摘要 · Abstract (English)
Neural networks are predominantly trained using gradient-based methods, yet in many applications their final predictions remain far from the accuracy attainable within the model's expressive capacity. We introduce Linearized Subspace Refinement (LSR), a general and architecture-agnostic framework that exploits the Jacobian-induced linear residual model at a fixed trained network state. By solving a reduced direct least-squares problem within this subspace, LSR computes a subspace-optimal solution of the linearized residual model, yielding a refined linear predictor with substantially improved accuracy over standard gradient-trained solutions, without modifying network architectures, loss formulations, or training procedures. Across supervised function approximation, data-driven operator learning, and physics-informed operator fine-tuning, we show that gradient-based training often fails to access this attainable accuracy, even when local linearization yields a convex problem. This observation indicates that loss-induced numerical ill-conditioning, rather than nonconvexity or model expressivity, can constitute a dominant practical bottleneck. In contrast, one-shot LSR systematically exposes accuracy levels not fully exploited by gradient-based training, frequently achieving order-of-magnitude error reductions. For operator-constrained problems with composite loss structures, we further introduce Iterative LSR, which alternates one-shot LSR with supervised nonlinear alignment, transforming ill-conditioned residual minimization into numerically benign fitting steps and yielding accelerated convergence and improved accuracy. By bridging nonlinear neural representations with reduced-order linear solvers at fixed linearization points, LSR provides a numerically grounded and broadly applicable refinement framework for supervised learning, operator learning, and scientific computing.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。