Hypothesize, Evaluate, Refine: A Scientific Agent for PDE Discovery with Unknown Spatial Coefficient Fields
Organizations: Fujian University of Technology · Fujian University of Technology Fujian Provincial Key Laboratory of Big Data Mining and Applications · Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences
Abstract
Discovering PDEs in heterogeneous media requires jointly identifying the governing operator and the unknown spatial fields that parameterize it. These tasks are coupled: changing field placement changes the differential law, while a sufficiently flexible field can conceal structural error on a single trajectory. We present Hypothesize, Evaluate, Refine for PDE Discovery (HER-PDE), a scientific-agent framework that discovers compositional PDE structure together with nonparametric, time-invariant coefficient fields. The Agent analyzes two noisy trajectories generated by different excitations, proposes complete expression-tree hypotheses, and combines creative structural exploration with local candidate refinement. Its Hypothesis Evaluation Interface (HEI) estimates only the fields explicitly declared in each hypothesis, never adds missing terms, and scores structures by bidirectional cross-excitation transfer. The selected law is subsequently audited on a sealed temporal interval. Across five controlled two-dimensional systems observed with 5 percent relative Gaussian state noise, the Agent recovers the generating operator in all five cases, including equivalent signed-field and product-rule parameterizations. Across nine unknown coefficient fields, the recovered fields attain a median Pearson correlation of approximately 0.85 and a median relative L2 error of approximately 0.28. These results show that agent-guided hypothesis refinement can recover heterogeneous governing laws without prescribing a parametric form for their spatial coefficients.
Figures & tables
| Command | Agent-visible operation and response | Cost |
|---|---|---|
| pde status | Reports the unique-evaluation budget, submission state, and current archive winner. | None |
| pde validate <file> | Parses a complete candidate and checks the candidate-language and evolution-law constraints without fitting fields. | None |
| pde score <file> | Canonicalizes and evaluates one complete candidate. The response contains its identifier, validity or stable failure code, directional transfer errors, symmetric score, rank, cache state, and counters. | 1/new |
| pde score-batch <files> | Applies score independently to up to 32 candidates; batching changes throughput but not candidate semantics. | 1/new |
| pde history | Returns the scored archive in chronological or ranked order without creating a new evaluation. | None |
| pde evidence <id> | Exports the candidate’s already-fitted coefficient fields on common anchors, together with conditioning, selected hyperparameters, and cross-trajectory stability diagnostics. | None |
| Case | Generating law | Agent-selected law | Structural relation | Search | Sealed |
|---|---|---|---|---|---|
| Groundwater flow | Exact; | 0.5140 | 0.1226 | ||
| Variable diffusion | Exact; | 0.2093 | 0.2890 | ||
| Reaction–diffusion | Product-rule equivalent; , | 0.4506 | 0.4272 | ||
| Advection–diffusion | Signed-field equivalent; , | 0.4944 | 0.5114 | ||
| Orthotropic diffusion | Exact; | 0.2041 | 0.2761 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Stage | Candidate or test | Agent update | |
|---|---|---|---|
| Phase I | Exploratory derivatives and cross-trial statistics | — | Enumerated diffusion, wave, advection, and reaction–diffusion as competing explanations. |
| Phase I | invalid | Used HEI’s language failure to rule out the second-order-time branch and restrict subsequent proposals to measured . | |
| Eval. 1 | 1.0246 | The Agent-authored zero multiplier makes this the effective null law , establishing the first numerical baseline. | |
| Eval. 2 | 1.4641 | Found that spatially varying diffusivity alone did not explain transfer between the two preparations. | |
| Eval. 8 | 4.2552 | Ruled out heterogeneous transport without a second-order component. |
| Case | Chronological structural progression | Selection evidence |
|---|---|---|
| Groundwater flow | Flat heterogeneous diffusion was followed by the shared conservative construction . Reaction, source, and heterogeneous transport extensions all increased the error. | A child with a unit -gradient term reached 0.5132, but added unsupported structure. The Agent submitted the compact parent; its sealed error, 0.122599, was also slightly below the child’s 0.122641. |
| Variable diffusion | The first candidate, , remained rank one while the Agent tested split directional fields (0.5045), reaction (0.3767), one-coordinate dependence (0.2628), and nonlinear modulation (0.2287). | The Agent retained the first candidate after 32 distinct scores. Its sealed error was 0.2890. |
| Reaction–diffusion | Reaction alone gave way to split diffusion plus reaction , then to a fully expanded local model . Linking the derivative and second-order terms through one shared diffusivity produced the conservative product-rule expansion . | Independent directional diffusivities scored 0.4592 and extra advection scored 0.4507. The Agent submitted the shared-field expansion; its sealed error was 0.4272. |
| Advection–diffusion | Diffusion-only and transport-only candidates exceeded 1.0. Their joint construction with reaction reached 0.6588 at evaluation 31; deleting the reaction produced the exact three-field structure at evaluation 37 with error 0.4944. | Missing-direction, linked-drift, nonlinear-reaction, conservative-placement, and split-diffusion controls were all worse. The sealed error was 0.5114. |
| Orthotropic diffusion | Shared isotropic diffusion improved to a two-field nonconservative diagonal form . Placing each field inside its directional derivative produced . | The explicit shared-field product-rule expansion at evaluation 22 had the identical score. The Agent submitted the compact nested form; its sealed error was 0.2761. |
| Rank | Eval. | Candidate equation | Structural audit | |
|---|---|---|---|---|
| 1 | 23 | 0.513184 | Integer-weighted gradient extension | |
| 2 | 11 | 0.513982 | Target; submitted | |
| 3 | 14 | 0.691988 | Extra spatial reaction | |
| 4 | 15 | 0.692116 | Extra spatial source | |
| 5 | 1 | 1.001298 | Agent-authored null law |
| Rank | Eval. | Candidate equation | Structural audit | |
|---|---|---|---|---|
| 1 | 1 | 0.209334 | Target; submitted | |
| 2 | 32 | 0.228653 | Nonlinear modulation | |
| 3 | 19 | 0.262803 | Missing dependence | |
| 4 | 21 | 0.304122 | Extra nonlinear term | |
| 5 | 17 | 0.365506 | Extra nonlinear reaction |
| Rank | Eval. | Candidate equation | Structural audit | |
|---|---|---|---|---|
| 1 | 7 | 0.450614 | Target; submitted | |
| 2 | 18 | 0.450719 | Extra spatial transport | |
| 3 | 5 | 0.451745 | Five independent fields | |
| 4 | 12 | 0.459191 | Split directional diffusion | |
| 5 | 16 | 0.462034 | Independent local fields |
| Rank | Eval. | Candidate equation | Structural audit | |
|---|---|---|---|---|
| 1 | 37 | 0.494444 | Target; submitted | |
| 2 | 62 | 0.547559 | Conservative diffusion placement | |
| 3 | 52 | 0.625828 | Extra nonlinear reaction | |
| 4 | 31 | 0.658754 | Extra linear reaction | |
| 5 | 39 | 0.675537 | Split diffusion and reaction |
| Rank | Eval. | Candidate equation | Structural audit | |
|---|---|---|---|---|
| 1 | 15 | 0.204067 | Target; submitted | |
| 2 | 22 | 0.204067 | Product-rule equivalent | |
| 3 | 29 | 0.247207 | Extra nonlinear term | |
| 4 | 4 | 0.298445 | Missing field derivatives | |
| 5 | 2 | 0.351889 | Shared isotropic field |
| Work / citation | Structure | Coefficient | Note |
|---|---|---|---|
| Equations of motion ( Crutchfield and McNamara 1987 ) | Mixed | Constant coefficients | Structure depends on representation. |
| GP structure ID ( Gray et al. 1998 ) | Open-form library | Constant coefficients | |
| GP-ODE ( Cao et al. 2000 ) | Open-form library | Constant coefficients | |
| ARE ( Bongard and Lipson 2007 ) | Open-form library | Constant coefficients | |
| Free-form laws ( Schmidt and Lipson 2009 ) | Open-form library | Constant coefficients |