Actor-Critic Learning for Extended Mean Field Control with Deterministic Policies
Ziheng Cheng, Xin Guo, Huyên Pham, Yufei Zhang
math.OC, cs.AI, cs.LG, stat.ML
Submitted: 2026-07-13
License: http://creativecommons.org/licenses/by/4.0/
The gist: This paper develops a model-free reinforcement learning framework for continuous--time extended mean field control problems, where both the dynamics and reward may depend on the joint distribution of
Terminology
Abstract
This paper develops a model-free reinforcement learning framework for continuous--time extended mean field control problems, where both the dynamics and reward may depend on the joint distribution of states and controls. We adopt deterministic feedback policies, under which the state--action distribution is induced directly as a push--forward of the state law. This avoids optimization over stochastic kernels and bypasses key limitations of existing approaches in extended mean field settings. We first establish a model--free sensitivity formula for parameterized McKean--Vlasov dynamics and use it to derive a deterministic policy gradient formula expressed through an advantage--rate function on the Wasserstein space. We then refine this formula by introducing local value and advantage--rate representations that depend on the state, action, and joint state--action distribution, yielding a policy gradient that includes both action derivatives and measure--derivative terms with respect to the control distribution. These characterizations lead to a martingale--based learning principle and motivate a continuous--time deep deterministic policy gradient algorithm combining particle approximations, measure--dependent neural networks, temporal--difference learning, and exploration in either action or parameter space. Numerical experiments on stochastic Cucker--Smale consensus control and optimal liquidation with trade crowding demonstrate the efficiency, stability, and robustness of the proposed method, including problems with explicit dependence on the control distribution.
Sources
- Policy Gradient for Continuous-Time Mean-Field Control
- Deterministic Policy Gradient for Reinforcement Learning with Continuous Time and State
- Deterministic Policy Gradient for Learning Equilibrium in Time-Inconsistent Control Problems
- Continuous control with deep reinforcement learning
- Learning operators on labelled conditional distributions with applications to mean field control of non exchangeable systems
- Model-free policy gradient for discrete-time mean-field control
- Extended mean field control: a global numerical solution via finite-dimensional approximation
- Parameter Space Noise for Exploration
- Convergence Rates of Time Discretization in Extended Mean Field Control
- Continuous-time q-learning for mean-field control with common noise, part-I: Theoretical foundations
- Continuous-time q-learning for mean-field control with common noise, part-II: q-learning algorithms
- Learning algorithms for mean field optimal control
- Unified continuous-time q-learning for mean-field game and mean-field control problems
Related papers
- Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noise
- Adam-HNAG: A Convergent Reformulation of Adam with Accelerated Rate
- Incremental Learning in Mirror Flows
- Second-Order Optimization with Asynchronous Curvature
- Online Control via Counterfactual Tracking
- Asynchronous Replanning in Two Population Linear Quadratic Mean Field Games: Information Requirements and Stability