Dymos拆分ODE为两个组件时出现IPOPT恢复失败问题

问题背景 #

基于赛车最优控制示例构建的OpenMDAO模型，包含三个独立组件：

• 车辆动力学模型：根据轮胎受力与当前速度计算车辆坐标系下的加速度
• 轮胎力计算组件：通过线性关系，基于车辆速度（状态量）与施加扭矩（控制量）计算轮胎力
• 轮胎抓地力限制组件：基于摩擦圆实现抓地力约束

模型拆分后运行时出现IPOPT恢复失败，复步法检查总导数显示多个缺陷约束对状态量的导数超出绝对/相对容差，但将车辆动力学与轮胎力计算组件合并后，优化运行完全正常。已通过复步法验证所有单个组件的偏导数无误，尝试将滑移角计算移至TireForce组件后问题仍存在。

复步法导数检查输出 #

'traj.phase0.collocation_constraint.defects:r'    | 'traj.phase0.states:n'  |  9.8677e-02 |  9.8723e-02 |  3.0820e-03 |  3.1218e-02 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:r'    | 'traj.phase0.states:r'  |  1.0607e+01 |  1.0607e+01 |  2.3431e-03 |  2.2091e-04 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:r'    | 'traj.phase0.states:u'  |  7.0988e-02 |  7.1729e-02 |  4.4107e-03 |  6.1491e-02 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:r'    | 'traj.phase0.states:v'  |  2.7646e-02 |  2.9618e-02 |  3.5225e-03 |  1.1893e-01 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:r'    | 'traj.phase0.states:xi' |  1.6765e+00 |  1.6593e+00 |  5.8082e-02 |  3.5004e-02 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:v'    | 'traj.phase0.states:n'  |  1.3974e+00 |  1.3982e+00 |  7.9276e-03 |  5.6700e-03 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:v'    | 'traj.phase0.states:r'  |  4.4467e+01 |  4.4466e+01 |  6.6054e-03 |  1.4855e-04 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:v'    | 'traj.phase0.states:u'  |  7.9225e-01 |  7.9746e-01 |  1.1748e-02 |  1.4732e-02 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:v'    | 'traj.phase0.states:v'  |  1.0619e+01 |  1.0620e+01 |  9.0666e-03 |  8.5377e-04 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+
| 'traj.phase0.collocation_constraint.defects:v'    | 'traj.phase0.states:xi' |  1.2890e+01 |  1.2893e+01 |  1.3847e-01 |  1.0740e-02 |  >ABS_TOL >REL_TOL |
+---------------------------------------------------+-------------------------+-------------+-------------+-------------+-------------+--------------------+

可能原因分析 #

1. 数值精度累积误差：组件拆分后，输入输出传递过程中引入的微小浮点误差，在链式求导时被放大，导致总导数与复步法结果的偏差超出容差。合并组件后直接计算内部变量导数，避免了跨组件的数值传递误差。
2. 雅可比组装的链式误差：虽然单个组件的偏导数验证正确，但OpenMDAO在组装总雅可比矩阵时，多层组件的链式计算可能引入累积误差，尤其是当组件间存在强耦合时。
3. 变量量级不匹配：拆分后各组件的输出变量量级差异较大，未进行合理缩放，导致缺陷约束对状态量的导数敏感性失衡，超出IPOPT的默认容差阈值。
4. 耦合关系的显式处理不足：车辆动力学与轮胎力之间存在强耦合，拆分后这种耦合变为跨组件的显式传递，缺乏隐式求解器的协调，导致导数计算的稳定性下降。

解决方案建议 #

• 统一变量缩放：对轮胎力、加速度等关键输出变量设置ref和ref0参数，将变量量级统一到相近范围（如[-1,1]或[0,1]），减少数值误差传播。
• 调整IPOPT配置：适当放宽abs_tol、rel_tol或constr_viol_tol参数，同时可尝试设置max_iter增加迭代次数，给IPOPT更多恢复空间。
• 优化导数验证精度：使用check_totals时设置step=1e-6或更小步长，重新验证总导数，确认偏差是否由步长过大导致；同时启用compact_print=True查看总雅可比的整体分布。
• 优化组件连接：检查组件间的数据流，移除不必要的中间变量传递，减少跨组件的依赖层级；对于强耦合部分，考虑保留为内部变量而非跨组件输入输出。
• 切换优化器验证：先用ScipyOptimizeDriver（如SLSQP）验证模型的可行性，若能正常优化，则问题大概率是IPOPT的配置或数值敏感性导致，可针对性调整IPOPT参数。
• 引入隐式组件处理耦合：将车辆动力学与轮胎力部分封装为隐式组件，使用Newton求解器处理内部耦合，提升导数计算的稳定性。

内容的提问来源于stack exchange，提问作者Tido