LLM-ACES: Closed-Loop Discovery of Dynamical Systems with LLM-Guided Adaptive Search
This work addresses the identifiability gap in discovering governing equations from limited data, a key challenge for modeling dynamical systems across scientific domains.
LLM-ACES introduces a closed-loop framework that uses LLM-guided adaptive search to jointly optimize symbolic hypothesis construction and data acquisition for discovering ODEs from data. On 122 ODE systems, it achieves the lowest median NMSE, outperforming baselines by orders of magnitude, with symbolic accuracy of 46.2% and 52.4% on ODEBench and ODEBase, and sample efficiency requiring one-tenth the data.
Recovering governing Ordinary Differential Equations (ODEs) from data is a central challenge in modeling dynamical systems across scientific domains. Existing approaches cast discovery as a static inference problem over fixed datasets, assuming that the observed trajectories are sufficiently informative. However, dynamical systems evolve over large state spaces, and limited data can make multiple equations observationally indistinguishable, leading to identifiability gaps and the recovery of incorrect governing equations. To address this, we introduce LLM-ACES, or LLM-guided Active Closed-loop Equation Search, a closed-loop framework that jointly optimizes symbolic hypothesis construction and adaptive data acquisition. In LLM-ACES, a large language model (LLM) proposes operator priors that partition the large search space into distinct regions, within which candidate equations are fit to the observed data. The disagreement among these candidates guides the acquisition of informative trajectories, creating a feedback loop that iteratively refines both the hypothesis space and the discovered dynamics. On 122 ODE systems spanning ODEBench and ODEBase, LLM-ACES achieves the lowest median NMSE, outperforming state-of-the-art baselines by several orders of magnitude while achieving a high symbolic accuracy of 46.2% and 52.4%, respectively. Our analysis further shows that LLM-ACES is sample-efficient, achieving better performance with one-tenth the data. Furthermore, LLM-ACES's feedback-driven data acquisition makes it robust to noise and recovers the correct symbolic structure, while baselines introduce spurious terms that fit the data locally but obscure the true governing relationships.