中文

群体动力学学习:逆优化与 JKO 格式的结合

机器学习 2026-03-04 v3 人工智能 机器学习

摘要

学习群体动力学涉及在给定离散时间点的样本演化快照下,恢复控制粒子演化的底层过程。最近的方法将其表述为概率空间中的能量最小化问题,并利用著名的 JKO 格式进行高效的时间离散化。在这项工作中,我们引入了 iJKOnet\texttt{iJKOnet},这是一种将 JKO 框架与逆优化技术相结合来学习群体动力学的方法。我们的方法依赖于常规的端到端对抗训练过程,不需要限制性的架构选择,例如输入凸神经网络。我们为该方法建立了理论保证,并展示了相比先前基于 JKO 的方法的性能提升。iJKOnet\texttt{iJKOnet} 的代码可在 https://github.com/MuXauJl11110/iJKOnet 获取。

关键词

引用

@article{arxiv.2506.01502,
  title  = {Learning of Population Dynamics: Inverse Optimization Meets JKO Scheme},
  author = {Mikhail Persiianov and Jiawei Chen and Petr Mokrov and Alexander Tyurin and Evgeny Burnaev and Alexander Korotin},
  journal= {arXiv preprint arXiv:2506.01502},
  year   = {2026}
}