对于神经网络来说,我们已经习惯了层状网络的思维:数据进来,经过第一层,然后第二层,第三层,最后输出结果。这个过程很像流水线,每一步都是离散的。
但是现实世界的变化是连续的,比如烧开水,谁的温度不是从30度直接跳到40度,而是平滑的上生。球从山坡滚下来速度也是渐渐加快的。这些现象背后都有连续的规律在支配。
微分方程就是描述这种连续变化的语言。它不关心某个时刻的具体数值,而是告诉你"变化的速度"。比如说,温度下降得有多快?球加速得有多猛?
Neural ODE的想法很直接:自然界是连续的,神经网络要是离散的?与其让数据在固定的层之间跳跃,不如让它在时间维度上平滑地演化。
微分方程的概念
微分方程其实就是描述变化的规则。
最简单的例子是咖啡冷却。刚泡好的咖啡温度高,冷却很快;温度接近室温时,冷却就变慢了。这个现象背后的规律是:冷却速度和温度差成正比。
比如说:90°C的咖啡在22°C房间里,温差68度,冷却很快;30°C的咖啡在同样环境里,温差只有8度,冷却就慢得多。这就是为什么咖啡从烫嘴快速降到能喝的温度,然后就一直保持温热状态。
这不只是个咖啡的故事,它展示了动态系统的核心特征:当前状态决定了变化的方向和速度。ODE捕捉的正是这种连续演化的规律。
1) 咖啡冷却曲线(指数衰减到室温)
import numpy as np
import matplotlib.pyplot as plt
咖啡冷却曲线 ----------
冷却模型参数:dT/dt = -k (T - T_room)
T0 = 90.0 # 初始温度 (°C)
T_room = 22.0 # 室温 (°C)
k = 0.35 # 冷却常数 (1/min)
t = np.linspace(0, 20, 300) # 分钟
T = T_room + (T0 - T_room) * np.exp(-k * t)
plt.figure(figsize=(7, 5))
plt.plot(t, T, linewidth=2)
plt.title("Coffee Cooling: An ODE in Action")
plt.xlabel("Time (minutes)")
plt.ylabel("Temperature (°C)")
plt.grid(True, alpha=0.3)
coffee_path = "/data/coffee_cooling_curve.png"
plt.tight_layout()
plt.savefig(coffee_path, dpi=200, bbox_inches="tight")
plt.show()
另一个例子是球滚下山坡。球刚开始几乎不动,但重力会让它加速。滚得越快摩擦阻力越大,最终速度会趋于稳定。整个过程可以用一个ODE来描述:
这个方程抓住了两个关键力量:重力让球加速、摩擦让球减速,速度的变化取决于这两个力的平衡。从数学上看,这个简单的方程能完整地描述球从静止到终端速度的整个过程。
import numpy as np
import matplotlib.pyplot as plt # ---------------- 参数 ----------------
g = 9.81 # 重力 (m/s^2)
theta_deg = 15.0 # 坡度角(度)
theta = np.deg2rad(theta_deg)
mu = 0.4 # 线性阻力系数 (1/s) v0 = 0.0 # 初始速度 (m/s)
x0 = 0.0 # 初始位置 (m)
t_end = 12.0 # 总仿真时间 (s)
n_steps = 1200 # 积分步数
---------------- 时间网格 ----------------
t = np.linspace(0.0, t_end, n_steps)
---------------- 向量场 ----------------
def f(y, ti):
x, v = y
dv = gnp.sin(theta) - muv
dx = v
return np.array([dx, dv], dtype=float)
---------------- RK4积分器 ----------------
def rk4(f, y0, t):
y = np.zeros((len(t), len(y0)), dtype=float)
y[0] = y0
for i in range(1, len(t)):
h = t[i] - t[i-1]
ti = t[i-1]
yi = y[i-1]
k1 = f(yi, ti)
k2 = f(yi + 0.5hk1, ti + 0.5h)
k3 = f(yi + 0.5hk2, ti + 0.5h)
k4 = f(yi + hk3, ti + h)
y[i] = yi + (h/6.0)(k1 + 2k2 + 2k3 + k4)
return y
---------------- 数值积分 ----------------
y0 = np.array([x0, v0])
traj = rk4(f, y0, t)
x_num = traj[:, 0]
v_num = traj[:, 1]
---------------- 解析解 ----------------
v_inf = (gnp.sin(theta)) / mu if mu != 0 else np.inf
v_ana = v_inf + (v0 - v_inf) * np.exp(-mu * t)
x_ana = x0 + v_inft + ((v0 - v_inf)/mu) * (1.0 - np.exp(-mu*t))
---------------- 图1:速度 ----------------
plt.figure(figsize=(8.5, 5))
plt.plot(t, v_num, linewidth=2, label="Velocity — RK4 (numeric)")
plt.plot(t, v_ana, linewidth=2, linestyle="--", label="Velocity — analytic")
plt.axhline(v_inf, linestyle=":", label=f"Terminal velocity = {v_inf:.2f} m/s")
plt.title(f"Ball Rolling Downhill — Velocity vs Time (θ={theta_deg:.1f}°, μ={mu})")
plt.xlabel("Time (s)")
plt.ylabel("Velocity (m/s)")
plt.grid(True, alpha=0.3)
plt.legend(frameon=False)
plt.tight_layout()
vel_png = "/mnt/data/ball_downhill_velocity.png"
vel_svg = "/mnt/data/ball_downhill_velocity.svg"
plt.savefig(vel_png, dpi=220, bbox_inches="tight")
plt.savefig(vel_svg, bbox_inches="tight")
plt.show()
---------------- 图2:位置 ----------------
plt.figure(figsize=(8.5, 5))
plt.plot(t, x_num, linewidth=2, label="Position — RK4 (numeric)")
plt.plot(t, x_ana, linewidth=2, linestyle="--", label="Position — analytic")
plt.title(f"Ball Rolling Downhill — Position vs Time (θ={theta_deg:.1f}°, μ={mu})")
plt.xlabel("Time (s)")
plt.ylabel("Position along slope (m)")
plt.grid(True, alpha=0.3)
plt.legend(frameon=False)
plt.tight_layout()
pos_png = "/mnt/data/ball_downhill_position.png"
pos_svg = "/mnt/data/ball_downhill_position.svg"
plt.savefig(pos_png, dpi=220, bbox_inches="tight")
plt.savefig(pos_svg, bbox_inches="tight")
plt.show()
vel_png, vel_svg, pos_png, pos_svg
重力把球往下拉,速度快速上升,但摩擦力越来越大,最终达到终端速度。ODE完美地捕捉了这个平滑的过程。
位置的变化也是如此:开始缓慢,然后加速,最后几乎匀速。这提醒我们,自然界的运动是连续的流,而不是离散的跳跃。
从深度网络到ODE
传统深度学习是离散的:
比如说ResNet的每一层都在做同样事:取当前隐藏状态,加上一些变换,然后传递给下一层。这和数值求解ODE的欧拉方法非常相似——通过小步长逼近连续变化。
或者可以说ResNet其实就是ODE的离散化版本。
更多层应该带来更强的学习能力。但实际上网络太深反而性能下降,原因是梯度消失——学习信号在层层传递中变得越来越弱。
ResNet的关键发现是是引入残差学习。不要求每层学习完整的变换,而是学习一个"修正项":
F(x)是残差,x是跳跃连接传递的原始输入。简单的说:保留原来的信息,只学习需要调整的部分。
跳跃连接字面上就是把输入x加到输出上,这让梯度能更容易地向后传播也防止了信息丢失。通过这个技巧,凯明大佬训练了152层的网络,ResNet不仅赢了2015年的ImageNet竞赛,也成为了现代计算机视觉的基础框架。
这是一个简单的ResNet块实现:
import torch.nn as nn
定义单个ResNet"块"
每个块学习残差函数F(x),然后在最后将输入x加回
class ResNetBlock(nn.Module):
def init(self, in_channels, out_channels):
super().init()
# 第一个卷积层: # - 应用3x3滤波器从输入中提取特征 # - padding=1确保输出大小与输入相同 self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=3, padding=1) # 非线性:ReLU将非线性模式引入网络 self.relu = nn.ReLU() # 第二个卷积层: # - 另一个3x3滤波器来细化特征 # - 仍然保持空间大小不变 self.conv2 = nn.Conv2d(out_channels, out_channels, kernel_size=3, padding=1) def forward(self, x): # 将输入保存为'残差' # 这将在稍后通过跳跃连接加回 residual = x # 通过第一个卷积 + ReLU激活传递输入 out = self.relu(self.conv1(x)) # 通过第二个卷积传递(还没有激活) out = self.conv2(out) # 将原始输入(残差)加到输出上 # 这是使ResNet特殊的"跳跃连接" out = out + residual # 再次应用ReLU以仅保留正激活 return self.relu(out)
关键在最后一行:返回的不是out,而是out + residual。这就是ResNet的精髓。
Neural ODE的核心思想
常规深度网络中,数据要经过固定数量的层。网络深度必须在训练前确定——10层、50层还是100层?Neural ODE彻底改变了这个思路。
不再用离散的层,而是让网络的隐藏状态在时间维度上连续演化。不是"通过100层处理输入",而是"从初始隐藏状态开始,让它按照某个规则连续演化"。
要知道隐藏状态在某个时刻的样子,就用ODE求解器,这个算法会问:状态变化有多快?需要多精确?步长应该多大?
这带来了一个关键特性:自适应深度。标准网络的深度是固定的,但Neural ODE中求解器自己决定需要多少步。简单数据用几步就够了,复杂数据就多用几步,网络在计算过程中自动调整"深度"。
Neural ODE的几个优势:
内存效率:不需要存储所有中间激活,只要起点和终点。
自适应计算:简单问题少用计算,复杂问题多用计算。
连续建模:天然适合物理、生物、金融等连续变化的系统。
可逆性:对生成模型特别有用。
构建Neural ODE
torchdiffeq
是PyTorch的Neural ODE库:
pip install torchdiffeq
import torch
import torch.nn as nn
from torchdiffeq import odeint
定义ODE的动力学函数:
import torch
import torch.nn as nn
class ODEFunc(nn.Module):
def init(self):
super().init()
# 定义参数化f_theta(h)的神经网络
# 输入:h(大小为2的状态向量)
# 输出:dh/dt(h的变化率,也是大小2)
self.net = nn.Sequential(
nn.Linear(2, 50), # 层:从2D状态 -> 50个隐藏单元
nn.Tanh(), # 非线性激活以获得灵活性
nn.Linear(50, 2) # 层:从50个隐藏单元 -> 2D输出
)
def forward(self, t, h): """ ODE函数的前向传播。 参数: t : 当前时间(标量,odeint需要但这里未使用) h : 当前状态(形状为[batch_size, 2]的张量) 返回: dh/dt : h的估计变化率(与h形状相同) """ return self.net(h)
这里f(h, t, θ)是个小神经网络,它描述了隐藏状态如何随时间变化。
设置初始状态和时间:
h0 = torch.tensor([[2., 0.]]) # 起始点
t = torch.linspace(0, 25, 100) # 时间步长
func = ODEFunc() # 你的神经ODE动力学(dh/dt = f(h))
求解ODE:
trajectory = odeint(func, h0, t)
print(trajectory.shape) # (时间, 批次, 特征)
这样我们就把神经网络转换成了连续系统。
案例研究:捕食者-猎物动力学
这是个经典的生态学问题。雪兔和加拿大猞猁的种群数量呈现周期性变化:兔子多了,猞猁有足够食物,数量增加;猞猁多了,兔子被吃得多,数量下降;兔子少了,猞猁没东西吃,数量也下降;猞猁少了,兔子又开始繁盛...这个循环不断重复。
这种动力学天然适合用微分方程建模,Neural ODE可以直接从历史数据中学习这个系统的演化规律,产生平滑的轨迹,并预测未来的种群变化。
更多案例:
github.com/yjndsrt/cn/issues/441
github.com/yjndsrt/cn/issues/440
github.com/yjndsrt/cn/issues/439
github.com/yjndsrt/cn/issues/438
github.com/yjndsrt/cn/issues/437
github.com/yjndsrt/cn/issues/436
github.com/yjndsrt/cn/issues/435
github.com/yjndsrt/cn/issues/434
github.com/yjndsrt/cn/issues/433
github.com/yjndsrt/cn/issues/432
github.com/yjndsrt/cn/issues/431
github.com/yjndsrt/cn/issues/430
github.com/yjndsrt/cn/issues/429
github.com/yjndsrt/cn/issues/428
github.com/yjndsrt/cn/issues/427
github.com/yjndsrt/cn/issues/426
github.com/yjndsrt/cn/issues/425
github.com/yjndsrt/cn/issues/424
github.com/yjndsrt/cn/issues/423
github.com/yjndsrt/cn/issues/422
github.com/yjndsrt/cn/issues/421
github.com/yjndsrt/cn/issues/420
github.com/yjndsrt/cn/issues/419
github.com/yjndsrt/cn/issues/418
github.com/yjndsrt/cn/issues/417
github.com/yjndsrt/cn/issues/416
github.com/yjndsrt/cn/issues/415
github.com/yjndsrt/cn/issues/414
github.com/yjndsrt/cn/issues/413