4 个月前

神经常微分方程

神经常微分方程

摘要

我们介绍了一种新的深度神经网络模型家族。与传统的离散隐藏层序列不同,我们使用神经网络来参数化隐藏状态的导数。网络的输出通过一个黑箱微分方程求解器计算得出。这些连续深度模型具有恒定的内存成本,能够根据每个输入调整其评估策略,并且可以显式地在数值精度和速度之间进行权衡。我们在连续深度残差网络和连续时间潜在变量模型中展示了这些特性。此外,我们还构建了连续归一化流(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 中提及
patrick-kidger/FasterNeuralDiffEq
pytorch
GitHub 中提及
kfallah/NODE-Denoiser
pytorch
GitHub 中提及
titu1994/tfdiffeq
tf
GitHub 中提及
rg321/anode_mtvso
pytorch
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 中提及
caidao22/pkg-cams
GitHub 中提及
zhuqunxi/My_numpy_NODE
GitHub 中提及
UofTrees/ProjectX2020
pytorch
GitHub 中提及
abaietto/neural_ode_classification
pytorch
GitHub 中提及
xuanqing94/torchdiffeq
pytorch
GitHub 中提及
Lauu1023/ODENet
pytorch
GitHub 中提及
locuslab/monotone_op_net
pytorch
GitHub 中提及
xwinxu/bayesian-sde
jax
GitHub 中提及
copperwire/neural_ode
tf
GitHub 中提及
BruceNolascoFC/RN-2021-1
pytorch
GitHub 中提及
m-qiang/cortexode
pytorch
GitHub 中提及
000Justin000/torchdiffeq
pytorch
GitHub 中提及
feicccccccc/Neural-ODE-Ex
pytorch
GitHub 中提及
tianfr/mononerf
pytorch
GitHub 中提及
HolmesShuan/OISR-PyTorch
pytorch
GitHub 中提及
Junghwan-brian/SDE-Net
tf
GitHub 中提及
teopb/ode-net-examples
pytorch
GitHub 中提及
alexandrejash/Chivvo
pytorch
GitHub 中提及
RainerGasper/NeuralODE
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-1Latent ODE (RNN enc.)
MSE (10^-2, 50% missing): 1.377
multivariate-time-series-forecasting-on-1RNN-VAE
MSE (10^-2, 50% missing): 1.782
multivariate-time-series-forecasting-on-2RNN-VAE
MSE stdev: 0.145
mse (10^-3): 3.055
multivariate-time-series-forecasting-on-2Latent ODE (RNN enc.)
MSE stdev: 0.052
mse (10^-3): 3.162
multivariate-time-series-forecasting-on-ushcnNeuralODE-VAE
MSE: 0.96
multivariate-time-series-forecasting-on-ushcnNeuralODE-VAE-Mask
MSE: 0.83
multivariate-time-series-imputation-on-1Latent ODE (RNN enc.)
mse (10^-3): 3.907
multivariate-time-series-imputation-on-1RNN-VAE
mse (10^-3): 5.930
multivariate-time-series-imputation-on-mujocoRNN-VAE
MSE (10^2, 50% missing): 6.100
multivariate-time-series-imputation-on-mujocoLatent ODE (RNN enc.)
MSE (10^2, 50% missing): 0.447
pose-estimation-on-2Nate
0..5sec: Jaau

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
神经常微分方程 | 论文 | HyperAI超神经