arXiv:2504.02260v1 交叉类型: cross 摘要: 通过可微编程的混合神经物理建模框架已经逐渐成为科学机器学习中强大的工具，它们通过将已知的物理知识与数据驱动的学习相结合，提高预测准确性和泛化能力。然而，大多数现有的混合框架依赖于显式的循环公式，这在长时间尺度预测中会导致数值不稳定性和误差累积。在这项工作中，我们引入了Im-PiNDiff，这是一种用于稳定和准确建模时空动力学的新颖隐式物理整合神经可微求解器。受深 equilibrium模型的启发，Im-PiNDiff 使用隐式固定点层推进状态，使其能够实现稳健的长时间模拟，同时保持全程端到端可微。为了实现可扩展的训练，我们引入了一种混合梯度传播策略，该策略结合了伴随状态方法与逆向模式自动微分。该方法消除了存储中间求解器状态的需要，并将内存复杂度与求解器迭代次数解耦，显著减少了训练开销。我们进一步采用检查点技术来管理长时间段滚动中的内存。在各种时空偏微分方程系统上的数值试验，包括对流-扩散过程、Burgers 动力学和多物理化学气相渗透过程，证明了 Im-PiNDiff 达到了优于显式和简单隐式基线的优越预测性能、增强的数值稳定性和显著降低的内存和运行时间成本。这项工作提供了一个原理明确、高效和可扩展的混合神经物理建模框架。