
摘要
我们介绍了一种新的深度神经网络模型家族。与传统的离散隐藏层序列不同,我们使用神经网络来参数化隐藏状态的导数。网络的输出通过一个黑箱微分方程求解器计算得出。这些连续深度模型具有恒定的内存成本,能够根据每个输入调整其评估策略,并且可以显式地在数值精度和速度之间进行权衡。我们在连续深度残差网络和连续时间潜在变量模型中展示了这些特性。此外,我们还构建了连续归一化流(Continuous Normalizing Flows),这是一种可以通过最大似然估计进行训练的生成模型,无需对数据维度进行分割或排序。对于训练过程,我们展示了如何高效地通过任何常微分方程(ODE)求解器进行反向传播,而无需访问其内部操作。这使得可以在更大的模型中端到端地训练常微分方程。
代码仓库
narock/machine_learning
GitHub 中提及
ipeter50/graph_ode
pytorch
GitHub 中提及
5yearsKim/HamiltonianFlow
pytorch
GitHub 中提及
lye0618/torchdiffeq
pytorch
GitHub 中提及
kakao/DAFT
pytorch
GitHub 中提及
xwinxu/bayesde
jax
GitHub 中提及
DrChainsaw/neuralODE4j
pytorch
GitHub 中提及
amirgholami/anode
pytorch
GitHub 中提及
jacobjinkelly/easy-neural-ode
jax
GitHub 中提及
patrick-kidger/FasterNeuralDiffEq
pytorch
GitHub 中提及
kfallah/NODE-Denoiser
pytorch
GitHub 中提及
titu1994/tfdiffeq
tf
GitHub 中提及
rg321/anode_mtvso
pytorch
GitHub 中提及
dlinzner-bcs/min_neuralODEs
GitHub 中提及
geoffsmith/gan-ode
pytorch
GitHub 中提及
gabrevaya/LatentDiffEq
GitHub 中提及
msurtsukov/neural-ode
pytorch
GitHub 中提及
wkirgsn/thermal-nn
tf
GitHub 中提及
ZainRaza14/Neural-ODE
GitHub 中提及
MilesCranmer/lagrangian_nns
jax
GitHub 中提及
TheStageAI/TorchIntegral
pytorch
GitHub 中提及
HosseinEbrahimiK/Normalizing-Flows-Reproducing
pytorch
GitHub 中提及
PranavMahajan25/Neural-ODE
GitHub 中提及
GitTeaching/Predicting-using-Neural-ODE
GitHub 中提及
caidao22/pkg-cams
GitHub 中提及
zhuqunxi/My_numpy_NODE
GitHub 中提及
UofTrees/ProjectX2020
pytorch
GitHub 中提及
abaietto/neural_ode_classification
pytorch
GitHub 中提及
AlphaGergedan/Sampling-HNNs
GitHub 中提及
xuanqing94/torchdiffeq
pytorch
GitHub 中提及
StuartHadfield/neural-odes
GitHub 中提及
Lauu1023/ODENet
pytorch
GitHub 中提及
locuslab/monotone_op_net
pytorch
GitHub 中提及
HosseinEbrahimiK/TrajectoryNet-Reproducing
pytorch
GitHub 中提及
xwinxu/bayesian-sde
jax
GitHub 中提及
copperwire/neural_ode
tf
GitHub 中提及
BruceNolascoFC/RN-2021-1
pytorch
GitHub 中提及
m-qiang/cortexode
pytorch
GitHub 中提及
jrbotvinick/learning-dynamics-on-invariant-measures
pytorch
GitHub 中提及
000Justin000/torchdiffeq
pytorch
GitHub 中提及
feicccccccc/Neural-ODE-Ex
pytorch
GitHub 中提及
gabrevaya/LatentDiffEq.jl
GitHub 中提及
tianfr/mononerf
pytorch
GitHub 中提及
HolmesShuan/OISR-PyTorch
pytorch
GitHub 中提及
Junghwan-brian/SDE-Net
tf
GitHub 中提及
dungxibo123/neural_ode
pytorch
teopb/ode-net-examples
pytorch
GitHub 中提及
alexandrejash/Chivvo
pytorch
GitHub 中提及
RainerGasper/NeuralODE
GitHub 中提及
boschresearch/numerics_independent_neural_odes
pytorch
GitHub 中提及
mandubian/pytorch-neural-ode
pytorch
GitHub 中提及
johnhw/simple_neural_ode
GitHub 中提及
rtqichen/torchdiffeq
官方
pytorch
GitHub 中提及
mandubian/neural-ode
tf
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| multivariate-time-series-forecasting-on-1 | Latent ODE (RNN enc.) | MSE (10^-2, 50% missing): 1.377 |
| multivariate-time-series-forecasting-on-1 | RNN-VAE | MSE (10^-2, 50% missing): 1.782 |
| multivariate-time-series-forecasting-on-2 | RNN-VAE | MSE stdev: 0.145 mse (10^-3): 3.055 |
| multivariate-time-series-forecasting-on-2 | Latent ODE (RNN enc.) | MSE stdev: 0.052 mse (10^-3): 3.162 |
| multivariate-time-series-forecasting-on-ushcn | NeuralODE-VAE | MSE: 0.96 |
| multivariate-time-series-forecasting-on-ushcn | NeuralODE-VAE-Mask | MSE: 0.83 |
| multivariate-time-series-imputation-on-1 | Latent ODE (RNN enc.) | mse (10^-3): 3.907 |
| multivariate-time-series-imputation-on-1 | RNN-VAE | mse (10^-3): 5.930 |
| multivariate-time-series-imputation-on-mujoco | RNN-VAE | MSE (10^2, 50% missing): 6.100 |
| multivariate-time-series-imputation-on-mujoco | Latent ODE (RNN enc.) | MSE (10^2, 50% missing): 0.447 |
| pose-estimation-on-2 | Nate | 0..5sec: Jaau |