Search papers, labs, and topics across Lattice.
This paper introduces Informed Neural Controlled Differential Equations (INDEQS), a novel graph-based NCDE method that integrates prior knowledge of directed graph structures into its architecture for enhanced time series forecasting. By differentiating between inner and outer mixing of hidden states and control inputs, INDEQS provides both a lightweight graph-constrained variant and a more expressive variant that adapts graph connections through data-driven learning. Evaluations on synthetic and real-world datasets demonstrate that INDEQS significantly reduces mean absolute error compared to uninformed NCDEs, especially on larger graphs, while also showing that continuous decoders outperform discrete ones in accuracy and temporal flexibility.
Outer informedness in INDEQS consistently reduces forecasting error on complex graphs, outperforming traditional methods even with similar parameter counts.
Neural Controlled Differential Equations (NCDE) provide a powerful continuous-time framework for forecasting time series, but standard graph-based extensions typically learn spatial structure purely from data, even in settings where a directed graph structure is known a priori. We introduce Informed Neural controlled Differential EQuationS (INDEQS), a graph-based NCDE forecasting method that incorporates prior knowledge of a directed graph at distinct architectural positions. INDEQS separates inner mixing of hidden states across graph nodes from outer mixing between vector field and control, and offers both a lightweight graph-constrained variant and a more expressive variant, learning additional graph connections from data via adaptive graph convolutions. To systematically study when graph informedness is beneficial in forecasting, we devise a continuous advection simulation on directed graphs, yielding synthetic spatio-temporal datasets with known ground-truth flow structure. We then evaluate INDEQS on two real-world tasks: river discharge forecasting on a hydrological network and traffic flow prediction on PeMS08. Across these synthetic and real-world benchmarks, outer informedness consistently improves mean absolute error over an uninformed NCDE with comparable parameter count, particularly on larger graphs, while inner informedness offers a more parameter-efficient alternative when strict adherence to a known adjacency is desired. A comparison of discrete convolutional and continuous-time decoders further shows that continuous decoders yield better accuracy and greater temporal flexibility on real-world tasks. An implementation of INDEQS and the advection simulation is available at https://github.com/Mitchi1/indeqs.
