Neural Ordinary Differential Equation Control of Dynamics on Graphs

We study the ability of neural networks to steer or control trajectories of dynamical systems on graphs, which we represent with neural ordinary differential equations (neural ODEs). To do so, we introduce a neural-ODE control (NODEC) framework and find that it can learn control signals that drive graph dynamical systems into desired target states. While we use loss functions that do not constrain the control energy, our results show that NODEC produces control signals that are highly correlated with optimal (or minimum energy) control signals. Finally, we empirically showcase the high performance and versatility of NODEC for various (non-)linear dynamics and loss functions on different graphs.