Skip to content
/ gde Public

Neural Graph Differential Equations (Neural GDEs)

License

Notifications You must be signed in to change notification settings

Zymrael/gde

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Neural Graph Differential Equations (Neural GDEs)

We introduce the framework of continuous-depth graph neural networks (GNNs). Graph neural ordinary differential equations (GDEs) are formalized as the counterpart to GNNs where the input-output relationship is determined by a continuum of GNN layers, blending discrete topological structures and differential equations. The proposed framework is shown to be compatible with various static and autoregressive GNN models. Results prove general effectiveness of GDEs: in static settings they offer computational advantages by incorporating numerical methods in their forward pass; in dynamic settings, on the other hand, they are shown to improve performance by exploiting the geometry of the underlying dynamics.

paper: arXiv link

This repository contains examples of Neural Graph Differential Equations (Neural GDE). The tutorial notebook contains abundant amounts of comments and all runnable top-to-bottom.

Neural GDEs rely on dgl and torchdiffeq.

NOTE: Neural GDE model zoo and additional tutorials are included in the torchdyn library: link

If you find our work useful, consider citing us:

@article{poli2019graph,
  title={Graph Neural Ordinary Differential Equations},
  author={Poli, Michael and Massaroli, Stefano and Park, Junyoung and Yamashita, Atsushi and Asama, Hajime and Park, Jinkyoo},
  journal={arXiv preprint arXiv:1911.07532},
  year={2019}
}

About

Neural Graph Differential Equations (Neural GDEs)

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published