MADQRL: Distributed Quantum Reinforcement Learning Framework for Multi-Agent Environments
Authors: Abhishek Sawaika, Samuel Yen-Chi Chen, Udaya Parampalli, Rajkumar Buyya
Organizations: Quantum Cloud Computing and Distributed Systems Lab (qCLOUDS), School of Computing and Information Systems, The University of Melbourne, Australia · Brookhaven National Laboratory, Upton, NY, USA
Reinforcement learning (RL) is one of the most practical ways to learn from real-life use-cases. Motivated from the cognitive methods used by humans makes it a widely acceptable strategy in the field of artificial intelligence. Most of the environments used for RL are often high-dimensional, and traditional RL algorithms becomes computationally expensive and challenging to effectively learn from such systems. Recent advancements in practical demonstration of quantum computing (QC) theories, such as compact encoding, enhanced representation and learning algorithms, random sampling, or the inherent stochastic nature of quantum systems, have opened up new directions to tackle these challenges. Quantum reinforcement learning (QRL) is seeking significant traction over the past few years. However, the current state of quantum hardware is not enough to cater for such high-dimensional environments with complex multi-agent setup. To tackle this issue, we propose a distributed framework for QRL where multiple agents learn independently, distributing the load of joint training from individual machines. Our method works well for environments with disjoint sets of action and observation spaces, but can also be extended to other systems with reasonable approximations. We analyze the proposed method on cooperative-pong environment and our results indicate ~10% improvement from other distribution strategies, and ~5% improvement from classical models of policy representation.
Figures & tables
Fig. 1 : Architecture of MADQRL framework. Agents learn independently to optimize for joint objective defined by the environment. This independent training happens on local observation, reward and action spaces, such that the joint policy can be approximated by the product of local policies. Here, the MDP representation of the environment is just an example for completeness of the system.
Flatten
Linear_Layer
VQC
RELU
Linear_layer
RELU
Table 2
Fig. 2 : A sample quantum circuit with 4 qubits, having angle encoding using RX gates and L=3 layers of strongly entangled variational layers, followed by full measurement in Z -Basis. Here we use three different entanglement layers.
Fig. 3 : A snapshot of the cooperative-pong game environment. The entire black box is the ”playing arena”, the small white dot at the center is the ”ball”, which moves freely based on the laws of physics, and the two elongated white area on the left and the right edges are the paddles, representing agents. Here, the right one is a cake shaped paddle, while the one on the left side is a standard block paddle.
Model - Distribution
Total Runtime (s)
Test Episodes
Trainable Weights
Sampling Saturation (%iterations)
Classical-Shared
377
193
40074
40
Classical-Joint
291
254
20560
35.8
Classical-Independent
391
264
18752
45
Quantum-Shared
2753
176
20140
37.9
Quantum-Joint SE
1529
259
10889
37.8
Quantum-Joint BE
1795
245
10893
37.5
TABLE I : Performance analysis on two agent cooperative pong game for different combinations of policy model and distribution strategies. A CNN and a QNN based architectures are used for classical and quantum models, respectively. There are 1170 tunable parameters in the VQCs used in quantum models.
Fig. 4 : Comparing learning curves for different distributed learning frameworks between (a) classical, (b) quantum–classical, and (c) quantum models. Results are reported for a 2-agent cooperative Pong game with the observation space reduced to 64×64 . Curves show the moving average of the mean reward with a window size of 5 , while the shaded regions indicate the corresponding standard deviation.
Quantum reinforcement learning studies how quantum circuits and algorithms can improve learning, planning, and control in classical or quantum environments.
Quantum reinforcement learning (QRL) is a promising approach to learn effective decision strategies across several applications with stochastic environments. Instead of directly modeling the random variables that govern these environments, existing QRL architectures indirectly approximate environment behavior by estimating expected outcomes, which limits their expressive power and adaptive potential. Overcoming such challenges requires a novel QRL approach that exploits the distributional nature of quantum computers to directly model environment random variables as quantum state distributions. Hence, in this paper, a novel framework dubbed quantum-native reinforcement learning (QnRL) is proposed. QnRL is a distributional RL framework that learns conditional distributions naturally in Hilbert space via superimposed and entangled quantum states. Thus, QnRL can directly model the behavior of stochastic learning environments via the natural properties of quantum systems. QnRL accomplishes this via a novel, proposed quantum amplitude kickback (QuAK) algorithm that enables comparing the n-th power of the m-th moment of multiple superimposed distributions. It is theoretically proven that a conditional action policy distribution is distilled from the moments of a quantum generative model entirely within Hilbert space via QuAK, and optimized via QnRL. This complex distribution composition is also shown to provide extra dimensions for expressing environment correlations that are unknown to purely classical and classically-sampled quantum distributional models. Experimental results across diverse environments show that QnRL achieves up to 82.9% higher evaluation scores, with up to 94.3% fewer parameters on average, more accurately estimates the expected return for unseen observations, and better adapts to varying stochastic conditions compared to the baseline.
Alexander DeRieux, Walid Saad
Bradley Department of Electrical and Computer Engineering, Virginia Tech Institute for Advanced Computing, Alexandria, VA 22305 USA
We present an empirical evaluation of quantum entanglement in agent coordination within quantum multi agent reinforcement learning (QMARL). While QMARL has attracted growing interest recently, most prior work evaluates quantum policies without provable baselines, making it impossible to rigorously distinguish quantum advantage from algorithmic coincidence. We address this directly by evaluating a decentralized QMARL framework with variational quantum circuit (VQC) actors with shared entangled states. In the CHSH game, which has a mathematically proven classical performance ceiling of 0.75 win rate, we show that entangled QMARL agents approach the Tsirelson limit of 0.854, providing clear evidence of their quantum advantage. We show that unentangled quantum circuits match the classical baseline, confirming that entanglement and not the quantum circuit itself is the active coordination mechanism. We also explore the effect of specific entanglement structures, as some Bell states enable coordination gains while others actively harm performance. On cooperative navigation (CoopNav), QMARL without entanglement achieves ∼2× improvement in success rate over classical MAA2C (∼0.85 versus ∼0.40), with the hybrid configuration, quantum actor paired with a classical centralised critic, outperforming both fully classical and fully quantum solutions. We present our experimental analysis and discuss future work.
Reinforcement learning (RL) provides a principled framework for decision-making in partially observable environments, which can be modeled as Markov decision processes and compactly represented through dynamic decision Bayesian networks. Recent advances demonstrate that inference on sparse Bayesian networks can be accelerated using quantum rejection sampling combined with amplitude amplification, leading to a computational speedup in estimating acceptance probabilities. Building on this result, we introduce Quantum Bayesian Reinforcement Learning (QBRL), a hybrid quantum-classical look-ahead algorithm for model-based RL in partially observable environments. We present a rigorous, oracle-free time complexity analysis under fault-tolerant assumptions for the quantum device. Unlike standard treatments that assume a black-box oracle, we explicitly specify the inference process, allowing our bounds to more accurately reflect the true computational cost. We show that, for environments whose dynamics form a sparse Bayesian network, horizon-based near-optimal planning can be achieved sub-quadratically faster through quantum-enhanced belief updates. On the other hand, we show that there is no quantum speed-up for environments that are either fully observable, or characterized by Bayesian networks whose maximum in-degree is not small. Furthermore, we present numerical experiments benchmarking QBRL against its classical counterpart on simple yet illustrative decision-making tasks. Our results offer a detailed analysis of how the quantum computational advantage translates into decision-making performance, highlighting that the magnitude of the advantage can vary significantly across different deployment settings.
Gilberto Cunha, Alexandra Ramôa, André Sequeira +2
High-Assurance Software Laboratory (HASLab), INESC TEC, Portugal · Department of Computer Science, University of Minho, Portugal · International Iberian Nanotechnology Laboratory (INL), Portugal