arXiv:2410.16295physics.comp-phcond-mat.dis-nn2024-10ICML被引 6

用Transformer逼近集体行为系统的平均场动态,理论可证效果。

Universal Approximation of Mean-Field Models via Transformers

  • 用Transformer建模粒子系统,天然适配粒子不可区分的对称性。
  • 实验证明能有效逼近群体飞行、神经网络训练等平均场模型。
  • 首次给出Transformer逼近平均场的数学误差界,理论支撑强。

本文研究使用Transformer逼近具有集体行为的相互作用粒子系统的平均场动力学。这类系统在物理、生物和工程中广泛应用,如意见形成、生物网络与群体机器人。其核心特征是粒子不可区分,导致动力学具有置换等变性。我们通过实验表明,Transformer能有效逼近多种平均场模型,包括用于群体飞行的Cucker-Smale模型和两层神经网络训练的平均场系统。通过数学理论验证:若有限维Transformer能有效逼近有限维向量场,则其期望值与无限维平均场向量场之间的$ L_2 $距离可被粒子数的函数统一控制。基于此,我们建立了真实平均场动力学与变压器所生成动力学之间的理论误差边界。

原文摘要 · Abstract (English)

This paper investigates the use of transformers to approximate the mean-field dynamics of interacting particle systems exhibiting collective behavior. Such systems are fundamental in modeling phenomena across physics, biology, and engineering, including opinion formation, biological networks, and swarm robotics. The key characteristic of these systems is that the particles are indistinguishable, leading to permutation-equivariant dynamics. First, we empirically demonstrate that transformers are well-suited for approximating a variety of mean field models, including the Cucker-Smale model for flocking and milling, and the mean-field system for training two-layer neural networks. We validate our numerical experiments via mathematical theory. Specifically, we prove that if a finite-dimensional transformer effectively approximates the finite-dimensional vector field governing the particle system, then the $L_2$ distance between the \textit{expected transformer} and the infinite-dimensional mean-field vector field can be uniformly bounded by a function of the number of particles observed during training. Leveraging this result, we establish theoretical bounds on the distance between the true mean-field dynamics and those obtained using the transformer.

Transformer平均场集体行为

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。