MDP Homomorphic Networks: Group Symmetries in Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
This paper introduces MDP homomorphic networks for deep reinforcement learning. MDP homomorphic networks are neural networks that are equivariant under symmetries in the joint state-action space of an MDP. Current approaches to deep reinforcement learning do not usually exploit knowledge about such structure. By building this prior knowledge into policy and value networks using an equivariance constraint, we can reduce the size of the solution space. We specifically focus on group-structured symmetries (invertible transformations). Additionally, we introduce an easy method for constructing equivariant network layers numerically, so the system designer need not solve the constraints by hand, as is typically done. We construct MDP homomorphic MLPs and CNNs that are equivariant under either a group of reflections or rotations. We show that such networks converge faster than unstructured baselines on CartPole, a grid world and Pong.
Key facts. Cited by 133 papers in the map (data through 7 Oct 2026). First posted on arXiv 30 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 17; 2022: 26; 2023: 30; 2024: 24; 2025: 18; 2026: 18.
133 papers.
Sort:
By area:
- 2021A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groupscited by 185
- 2022Sym-NCO: Leveraging Symmetricity for Neural Combinatorial Optimizationcited by 118
- 2022Equivariance with Learned Canonicalization Functionscited by 106
- 2021Residual Pathway Priors for Soft Equivariance Constraintscited by 74
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 73
- 2022Learning Symmetric Embeddings for Equivariant World Modelscited by 54
- 2021Towards Robust Bisimulation Metric Learningcited by 49
- 2022Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximationscited by 42
- 2025Towards General-Purpose Model-Free Reinforcement Learningcited by 40
- 2021Return-Based Contrastive Representation Learning for Reinforcement Learningcited by 36
- 2022Trustworthy Reinforcement Learning Against Intrinsic Vulnerabilities: Robustness, Safety, and Generalizabilitycited by 32
- 2022The Surprising Effectiveness of Equivariant Models in Domains with Latent Symmetrycited by 32
38 more
- 2021Koopman Q-learning: Offline Reinforcement Learning via Symmetries of Dynamicscited by 27
- 2021A Dataset Perspective on Offline Reinforcement Learningcited by 27
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligencecited by 26
- 2023Learning Probabilistic Symmetrization for Architecture Agnostic Equivariancecited by 25
- 2021Multi-Agent MDP Homomorphic Networkscited by 24
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatchcited by 24
- 2023Learning Layer-wise Equivariances Automatically using Gradientscited by 23
- 2022Toward Compositional Generalization in Object-Oriented World Modelingcited by 22
- 2022Equi-Tuning: Group Equivariant Fine-Tuning of Pretrained Modelscited by 22
- 2022Maximum Class Separation as Inductive Bias in One Matrixcited by 21
- 2022Homomorphism Autoencoder -- Learning Group Structured Representations from Observed Transitionscited by 18
- 2022Unsupervised Representation Learning in Deep Reinforcement Learning: A Reviewcited by 18
- 2023A General Theory of Correct, Incorrect, and Extrinsic Equivariancecited by 18
- 2024Symmetry Considerations for Learning Task Symmetric Robot Policiescited by 18
- 2024One-Shot Imitation Learning with Invariance Matching for Robotic Manipulationcited by 18
- 2024Equivariant Reinforcement Learning under Partial Observabilitycited by 17
- 2022Equivariant Networks for Zero-Shot Coordinationcited by 16
- 2023Structure in Deep Reinforcement Learning: A Survey and Open Problemscited by 16
- 2022Unsupervised Learning of Equivariant Structure from Sequencescited by 15
- 2023Neurosymbolic Grounding for Compositional World Modelscited by 14
- 2023Symmetry-Aware Robot Design with Structured Subgroupscited by 13
- 2023On Robot Grasp Learning Using Equivariant Modelscited by 12
- 2023Efficient Equivariant Transfer Learning from Pretrained Modelscited by 11
- 2023Flow Factorized Representation Learningcited by 11
- 2022Continual Vision-based Reinforcement Learning with Group Symmetriescited by 10
- 2023Subequivariant Graph Reinforcement Learning in 3D Environmentscited by 10
- 2023Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learningcited by 10
- 2025Geometry-aware RL for Manipulation of Varying Shapes and Deformable Objectscited by 10
- 2023Sample Efficient Reinforcement Learning in Mixed Systems through Augmented Samples and Its Applications to Queueing Networkscited by 9
- 2023ESP: Exploiting Symmetry Prior for Multi-Agent Reinforcement Learningcited by 9
- 2024Morphological Symmetries in Roboticscited by 9
- 2024Boosting Sample Efficiency and Generalization in Multi-agent Reinforcement Learning via Equivariancecited by 9
- 2024Approximate Equivariance in Reinforcement Learningcited by 9
- 2021Neural Networks for Learning Counterfactual G-Invariances from Single Environmentscited by 8
- 2022Learning from Symmetry: Meta-Reinforcement Learning with Symmetrical Behaviors and Language Instructionscited by 8
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractionscited by 8
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Surveycited by 8
50 most cited of 133 All 133 (JSON)
- 2026Sample Complexity of Equivariant Reinforcement Learning29 Sep 2026
- 2026Categorical Internalisation of Environmental Groupoids for Generalisable POMDP Solving23 Sep 2026
- 2026Groupoid-Based Internal State Representations for Reinforcement Learning with Local Symmetries11 Sep 2026
- 2026From Reasoning Strings to Partial Orders: Verifier-Certified Rule Transport through Quotient Policy Optimization20 Aug 2026
- 2026Terminal Symmetry as a Carrier of Asymmetric Process Knowledge: Statewise Refinement for Anytime Verified Construction11 Aug 2026
- 2026Learning Not to Optimize: Physics-Informed Action-Space Reshaping for Intent-Based Network Control2 Aug 2026
- 2026Depth-Regularized JEPA World Models Learn More Transferable Representations from Real Outdoor Robot Data15 Jul 2026
- 2026Compositional Behavioral Semantics for State Abstraction in Reinforcement Learning24 Jun 2026
- 2026Adaptive state-action abstractions via rate-distortion4 Jun 2026
- 2026Task-Induced Representational Invariances Depend on Learning Objective in Deep RL1 Jun 2026
- 2026Reflex: Reinforcement Learning with Reflection Symmetry Exploitation in State-Based Continuous Control22 May 2026
- 2026Equivariant Reinforcement Learning for Clifford Quantum Circuit Synthesis11 May 2026
38 more
- 2026Equivariant Multi-agent Reinforcement Learning for Multimodal Vehicle-to-Infrastructure Systems8 Apr 2026
- 2026Tunable Soft Equivariance with Guarantees27 Mar 2026
- 2026PRISM: Parallel Reward Integration with Symmetry for MORL20 Feb 2026
- 2026Geometry of Uncertainty: Learning Metric Spaces for Multimodal State Estimation in RL12 Feb 2026
- 2026Group-Invariant Unsupervised Skill Discovery: Symmetry-aware Skill Representations for Generalizable Behavior20 Jan 2026
- 2026Flow Equivariant World Models: Memory for Partially Observed Dynamic Environments3 Jan 2026
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limits12 Dec 2025
- 2025Partially Equivariant Reinforcement Learning in Symmetry-Breaking Environments30 Nov 2025
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025Robust Remote Reinforcement Learning over Unreliable Communication Channels using Homomorphic State Encoding11 Aug 2025
- 2025Geometry of Neural Reinforcement Learning in Continuous State and Action Spaces28 Jul 2025
- 2025SE(3)-Equivariant Diffusion Policy in Spherical Fourier Space2 Jul 2025
- 2025Learning Abstract World Models with a Group-Structured Latent Space2 Jun 2025
- 2025Surrogate Modeling of 3D Rayleigh-Benard Convection with Equivariant Autoencoders19 May 2025
- 2025Morphologically Symmetric Reinforcement Learning for Ambidextrous Bimanual Manipulation8 May 2025
- 2025Symmetry-Preserving Architecture for Multi-NUMA Environments (SPANE): A Deep Reinforcement Learning Approach for Dynamic VM Scheduling21 Apr 2025
- 2025PEnGUiN: Partially Equivariant Graph NeUral Networks for Sample Efficient MARL19 Mar 2025
- 2025Permutation Equivariant Neural Networks for Symmetric Tensors14 Mar 2025
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Survey12 Mar 2025
- 2025Generalization Bounds for Equivariant Networks on Markov Data1 Mar 2025
- 2025Learning Decentralized Swarms Using Rotation Equivariant Graph Neural Networks24 Feb 2025
- 2025Geometry-aware RL for Manipulation of Varying Shapes and Deformable Objects10 Feb 2025
- 2025Towards General-Purpose Model-Free Reinforcement Learning27 Jan 2025
- 2025Symmetries-enhanced Multi-Agent Reinforcement Learning2 Jan 2025
- 2024Equivariant Action Sampling for Reinforcement Learning and Planning16 Dec 2024
- 2024Approximate Equivariance in Reinforcement Learning6 Nov 2024
- 2024Sample-efficient Bayesian Optimisation Using Known Invariances22 Oct 2024
- 2024Reinforcement Learning with Euclidean Data Augmentation for State-Based Continuous Control16 Oct 2024
- 2024Boosting Sample Efficiency and Generalization in Multi-agent Reinforcement Learning via Equivariance3 Oct 2024
- 2024Leveraging Symmetry to Accelerate Learning of Trajectory Tracking Controllers for Free-Flying Robotic Systems17 Sep 2024
- 2024Equivariant Reinforcement Learning under Partial Observability26 Aug 2024
- 2024Geometric Active Exploration in Markov Decision Processes: the Benefit of Abstraction18 Jul 2024
- 2024Subequivariant Reinforcement Learning in 3D Multi-Entity Physical Environments17 Jul 2024
- 2024The Perils of Optimizing Learned Reward Functions: Low Training Error Does Not Guarantee Low Regret22 Jun 2024
- 2024SiT: Symmetry-Invariant Transformers for Generalisation in Reinforcement Learning21 Jun 2024
- 2024Equivariant Offline Reinforcement Learning20 Jun 2024
- 2024Spatio-temporal Value Semantics-based Abstraction for Dense Deep Reinforcement Learning24 May 2024
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatch24 May 2024
50 newest of 133 All 133 (JSON)
- 2021Koopman Q-learning: Offline Reinforcement Learning via Symmetries of Dynamicscited by 27
- 2021A Dataset Perspective on Offline Reinforcement Learningcited by 27
- 2024One-Shot Imitation Learning with Invariance Matching for Robotic Manipulationcited by 18
- 2023Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learningcited by 10
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Surveycited by 8
- 2024Equivariant Offline Reinforcement Learningcited by 5
- 2023Equivariant Data Augmentation for Generalization in Offline Reinforcement Learningcited by 3
- 2021Data Augmentation through Expert-guided Symmetry Detection to Improve Performance in Offline Reinforcement Learningcited by 1
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Survey12 Mar 2025
- 2024Equivariant Offline Reinforcement Learning20 Jun 2024
- 2024One-Shot Imitation Learning with Invariance Matching for Robotic Manipulation21 May 2024
- 2023Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learning27 Oct 2023
- 2023Equivariant Data Augmentation for Generalization in Offline Reinforcement Learning14 Sep 2023
- 2021Data Augmentation through Expert-guided Symmetry Detection to Improve Performance in Offline Reinforcement Learning18 Dec 2021
- 2021A Dataset Perspective on Offline Reinforcement Learning8 Nov 2021
- 2021Koopman Q-learning: Offline Reinforcement Learning via Symmetries of Dynamics2 Nov 2021
- 2022Learning Symmetric Embeddings for Equivariant World Modelscited by 54
- 2022Toward Compositional Generalization in Object-Oriented World Modelingcited by 22
- 2023Neurosymbolic Grounding for Compositional World Modelscited by 14
- 2026Flow Equivariant World Models: Memory for Partially Observed Dynamic Environmentscited by 6
- 2023Equivariant MuZerocited by 3
- 2026Depth-Regularized JEPA World Models Learn More Transferable Representations from Real Outdoor Robot Datacited by 2
- 2025Learning Abstract World Models with a Group-Structured Latent Spacecited by 1
- 2026Depth-Regularized JEPA World Models Learn More Transferable Representations from Real Outdoor Robot Data15 Jul 2026
- 2026Flow Equivariant World Models: Memory for Partially Observed Dynamic Environments3 Jan 2026
- 2025Learning Abstract World Models with a Group-Structured Latent Space2 Jun 2025
- 2023Neurosymbolic Grounding for Compositional World Models19 Oct 2023
- 2023Equivariant MuZero9 Feb 2023
- 2022Toward Compositional Generalization in Object-Oriented World Modeling28 Apr 2022
- 2022Learning Symmetric Embeddings for Equivariant World Models24 Apr 2022
- 2021A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groupscited by 185
- 2022Sym-NCO: Leveraging Symmetricity for Neural Combinatorial Optimizationcited by 118
- 2022Equivariance with Learned Canonicalization Functionscited by 106
- 2021Residual Pathway Priors for Soft Equivariance Constraintscited by 74
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 73
- 2021Towards Robust Bisimulation Metric Learningcited by 49
- 2022Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximationscited by 42
- 2025Towards General-Purpose Model-Free Reinforcement Learningcited by 40
- 2021Return-Based Contrastive Representation Learning for Reinforcement Learningcited by 36
- 2022Trustworthy Reinforcement Learning Against Intrinsic Vulnerabilities: Robustness, Safety, and Generalizabilitycited by 32
- 2022The Surprising Effectiveness of Equivariant Models in Domains with Latent Symmetrycited by 32
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligencecited by 26
12 most cited of 114 All 114 (JSON)
- 2026Sample Complexity of Equivariant Reinforcement Learning29 Sep 2026
- 2026Categorical Internalisation of Environmental Groupoids for Generalisable POMDP Solving23 Sep 2026
- 2026Groupoid-Based Internal State Representations for Reinforcement Learning with Local Symmetries11 Sep 2026
- 2026From Reasoning Strings to Partial Orders: Verifier-Certified Rule Transport through Quotient Policy Optimization20 Aug 2026
- 2026Terminal Symmetry as a Carrier of Asymmetric Process Knowledge: Statewise Refinement for Anytime Verified Construction11 Aug 2026
- 2026Learning Not to Optimize: Physics-Informed Action-Space Reshaping for Intent-Based Network Control2 Aug 2026
- 2026Compositional Behavioral Semantics for State Abstraction in Reinforcement Learning24 Jun 2026
- 2026Adaptive state-action abstractions via rate-distortion4 Jun 2026
- 2026Task-Induced Representational Invariances Depend on Learning Objective in Deep RL1 Jun 2026
- 2026Reflex: Reinforcement Learning with Reflection Symmetry Exploitation in State-Based Continuous Control22 May 2026
- 2026Equivariant Reinforcement Learning for Clifford Quantum Circuit Synthesis11 May 2026
- 2026Equivariant Multi-agent Reinforcement Learning for Multimodal Vehicle-to-Infrastructure Systems8 Apr 2026
12 newest of 114 All 114 (JSON)