Scientific machine learning in ecological systems: A study on the predator-prey dynamics

Ranabir Devgupta,Raj Abhijit Dandekar,Rajat Dandekar,Sreedath Panat
2024-11-11
Abstract:In this study, we apply two pillars of Scientific Machine Learning: Neural Ordinary Differential Equations (Neural ODEs) and Universal Differential Equations (UDEs) to the Lotka Volterra Predator Prey Model, a fundamental ecological model describing the dynamic interactions between predator and prey populations. The Lotka-Volterra model is critical for understanding ecological dynamics, population control, and species interactions, as it is represented by a system of differential equations. In this work, we aim to uncover the underlying differential equations without prior knowledge of the system, relying solely on training data and neural networks. Using robust modeling in the Julia programming language, we demonstrate that both Neural ODEs and UDEs can be effectively utilized for prediction and forecasting of the Lotka-Volterra system. More importantly, we introduce the forecasting breakdown point: the time at which forecasting fails for both Neural ODEs and UDEs. We observe how UDEs outperform Neural ODEs by effectively recovering the underlying dynamics and achieving accurate forecasting with significantly less training data. Additionally, we introduce Gaussian noise of varying magnitudes (from mild to high) to simulate real-world data perturbations and show that UDEs exhibit superior robustness, effectively recovering the underlying dynamics even in the presence of noisy data, while Neural ODEs struggle with high levels of noise. Through extensive hyperparameter optimization, we offer insights into neural network architectures, activation functions, and optimizers that yield the best results. This study opens the door to applying Scientific Machine Learning frameworks for forecasting tasks across a wide range of ecological and scientific domains.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is how to use Scientific Machine Learning (SciML) methods to predict and understand population dynamics in the predator - prey dynamics of ecosystems. Specifically, the paper focuses on the following key issues: 1. **Can Universal Differential Equations (UDEs) be used to learn the interaction terms in the predator - prey system**: By replacing certain terms in traditional models with neural networks, can UDEs effectively discover unknown interaction mechanisms? 2. **Comparison of the prediction performance of Neural ODEs and UDEs in ecological modeling**: Study the prediction effects of Neural ODEs and UDEs in the predator - prey model, and evaluate which method performs better under different amounts of data. 3. **Data requirements and prediction failure points of Neural ODEs and UDEs**: Explore whether these two methods can recover interaction terms from less training data, and determine their "forecasting breakdown point", that is, the point in time when the prediction accuracy drops significantly. 4. **Is UDEs more effective in prediction than Neural ODEs**: Especially in the presence of noisy data, can UDEs better recover the underlying dynamics of the system and achieve more accurate predictions? To answer these questions, the author selected the classic Lotka - Volterra predator - prey model as a case study. This model is described by the following two first - order ordinary differential equations: \[ \frac{dx}{dt}=\alpha x-\beta xy \] \[ \frac{dy}{dt}=-\delta y + \gamma xy \] where \(x(t)\) represents the number of prey populations, \(y(t)\) represents the number of predator populations, and \(\alpha, \beta, \gamma, \delta\) are constant parameters that control the interaction dynamics. Through experiments, the author demonstrated the advantages of UDEs in the following aspects: - **Recovering underlying dynamics from less training data**: UDEs can more effectively learn the underlying dynamics of the system from a small amount of training data. - **Robustness to noisy data**: UDEs show better robustness in a high - noise environment and can more accurately recover the underlying dynamics of the system. - **Prediction failure point**: The prediction failure point of UDEs is later than that of Neural ODEs, indicating its superiority in long - term prediction tasks. In addition, the author also carried out extensive hyper - parameter optimization to explore the influence of neural network architecture, activation functions, and optimization strategies on the performance of the two methods. These research results provide valuable insights for the application of scientific machine learning frameworks in ecology and other scientific fields.