Learning Locally Interacting Discrete Dynamical Systems: Towards Data-Efficient and Scalable Prediction
Beomseok Kang, Harshit Kumar, Minah Lee, Biswadeep Chakraborty, Saibal Mukhopadhyay
TL;DR
AR-NCA addresses predicting evolution in locally interacting discrete dynamical systems by learning unknown local transition rules with a memory-enabled, permutation-invariant neural cellular automaton. It combines a per-cell memory (LSTM) with cellular self-attention to efficiently propagate neighbor dynamics into each cell, yielding data-efficient and scalable predictions. Empirical results on forest fire, host-pathogen, and stock-market-like models show AR-NCA outperforms video-prediction networks and other NCAs, especially under limited data and large spatial scales. The approach offers a practical framework for data-efficient modeling of complex spatiotemporal systems driven by local interactions, with potential extensions to richer state spaces and continuous dynamics.
Abstract
Locally interacting dynamical systems, such as epidemic spread, rumor propagation through crowd, and forest fire, exhibit complex global dynamics originated from local, relatively simple, and often stochastic interactions between dynamic elements. Their temporal evolution is often driven by transitions between a finite number of discrete states. Despite significant advancements in predictive modeling through deep learning, such interactions among many elements have rarely explored as a specific domain for predictive modeling. We present Attentive Recurrent Neural Cellular Automata (AR-NCA), to effectively discover unknown local state transition rules by associating the temporal information between neighboring cells in a permutation-invariant manner. AR-NCA exhibits the superior generalizability across various system configurations (i.e., spatial distribution of states), data efficiency and robustness in extremely data-limited scenarios even in the presence of stochastic interactions, and scalability through spatial dimension-independent prediction.
