用双网络PINN求解弹簧质量阻尼系统最优控制,结果与经典方法高度一致。
Dual-Network PINNs for Optimal Control: A Reproducible Benchmark on the Mass-Spring-Damper System
- 设计双神经网络:状态网络精确满足边界条件,控制网络无约束优化。
- 最优代价误差仅0.0001,状态与控制误差在经典方法范围内。
- 提供可复现的Colab代码,适合初学者入门物理信息神经网络控制。
本文针对弹簧质量阻尼系统的线性二次型最优控制问题,构建了一个透明且可复现的双网络物理信息神经网络(PINN)基准测试。通过庞特里亚金极小值原理单次射击法和梯形配点直接转录法两种经典方法求解,并重构为约束优化问题,由两个前馈神经网络求解:一个状态网络采用复合三次函数与掩码基底精确施加边界条件,一个无约束控制网络。复合损失函数结合配点处的物理残差与梯形积分形式的成本泛函,由单一标量超参数加权。在该基准上,PINN将最优成本重现至四位有效数字,终端状态约束按构造精确满足,状态与控制的逐点误差落在两种经典参考解的散布范围内。训练速度比经典射击法慢约两个数量级,如实报告。贡献在于方法清晰而非创新:该公式及配套Google Colab实现旨在降低初学者探索基于PINN的最优控制门槛,无需掌握伴随方法或两点边值问题知识。
原文摘要 · Abstract (English)
This work presents a transparent and reproducible benchmark study of a direct dual-network Physics-Informed Neural Network (PINN) formulation for the optimal control of a mass-spring-damper system. The classical linear-quadratic optimal control problem is solved by two independent classical methods -- Pontryagin's Minimum Principle with single shooting, and direct transcription through trapezoidal collocation -- and recast as a constrained optimization problem solved by two feedforward neural networks: a state network whose boundary conditions are enforced exactly through a composite cubic-and-mask ansatz, and an unconstrained control network. The composite loss combines the physics residual at the collocation points with a trapezoidal approximation of the cost functional, weighted by a single scalar hyperparameter. On the benchmark considered, the PINN reproduces the classical optimal cost to four significant digits, satisfies the terminal state constraints exactly by construction, and produces pointwise state and control errors that fall within the spread of the two classical references. Training is approximately two orders of magnitude slower than classical shooting on this benchmark, which is honestly reported. The contribution is methodological clarity rather than methodological novelty: the formulation and the accompanying Google Colab implementation are intended to lower the barrier to entry for practitioners exploring PINN-based optimal control without prior exposure to adjoint methods or two-point boundary value problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。