Simultaneous inference of environmental and interaction forces in collective dynamics
Organizations: Department of Mathematics, Clarkson University, Potsdam, NY, United States · Department of Mathematics, University of Houston, Houston, TX, United States
Abstract
Collective dynamics arise in a wide range of physical, biological, and engineering applications. Examples include cell migration, swarm robotics, social dynamics, and animal behavior. A defining characteristic of these systems is the emergence of large-scale coordination from local interactions among agents; a fundamental question is thus to understand the local interactions that give rise to the observed emergent dynamics. We are interested in methods for learning interactions generally, which can describe a wide class of physical systems exhibiting collective dynamics defined by an interaction kernel, without a priori assumptions on the analytical form of this kernel (i.e. it is nonparametric). The advantage of this kernel-based approach is that it incorporates the underlying physics of the model (i.e. collective dynamics), which more general equation-learning approaches may ignore, potentially limiting their effectiveness for model accuracy and predictions. In this work, we extend existing variational learning approaches to collective systems with both interaction kernels and environmental/intra-agent forces. The proposed framework simultaneously infers the interaction kernel non-parametrically while learning the environmental force using either semi-parametric or fully nonparametric representations. The methodology is validated on several benchmark models exhibiting synchronization, alignment, attraction-repulsion, and external environmental forces. We also introduce a model-selection procedure based on our nonparametric learning framework to identify models that optimally explain a given set of trajectory observations. By exploiting the feature-identification capability of the learned models, the proposed procedure can distinguish among different collective dynamics frameworks and recover mechanistic interaction mechanisms directly from trajectory data.
Figures & tables
| Candidate ( | Candidate Model | Complexity |
| 1 | ||
| 2 | ||
| 3 | ||
| 3 | ||
| 4 | ||
| 4 |
| Data set | Function |
| Training ( samples) | Obtain estimators for mechanistic forces for each model |
| Validation ( samples) | Compare candidate models and tune hyperparameters to select candidate model |
| Testing ( samples) | Measure final evaluation of models and report performance |
| Metric | Data | Time horizon | Function |
| Obtain admissible frameworks (trajectory data) | |||
| Obtain admissible frameworks (residual data) | |||
| Primary validation metric (trajectories) | |||
| Residual consistency |
| Metric | Non-parametric | Semi-Parametric |
| Data set | Statistic | Fully non-parametric | Semi-parametric |
| Metric | Non-parametric | Semi-Parametric |
| Data set | Statistic | Fully non-parametric | Semi-parametric |
| Metric | Non-parametric | Semi-Parametric |
| Data set | Statistic | Non-parametric | Semi-Parametric |
| Data set | Statistic | Non-parametric | Semi-parametric |
| Kuramoto | |||
| SPP | |||
| Order | |||
| 2 | - | ✓ | - |
| Candidate | Order | Complexity | Rank | Selected | ||||
| 2 | 0.000626 | 0.000004 | 0.000259 | 3 | 0.000509 | 1 | Yes | |
| 2 | 0.000708 | 0.000004 | 0.000437 | 4 | 0.000575 | 2 | No | |
| 2 | 0.000624 | 0.000004 | 0.000255 | 5 | 0.000507 | 3 | No | |
| 2 | 0.000717 | 0.000004 | 0.000464 | 6 | 0.000583 | 4 | No | |
| 1 | 0.593726 | 0.003266 | 0.732061 | 2 | 0.482466 | 5 | No | |
| 2 | 0.593726 | 0.003266 | 0.732061 | 4 | 0.482466 | 6 | No |
| Order | |||
| 2 | - | ✓ | ✓ |
| Candidate | Order | Complexity | Rank | Selected | ||||
| 2 | 0.000067 | 0.000002 | 0.000016 | 4 | 0.000040 | 1 | Yes | |
| 2 | 0.000068 | 0.000002 | 0.000017 | 6 | 0.000041 | 2 | No | |
| 1 | 0.216497 | 0.009056 | 0.102124 | 2 | 0.130341 | 3 | No | |
| 2 | 0.216497 | 0.009056 | 0.102124 | 4 | 0.130341 | 4 | No | |
| 2 | 0.224109 | 0.015654 | 0.272981 | 3 | 0.134924 | 5 | No | |
| 2 | 0.224109 | 0.015654 | 0.272981 | 5 | 0.134924 | 6 | No |
| Order | |||
| 2 | ✓ | ✓ | ✓ |
| Candidate | Order | Complexity | Rank | Selected | ||||
| 2 | 0.063176 | 0.000608 | 0.072231 | 6 | 0.105255 | 1 | Yes | |
| 2 | 0.127328 | 0.003480 | 0.354430 | 4 | 0.212138 | 2 | No | |
| 1 | 0.237008 | 0.003800 | 0.396505 | 2 | 0.394873 | 3 | No | |
| 2 | 0.237008 | 0.003800 | 0.396505 | 4 | 0.394873 | 4 | No | |
| 2 | N/A | N/A | N/A | 3 | N/A | 5 | No | |
| 2 | N/A | N/A | N/A | 5 | N/A | 6 | No |
| Order | |||
| 1 | ✓ | – | – |
| Candidate | Order | Complexity | Rank | Selected | ||||
| 1 | 0.186886 | 0.006227 | 0.016430 | 1 | 0.005228 | 1 | Yes | |
| 1 | 0.186580 | 0.006351 | 0.015867 | 2 | 0.005220 | 2 | No | |
| 2 | 0.930454 | 0.020764 | 0.099662 | 3 | 0.037002 | 3 | No | |
| 2 | 0.927506 | 0.042013 | 0.147101 | 5 | 0.036885 | 4 | No | |
| 2 | 0.983127 | 0.034845 | 0.335785 | 3 | 0.039097 | 5 | No | |
| 2 | N/A | N/A | N/A | 4 | N/A | 6 | No |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| M | L | ||
|---|---|---|---|
| 1 | 51 | 1.3881\text{\times}{10}^{00}$\pm$2.2499\text{\times}{10}^{00} | 3.6814\text{\times}{10}^{01}$\pm$6.4836\text{\times}{10}^{01} |
| 1 | 101 | 1.8696\text{\times}{10}^{00}$\pm$2.0156\text{\times}{10}^{00} | 4.8435\text{\times}{10}^{01}$\pm$5.6091\text{\times}{10}^{01} |
| 1 | 201 | 2.8650\text{\times}{10}^{00}$\pm$3.9872\text{\times}{10}^{00} | 6.4122\text{\times}{10}^{01}$\pm$8.9776\text{\times}{10}^{01} |
| 1 | 401 | 1.7684\text{\times}{10}^{00}$\pm$2.3126\text{\times}{10}^{00} | 4.2537\text{\times}{10}^{01}$\pm$5.9298\text{\times}{10}^{01} |
| 1 | 801 | 2.6533\text{\times}{10}^{00}$\pm$3.9945\text{\times}{10}^{00} | 7.0362\text{\times}{10}^{01}$\pm$1.1683\text{\times}{10}^{02} |
| 4 | 51 | 1.2803\text{\times}{10}^{-02}$\pm$2.1729\text{\times}{10}^{-03} | 8.1026\text{\times}{10}^{-02}$\pm$6.6889\text{\times}{10}^{-02} |
| M | L | ||
|---|---|---|---|
| 1 | 51 | 6.1893\text{\times}{10}^{-03}$\pm$6.2809\text{\times}{10}^{-03} | 6.1180\text{\times}{10}^{-02}$\pm$1.3356\text{\times}{10}^{-02} |
| 1 | 101 | 4.4905\text{\times}{10}^{-03}$\pm$4.3087\text{\times}{10}^{-03} | 1.1115\text{\times}{10}^{-01}$\pm$8.8252\text{\times}{10}^{-02} |
| 1 | 201 | 1.3141\text{\times}{10}^{-02}$\pm$2.9193\text{\times}{10}^{-02} | 8.5239\text{\times}{10}^{-02}$\pm$3.5008\text{\times}{10}^{-02} |
| 1 | 401 | 5.4909\text{\times}{10}^{-03}$\pm$3.4038\text{\times}{10}^{-03} | 8.5608\text{\times}{10}^{-02}$\pm$3.6644\text{\times}{10}^{-02} |
| 1 | 801 | 8.6348\text{\times}{10}^{-03}$\pm$1.2248\text{\times}{10}^{-02} | 7.7069\text{\times}{10}^{-02}$\pm$1.3662\text{\times}{10}^{-02} |
| 4 | 51 | 1.9382\text{\times}{10}^{-04}$\pm$3.1394\text{\times}{10}^{-05} | 1.2214\text{\times}{10}^{-02}$\pm$9.3570\text{\times}{10}^{-03} |
| M | L | ||
|---|---|---|---|
| 1 | 51 | 1.1381\text{\times}{10}^{00}$\pm$1.2043\text{\times}{10}^{00} | 2.7564\text{\times}{10}^{00}$\pm$3.3192\text{\times}{10}^{00} |
| 1 | 101 | 1.4278\text{\times}{10}^{00}$\pm$1.7454\text{\times}{10}^{00} | 1.3453\text{\times}{10}^{00}$\pm$1.0674\text{\times}{10}^{00} |
| 1 | 201 | 5.3246\text{\times}{10}^{-01}$\pm$5.6657\text{\times}{10}^{-01} | 2.1987\text{\times}{10}^{00}$\pm$2.1506\text{\times}{10}^{00} |
| 1 | 401 | 6.1445\text{\times}{10}^{-01}$\pm$5.5824\text{\times}{10}^{-01} | 1.0916\text{\times}{10}^{00}$\pm$1.5755\text{\times}{10}^{00} |
| 1 | 801 | 8.0262\text{\times}{10}^{-01}$\pm$3.6660\text{\times}{10}^{-01} | 1.1428\text{\times}{10}^{00}$\pm$5.8255\text{\times}{10}^{-01} |
| 4 | 51 | 1.3722\text{\times}{10}^{-01}$\pm$3.5001\text{\times}{10}^{-02} | 1.9606\text{\times}{10}^{00}$\pm$4.5055\text{\times}{10}^{00} |
| Parameter | Kuramoto | Phototaxis | SPP | SPP | CS2D | SPPCS | OP |
| profile | - | - | clumps | milling | - | clumps | - |
| System order | 1 | 2 | 2 | 2 | 2 | 2 | 1 |
| State dimension | 1 | 2 | 2 | 2 | 2 | 2 | 1 |
| Number of agents | 10 | 10 | 10 | 10 | 10 | 10 | 10 |
| training trajectories | 20 | 20 | 50 | 50 | 20 | 50 | 20 |
| testing trajectories | 10 | 10 | 10 | 10 | 10 | 10 | 10 |
| Parameter | Kuramoto | Phototaxis | SPP | CS2D | SPPCS | OP |
| Interaction type | Energy | Alignment | Energy | Alignment | Energy Alignment | Energy |
| Radial basis functions | 10 | 8 | 10 | 10 | 10 | 50 |
| Energy basis functions | – | 10 | – | 10 | 10 | 50 |
| Alignment basis functions | – | 10 | – | 10 | 10 | 50 |
| Environment basis parameter | 10 | 10 | 10 | 10 | 10 | 10 |
| Environment basis size per state | 10 | 100 | 100 | 100 | 100 | 10 |
| System | Initial condition |
| Kuramoto | |
| Phototaxis | |
| SPP | |
| CS2D | |
| SPPCS | |
| OP |
| System | Component | Term | ||
| Kuramoto | 0.009\,999\,999\,999\,999\,9$\pm$2.087\,722\,866\,931\,242\,4\text{\times}{10}^{-16} | |||
| Phototaxis | -0.070\,710\,681\,492\,353\,88$\pm$1.423\,730\,942\,286\,253\,9\text{\times}{10}^{-07} | |||
| 1.000\,000\,319\,453\,960\,8$\pm$3.538\,386\,027\,474\,474\text{\times}{10}^{-06} | ||||
| -2.967\,810\,805\,171\,508\text{\times}{10}^{-07}$\pm$2.676\,347\,740\,402\,615\text{\times}{10}^{-06} | ||||
| 0.070\,710\,663\,656\,462\,13$\pm$5.863\,941\,042\,682\,641\text{\times}{10}^{-08} | ||||
| -4.931\,084\,525\,950\,616\text{\times}{10}^{-07}$\pm$2.709\,497\,747\,258\,749\text{\times}{10}^{-06} |
| Parameter | Definition |
|---|---|
| Dynamical system | |
| Number of agents in the system | |
| Dimension of the state space of each agent | |
| State of the -th agent at time | |
| Full stacked state of the system at time | |
| Environmental force function (assumed identical for all agents) | |
Emergent Behavior in Multi-Agent Systems
Emergent behavior in multi-agent systems studies how local agent interactions produce collective patterns such as cooperation, norms, and instability.
See all 172 papers