如何用DifferentialEquations.jl实现ODE求解器的提前终止？

要实现特定条件满足时让ODE求解器提前终止，你需要使用回调（Callback）机制，这是DifferentialEquations.jl包的标准功能。以下针对你的Lorenz系统需求给出具体实现方案：

1. 规范ODE定义（可选但推荐） #

先修正你的lorenz!函数，补充标准ODE函数所需的p（参数）和t（时间）参数，避免潜在兼容性问题：

using DifferentialEquations

function lorenz!(du, u, p, t)
    du[1] = 10.0*(u[2]-u[1])
    du[2] = u[1]*(28.0-u[3]) - u[2]
    du[3] = u[1]*u[2] - (8/3)*u[3]
end

2. 构建连续监测终止回调 #

针对「当u[3]>10时终止求解」的需求，推荐使用ContinuousCallback——它会连续监测条件，精确捕捉条件触发的时刻：

完整代码示例 #

using DifferentialEquations

function lorenz!(du, u, p, t)
    du[1] = 10.0*(u[2]-u[1])
    du[2] = u[1]*(28.0-u[3]) - u[2]
    du[3] = u[1]*u[2] - (8/3)*u[3]
end

u0 = [1.0; 0.0; 0.0]
tspan = (0.0, 100.0)
prob = ODEProblem(lorenz!, u0, tspan)

# 定义触发条件：当返回值从负变正时触发回调
condition(u, t, integrator) = u[3] - 10.0
# 定义终止动作：停止求解器
affect!(integrator) = terminate!(integrator)
# 创建连续回调实例
cb = ContinuousCallback(condition, affect!)

# 传入回调求解ODE
sol = solve(prob, callback=cb)

代码说明 #

• condition(u, t, integrator)：返回u[3]-10，当该值从负数变为正数（即u[3]刚超过10的瞬间）时触发回调。
• affect!(integrator)：调用terminate!(integrator)直接终止求解过程。
• ContinuousCallback会在求解过程中实时监测条件，确保终止时刻的精度。

3. 离散监测的替代方案 #

如果不需要精确捕捉触发时刻，仅需在每步求解完成后检查条件，可以使用DiscreteCallback，性能略高：

# 离散条件：每步结束后检查u[3]是否大于10
condition_discrete(u, t, integrator) = u[3] > 10.0
cb_discrete = DiscreteCallback(condition_discrete, affect!)

sol = solve(prob, callback=cb_discrete)

验证终止结果 #

求解完成后，可通过以下代码确认终止状态：

println("终止时间: ", sol.t[end])
println("终止状态u[3]: ", sol.u[end][3])

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