Learning Sampled-data Control for Swarms via MeanFlow
Authors: Anqi Dong, Yongxin Chen, Karl H. Johansson, Johan Karlsson
Organizations: Department of Decision and Control Systems, Department of Mathematics, and the Digital Future, KTH Royal Institute of Technology, SE-114 28 Stockholm, Sweden · Institute for Robotics and Intelligent Machines, Georgia Institute of Technology, Atlanta, GA 30332, USA · Department of Decision and Control Systems and the Digital Future, KTH Royal Institute of Technology, Lindstedtsvägen 25, SE-114 28 Stockholm, Sweden · Department of Mathematics and the Digital Future, KTH Royal Institute of Technology, Lindstedtsvägen 25, SE-114 28 Stockholm, Sweden
Steering large-scale swarms with only limited control updates is often needed due to communication or computational constraints, yet most learning-based approaches do not account for this and instead model instantaneous velocity fields. As a result, the natural object for decision making is a finite-window control quantity rather than an infinitesimal one. To address this gap, we consider the recent machine learning framework MeanFlow and generalize it to the setting with general linear dynamic systems. This results in a new sampled-data learning framework that operates directly in control space and that can be applied for swarm steering. To this end, we learn the finite-horizon coefficient that parameterizes the minimum-energy control applied over each interval, and derive a differential identity that connects this quantity to a local bridge-induced supervision signal. This identity leads to a simple stop-gradient regression objective, allowing the interval coefficient field to be learned efficiently from bridge samples. The learned policy is deployed through sampled-data updates, guaranteeing that the resulting controller exactly respects the prescribed linear time-invariant dynamics and actuation channel. The resulting method enables few-step swarm steering at scale, while remaining consistent with the finite-window actuation structure of the underlying control system.
Figures & tables
Fig. 1: Swarm distribution evolves from an initial law ρ0 at t0=0 to a target law ρ1 at tK=1 under the linear dynamics z˙=Az+Bu . The proposed model learns an interval coefficient ck=cθ(zk,tk,tk+1) , which generates the finite-horizon minimum-energy control over [tk,tk+1] .
z0∼ρ0,z1∼ρ1,t∼U(0,1),r∼U(t,1)
Algorithm 1 Training the interval-coefficient model cθ
Multi-agent control coordinates interacting autonomous agents to achieve shared goals while respecting safety, communication, and resource constraints.