Entropy-Regularized Optimal Transport
for Time-Varying Multi-Agent Coverage Control

Coordinating mobile agents toward a desired spatial configuration is a fundamental problem in the control of multi-agent systems. In many applications, however, prescribing individual target positions is unnecessarily restrictive: the objective is instead specified at the collective or probabilistic level through a desired spatial distribution [1, 2]. This motivates a distributional formulation in which the agents are represented by their empirical measure and are steered toward a possibly time-varying target density, without prescribing an individual target position for each agent.

Optimal transport (OT) provides a natural framework for this problem, as the Wasserstein distance incorporates the geometry of the physical domain and allows discrete agent distributions to be compared directly with the continuous target densities [3]. While several approaches to density control rely on mean-field or continuum descriptions [4, 5, 6], here we retain the finite-dimensional nature of the multi-agent dynamics and formulate the resulting discrete-to-continuous problem through semi-discrete OT [7]. This provides a direct connection between individual agent positions and collective objectives in probability space. This perspective is also related to coverage control, where spatial densities encode the relative importance of different regions and classical formulations lead to Voronoi-based deployments [8, 9, 2]. Recent works have connected coverage objectives with semi-discrete OT and Laguerre partitions [10, 11, 12]. Building on this interpretation, our previous work [13] formulated time-varying coverage as an unregularized semi-discrete OT problem, in which agent positions and Kantorovich dual variables evolve jointly to track the first-order optimality conditions associated with a time-varying density. Such a formulation, however, inherits the hard partition induced by Laguerre cells, so that transport assignments change through moving cell boundaries. This lack of smoothness may be undesirable in a continuously evolving feedback problem, as moving cell boundaries complicate sensitivities, require geometric regularity assumptions, and lead to nonsmooth closed-loop analysis.

In this work, we introduce entropic regularization into the time-varying semi-discrete optimal transport formulation we first proposed in its unregularized form in [13]. Entropic OT replaces hard cell-based assignments with smooth transport weights, thereby regularizing the dependence of the transport problem on both the agent positions and the dual variables [3, 7]. It has also been employed for model predictive control in the discrete-discrete setting [14]. In contrast to these existing uses, the present work employs entropy not merely as a computational tool, as is common in OT [15], but as a control-oriented regularization that facilitates the design of smooth continuous-time feedback laws. The regularization parameter provides a natural trade-off between fidelity to the original Wasserstein problem and smoothness of the induced transport allocation. More broadly, the resulting formulation provides a framework for time-varying multi-agent density tracking and formation control in probability space. Unlike continuum approaches, it acts directly on a finite collection of agents, while allowing the target density to encode time-varying formations without prescribing individual agent assignments. In this Letter, we focus on the centralized setting. Our contributions are threefold: (i) we formulate time-varying density tracking for finite-agent systems as an entropy-regularized semi-discrete OT problem; (ii) we derive the associated regularized optimality conditions and continuous-time dynamics for the agent positions and dual variables, replacing the nonsmooth Laguerre partition with smooth transport weights; and (iii) we investigate the influence of the entropic regularization parameter.

Scalars are lowercase ($x$), vectors bold lowercase ($\mathbf{x}$), and matrices bold uppercase ($\mathbf{A}$), with maximum eigenvalue $\lambda_{\mathrm{max}}(\mathbf{A})$. The operator $\text{vec}(\cdot)$ stacks vectors, e.g., $\mathbf{p}:=\text{vec}(\mathbf{p}_{1},\ldots,\mathbf{p}_{N})$. The $N$-dimensional identity matrix is denoted by $\mathbf{I}_{N}$. Matrices of zeros and ones are denoted by $\mathbf{0}_{N\times M}$ and $\mathbf{1}_{N\times M}$, respectively. When $M=1$, they are $\mathbf{0}_{N}$ and $\mathbf{1}_{N}$. The Kronecker product is $\otimes$, while $\delta_{ij}$ is the Kronecker delta. The set of Borel probability measures on $\mathbb{R}^{d}$ with finite second moment is $\mathcal{P}_{2}(\mathbb{R}^{d})$, and $\mathcal{W}_{p}(\cdot,\cdot)$ denotes the $p$-Wasserstein distance. As in [10], $\mathcal{W}_{2}^{2}$ is the transport value associated with the half-squared Euclidean cost $\tfrac{1}{2}\lVert\cdot\rVert_{2}^{2}$.

Building upon [13], our aim is to control an ensemble of finite $N$ agents to approximate a continuous, possibly time-varying distribution by minimizing the instantaneous entropic semi-discrete transport cost.

Consider a convex compact domain $\Omega\subseteq\mathbb{R}^{d}$, and let $\bar{\rho}:\Omega\times\mathbb{R}_{\geq 0}\to\mathbb{R}_{>0}$ denote a (possibly time-varying) target density function with unitary mass. The absolutely continuous target measure $\bar{\mu}_{t}\in\mathcal{P}_{2}(\Omega)$ is induced by $\bar{\rho}(\mathbf{x},t)$.

This guarantees that $\bar{\rho}$ is sufficiently regular and $\partial_{t}\bar{\rho}$ exists. The $i$-th controlled agent is located at position $\mathbf{p}_{i}(t)\in\mathbb{R}^{d}$ at time $t$, and we collect the positions of all $N$ agents in $\mathbf{p}:=\mathrm{vec}(\mathbf{p}_{1},\dots,\mathbf{p}_{N})$. Each agent evolves according to:

where $\mathbf{u}_{i}(t)\in\mathbb{R}^{d}$ is the control input driving the $i$-th agent. The empirical measure of the agents at time $t$, denoted by $\mu_{t}$, is defined as the normalized sum of Dirac delta measures centered at the agent positions:

Assuming that the agents have access to $\bar{\rho}$ and its time derivative $\partial_{t}\bar{\rho}$, the objective is to design a control law that computes $\mathbf{u}_{i}(t)$ for $i=1,\dots,N$, so that, at each time $t$, the positions of the agents minimize the mismatch between the target measure $\bar{\mu}_{t}$ and the empirical measure (6), i.e.,

for the entropy regularization parameter $\varepsilon>0$, with $F^{\varepsilon}$ given by (3). Note that the instantaneous Wasserstein minimization is fundamentally different from dynamic OT in the Benamou-Brenier formulation [19].

For every $(\mathbf{p},\boldsymbol{\phi})$ and every $\varepsilon>0$, the soft assignment (4) satisfies $\Pi_{i}(\mathbf{x},\mathbf{p}(t),\boldsymbol{\phi}(t))>0$. We define the regularized masses and barycenters as

Thus, $a_{i}^{\varepsilon}>0$ and every $\mathbf{b}_{i}^{\varepsilon}$ is well defined, without requiring an additional nonempty-cell assumption. Here $a_{i}^{\varepsilon}\in\mathbb{R}_{>0}$ and $\mathbf{b}_{i}^{\varepsilon}\in\Omega$, and we collect them as $\mathbf{a}^{\varepsilon}:=\mathrm{vec}(a_{1}^{\varepsilon},\dots,a_{N}^{\varepsilon})\in\mathbb{R}^{N}_{>0}$ and $\mathbf{b}^{\varepsilon}:=\mathrm{vec}(\mathbf{b}_{1}^{\varepsilon},\dots,\mathbf{b}_{N}^{\varepsilon})\in\Omega^{N}$. These converge to their unregularized counterparts, the mass and barycenter of the Laguerre cell (2), as $\varepsilon\to 0$ [3]. Direct differentiation yields

Setting (9) to zero, the regularized first-order conditions are

We will then design dynamics for $(\mathbf{p},\boldsymbol{\phi})$ to achieve the above conditions. To do so, we first differentiate (8) in time, obtaining

where $\partial_{t}\bar{\rho}$ is given, while

Exploiting the softmax properties, we have

Therefore, we define the sensitivity matrix $\mathbf{S}_{pb}^{\varepsilon}\in\mathbb{R}^{dN\times dN}$, with $d\times d$ block entries

This construction allows us to compactly write

with

We now exploit this structure to design primal and dual dynamics that track the time-varying first-order optimality conditions (10) by directly imposing the desired error dynamics. Since the dual variables $\boldsymbol{\phi}$ are defined up to an additive constant, we remove one decision variable to resolve this ambiguity and, without loss of generality, set $\phi_{N}=0$. The inner objective is concave and becomes strictly concave once this gauge is fixed; the outer minimization, however, is nonconvex, as is standard for this class of problems [10, 13, 8]. We therefore target stationary points.

In all experiments, the integrals in (11) are evaluated by tensor-product quadrature on a uniform grid of $50$ points per dimension in the 2D experiment and $5000$ points in the 1D experiment, (14) is integrated with forward Euler with $\Delta t=0.01$, the gains are $K_{x}=10$, $K_{\phi}=10$, and $\boldsymbol{\phi}(0)=\mathbf{0}_{N}$. We first validate the proposed approach in a representative two-dimensional example. We consider $N=6$ agents tasked with covering an evolving density and set $\varepsilon=6$. The initial density is Gaussian $\bar{\rho}(\mathbf{x},0)=\mathcal{N}([10,10]^{\top},10\mathbf{I}_{2})$ and evolves according to the advection-diffusion equation

complemented with no-flux boundary conditions. We consider a uniform drift $\mathbf{v}=[4,8]^{\top}$ and a diffusion coefficient $D=2$. This setting may represent, for instance, a contaminant plume transported by a uniform fluid current while diffusing  [20, 2]. The agents must therefore continuously redistribute to monitor the evolving concentration field. Figure 1a–c shows snapshots at the initial, intermediate, and final times, with $\partial_{t}\bar{\rho}$ computed directly from (20). Starting from random initial positions uniform in $[0,15]^{2}$, the agents redistribute to cover the density while tracking its drift, and progressively spread out as it diffuses, adapting to the varying target density. The mass and positions error also exponentially decay to zero (their evolutions are omitted here for the sake of brevity). The Wasserstein distance decreases rapidly during the initial transient and then grows linearly, as diffusion increases the variance of the target density and hence the finite-agent quantization error (Fig. 1d).

Next, we investigate the effect of $\varepsilon$ in a one-dimensional domain. We consider $N=5$ agents covering a Gaussian density with a time-varying mean, namely, $\bar{\rho}(x,t)=\mathcal{N}(m(t),9)$, where $\dot{m}(t)=2t$ and $m(0)=10$, over a time horizon $t\in[0,5]$. Figures 2(a)–2(c) show the final agent configurations for different values of $\varepsilon$. For $\varepsilon=0.1$, the agents effectively cover the density while maintaining distinct positions: three agents are closely spaced near the peak of the Gaussian, whereas the remaining two lie farther away toward its tails. For the intermediate value $\varepsilon=6$, the three central agents collapse to the same position, while the two outer agents remain distinct. Finally, for $\varepsilon=10>\lambda_{\mathrm{max}}(\Sigma)=9$, all agents collapse to a single point corresponding to the global barycenter of the target density, consistently with Proposition 2. Figure 2(d) further confirms these theoretical findings, where we vary $\varepsilon\in[0.1,15]$. For small values of $\varepsilon$, the standard deviation of the agent positions, normalized by its value in the unregularized case, remains close to one, indicating that the ensemble spreads similarly to the baseline configuration. As $\varepsilon$ increases, the normalized standard deviation decreases and becomes zero near the theoretical critical threshold, remaining zero for all values exceeding the variance of the target density. Accordingly, the normalized Wasserstein distance remains close to one for small $\varepsilon$ and increases to approximately 3 when all agents collapse.

Finally, we compare the proposed solution with the Voronoi-based approach [9], obtained in the unregularized case by setting $\boldsymbol{\phi}(t)=\mathbf{0}_{N}$ for all $t\geq 0$. Since the Voronoi partition does not enforce equal cell masses, the agents concentrate less around the Gaussian peak than under the unregularized OT solution, resulting in a larger final approximation error. Moreover, the entropic controller remains more accurate for all tested values $\varepsilon\leq 3$, as reported in Table I for four representative values of $\varepsilon$. Finally, the suboptimality gap $\mathcal{W}_{2,\varepsilon}^{2}-\mathcal{W}_{2}^{2}$ consistently remains below the upper bound $\varepsilon\log N$ established in Proposition 1. Notice that at a collapsed configuration, all agents coincide and the regularized optimal coupling has zero KL divergence, consequently, $\mathcal{W}_{2,\varepsilon}^{2}=\mathcal{W}_{2}^{2}$.

This paper introduced an entropy-regularized optimal transport framework for time-varying coverage control, with application to multi-agent density tracking. The smooth assignment induced by entropic regularization enabled a control law that exponentially tracks the evolving first-order optimality conditions, together with an explicit threshold above which the collapsed configuration becomes locally optimal. Numerical experiments confirmed the theory and, for moderate regularization, have comparable performance with our unregularized approach [13], while outperforming Voronoi-based coverage [9]. Future work will focus on distributed implementations, adaptive selection of $\varepsilon$, extensions to constrained dynamics and collision avoidance, and a theoretical characterization of the conditions ensuring the invertibility of $\mathbf{M}^{\varepsilon}$.