Mar, 2024

使用神经网络对受约束系统进行建模的两阶段训练方法

TL;DR本文详细描述了一种简单、有效且无需惩罚参数的两阶段训练方法,用于模型约束系统。通过将约束优化问题重写为解决两个无约束子问题的两阶段,实现了找到可行神经网络参数和最优神经网络参数。实验证明,该方法可以产生满足约束的模型,并提升预测性能,确保关键系统属性的合规性并减少数据需求量。此外,我们还展示了该方法改善了求解最优解的收敛性和解释可行的神经常微分方程模型的能力。我们的两阶段训练方法适用于任何神经网络架构。