SO(2)-Equivariant Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Equivariant neural networks enforce symmetry within the structure of their convolutional layers, resulting in a substantial improvement in sample efficiency when learning an equivariant or invariant function. Such models are applicable to robotic manipulation learning which can often be formulated as a rotationally symmetric problem. This paper studies equivariant model architectures in the context of Q-learning and actor-critic reinforcement learning. We identify equivariant and invariant characteristics of the optimal Q-function and the optimal policy and propose equivariant DQN and SAC algorithms that leverage this structure. We present experiments that demonstrate that our equivariant versions of DQN and SAC can be significantly more sample efficient than competing algorithms on an important class of robotic manipulation problems.
Key facts. Cited by 79 papers in the map (data through 7 Oct 2026). First posted on arXiv 8 Mar 2022.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2022: 10; 2023: 18; 2024: 18; 2025: 22; 2026: 11.
79 papers.
Sort:
By area:
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learningcited by 72
- 2024Equivariant Diffusion Policycited by 61
- 2022Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learningcited by 54
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 51
- 2024Does equivariance matter at scale?cited by 49
- 2023EDGI: Equivariant Diffusion for Planning with Embodied Agentscited by 35
- 2022The Surprising Effectiveness of Equivariant Models in Domains with Latent Symmetrycited by 32
- 2024Fourier Transporter: Bi-Equivariant Robotic Manipulation in 3Dcited by 32
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2022SEIL: Simulation-augmented Equivariant Imitation Learningcited by 22
- 2023Learning to Extrapolate: A Transductive Approachcited by 19
- 2023A General Theory of Correct, Incorrect, and Extrinsic Equivariancecited by 18
38 more
- 2024Symmetry Considerations for Learning Task Symmetric Robot Policiescited by 18
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policiescited by 18
- 2024Equivariant Reinforcement Learning under Partial Observabilitycited by 17
- 2022Equivariant Networks for Zero-Shot Coordinationcited by 16
- 2023One-shot Imitation Learning via Interaction Warpingcited by 16
- 2022Unsupervised Learning of Equivariant Structure from Sequencescited by 15
- 2022Relational Reasoning via Set Transformers: Provable Efficiency and Applications to MARLcited by 12
- 2023On Robot Grasp Learning Using Equivariant Modelscited by 12
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentationcited by 12
- 2022Continual Vision-based Reinforcement Learning with Group Symmetriescited 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
- 2024Approximate Equivariance in Reinforcement Learningcited by 9
- 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
- 2024Symmetry-aware Reinforcement Learning for Robotic Assembly under Partial Observability with a Soft Wristcited by 7
- 2025SE(3)-Equivariant Diffusion Policy in Spherical Fourier Spacecited by 7
- 2023Symmetric Replay Training: Enhancing Sample Efficiency in Deep Reinforcement Learning for Combinatorial Optimizationcited by 6
- 2023E(2)-Equivariant Graph Planning for Navigationcited by 6
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policiescited by 6
- 2025Constraint-Preserving Data Generation for Visuomotor Policy Learningcited by 6
- 2025Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetrycited by 6
- 2023SO(2)-Equivariant Downwash Models for Close Proximity Flightcited by 5
- 2024Advanced deep-reinforcement-learning methods for flow control: group-invariant and positional-encoding networks improve learning speed and qualitycited by 5
- 2025Latent Mixture of Symmetries for Sample-Efficient Dynamic Learningcited by 5
- 2022Equivariant Reinforcement Learning for Quadrotor UAVcited by 4
- 2023rm E(3)-Equivariant Actor-Critic Methods for Cooperative Multi-Agent Reinforcement Learningcited by 4
- 2023Learning Regularized Graphon Mean-Field Games with Unknown Graphonscited by 4
- 2024Synthesizing Neural Network Controllers with Closed-Loop Dissipativity Guaranteescited by 4
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policycited by 4
- 20253D Equivariant Visuomotor Policy Learning via Spherical Projectioncited by 4
- 2025EfficientFlow: Efficient Equivariant Flow Policy Learning for Embodied AIcited by 4
- 2023Equivariant MuZerocited by 3
- 2023Equivariant Data Augmentation for Generalization in Offline Reinforcement Learningcited by 3
- 2024Equivariant Ensembles and Regularization for Reinforcement Learning in Map-based Path Planningcited by 3
- 2024SiT: Symmetry-Invariant Transformers for Generalisation in Reinforcement Learningcited by 3
- 2025Symmetry-Guided Memory Augmentation for Efficient Locomotion Learningcited by 3
50 most cited of 79 All 79 (JSON)
- 2026Sample Complexity of Equivariant Reinforcement Learning29 Sep 2026
- 2026SymmGrid: Super-Scaling On-Robot Learning with Parallelized Symmetries and Egocentric-Exocentric Visual Perception29 Jul 2026
- 2026Pix2Act: Image-Space Manipulation Policies with Equivariant Augmentation13 Jul 2026
- 2026Conformal Orbit-Valid Trust Horizons for Equivariant World Models23 Jun 2026
- 2026EquiVLA: A General Framework for Rotationally Equivariant Vision-Language-Action Models18 Jun 2026
- 2026Reflex: Reinforcement Learning with Reflection Symmetry Exploitation in State-Based Continuous Control22 May 2026
- 2026Efficient Hybrid SE(3)-Equivariant Visuomotor Flow Policy via Spherical Harmonics for Robot Manipulation24 Mar 2026
- 2026Generalized Reduction to the Isotropy for Flexible Equivariant Neural Fields8 Mar 2026
- 2026Cost-Aware Diffusion Active Search23 Feb 2026
- 2026PRISM: Parallel Reward Integration with Symmetry for MORL20 Feb 2026
- 2026Group-Invariant Unsupervised Skill Discovery: Symmetry-aware Skill Representations for Generalizable Behavior20 Jan 2026
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limits12 Dec 2025
38 more
- 2025Drawback of Enforcing Equivariance and its Compensation via the Lens of Expressive Power10 Dec 2025
- 2025Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetry5 Dec 2025
- 2025EfficientFlow: Efficient Equivariant Flow Policy Learning for Embodied AI1 Dec 2025
- 2025Partially Equivariant Reinforcement Learning in Symmetry-Breaking Environments30 Nov 2025
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025Latent Mixture of Symmetries for Sample-Efficient Dynamic Learning4 Oct 2025
- 2025Multi-Group Equivariant Augmentation for Reinforcement Learning in Robot Manipulation15 Aug 2025
- 2025Constraint-Preserving Data Generation for Visuomotor Policy Learning5 Aug 2025
- 2025SE(3)-Equivariant Diffusion Policy in Spherical Fourier Space2 Jul 2025
- 20253D Equivariant Visuomotor Policy Learning via Spherical Projection22 May 2025
- 2025Equivariant Eikonal Neural Networks: Grid-Free, Scalable Travel-Time Prediction on Homogeneous Spaces21 May 2025
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policy19 May 2025
- 2025Morphologically Symmetric Reinforcement Learning for Ambidextrous Bimanual Manipulation8 May 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
- 2025Your Language Model May Think Too Rigidly: Achieving Reasoning Consistency with Symmetry-Enhanced Training25 Feb 2025
- 2025Geometry-aware RL for Manipulation of Varying Shapes and Deformable Objects10 Feb 2025
- 2025Coarse-to-Fine 3D Keyframe Transporter3 Feb 2025
- 2025Symmetry-Guided Memory Augmentation for Efficient Locomotion Learning3 Feb 2025
- 2025MatrixNet: Learning over symmetry groups using learned group representations16 Jan 2025
- 2024Approximate Equivariance in Reinforcement Learning6 Nov 2024
- 2024Problem Space Transformations for Out-of-Distribution Generalisation in Behavioural Cloning6 Nov 2024
- 2024Does equivariance matter at scale?30 Oct 2024
- 2024Reinforcement Learning with Euclidean Data Augmentation for State-Based Continuous Control16 Oct 2024
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policies23 Sep 2024
- 2024Equivariant Reinforcement Learning under Partial Observability26 Aug 2024
- 2024Advanced deep-reinforcement-learning methods for flow control: group-invariant and positional-encoding networks improve learning speed and quality25 Jul 2024
- 2024Equivariant Diffusion Policy1 Jul 2024
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learning1 Jul 2024
- 2024SiT: Symmetry-Invariant Transformers for Generalisation in Reinforcement Learning21 Jun 2024
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policies17 Jun 2024
- 2024Synthesizing Neural Network Controllers with Closed-Loop Dissipativity Guarantees10 Apr 2024
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentation28 Mar 2024
- 2024Exploiting Symmetry in Dynamics for Model-Based Reinforcement Learning with Asymmetric Rewards27 Mar 2024
- 2024Equivariant Ensembles and Regularization for Reinforcement Learning in Map-based Path Planning19 Mar 2024
- 2024Symmetry Considerations for Learning Task Symmetric Robot Policies7 Mar 2024
- 2024Symmetry-aware Reinforcement Learning for Robotic Assembly under Partial Observability with a Soft Wrist28 Feb 2024
50 newest of 79 All 79 (JSON)
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learningcited by 72
- 2024Equivariant Diffusion Policycited by 61
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 51
- 2023EDGI: Equivariant Diffusion for Planning with Embodied Agentscited by 35
- 2025SE(3)-Equivariant Diffusion Policy in Spherical Fourier Spacecited by 7
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policycited by 4
- 20253D Equivariant Visuomotor Policy Learning via Spherical Projectioncited by 4
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limitscited by 1
- 2026Efficient Hybrid SE(3)-Equivariant Visuomotor Flow Policy via Spherical Harmonics for Robot Manipulationcited by 1
- 2026Efficient Hybrid SE(3)-Equivariant Visuomotor Flow Policy via Spherical Harmonics for Robot Manipulation24 Mar 2026
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limits12 Dec 2025
- 2025SE(3)-Equivariant Diffusion Policy in Spherical Fourier Space2 Jul 2025
- 20253D Equivariant Visuomotor Policy Learning via Spherical Projection22 May 2025
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policy19 May 2025
- 2024Equivariant Diffusion Policy1 Jul 2024
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learning1 Jul 2024
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation6 Sep 2023
- 2023EDGI: Equivariant Diffusion for Planning with Embodied Agents22 Mar 2023
- 2022SEIL: Simulation-augmented Equivariant Imitation Learningcited by 22
- 2023One-shot Imitation Learning via Interaction Warpingcited by 16
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentationcited by 12
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Surveycited by 8
- 2023Equivariant Data Augmentation for Generalization in Offline Reinforcement Learningcited by 3
- 2025Coarse-to-Fine 3D Keyframe Transportercited by 2
- 2024Problem Space Transformations for Out-of-Distribution Generalisation in Behavioural Cloningcited by 0
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Survey12 Mar 2025
- 2025Coarse-to-Fine 3D Keyframe Transporter3 Feb 2025
- 2024Problem Space Transformations for Out-of-Distribution Generalisation in Behavioural Cloning6 Nov 2024
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentation28 Mar 2024
- 2023Equivariant Data Augmentation for Generalization in Offline Reinforcement Learning14 Sep 2023
- 2023One-shot Imitation Learning via Interaction Warping21 Jun 2023
- 2022SEIL: Simulation-augmented Equivariant Imitation Learning31 Oct 2022
- 2023Equivariant MuZerocited by 3
- 2026Conformal Orbit-Valid Trust Horizons for Equivariant World Modelscited by 1
- 2026Conformal Orbit-Valid Trust Horizons for Equivariant World Models23 Jun 2026
- 2023Equivariant MuZero9 Feb 2023
- 2022Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learningcited by 54
- 2024Does equivariance matter at scale?cited by 49
- 2022The Surprising Effectiveness of Equivariant Models in Domains with Latent Symmetrycited by 32
- 2024Fourier Transporter: Bi-Equivariant Robotic Manipulation in 3Dcited by 32
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2023Learning to Extrapolate: A Transductive Approachcited by 19
- 2023A General Theory of Correct, Incorrect, and Extrinsic Equivariancecited by 18
- 2024Symmetry Considerations for Learning Task Symmetric Robot Policiescited by 18
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policiescited by 18
- 2024Equivariant Reinforcement Learning under Partial Observabilitycited by 17
- 2022Equivariant Networks for Zero-Shot Coordinationcited by 16
- 2022Unsupervised Learning of Equivariant Structure from Sequencescited by 15
12 most cited of 58 All 58 (JSON)
- 2026Sample Complexity of Equivariant Reinforcement Learning29 Sep 2026
- 2026SymmGrid: Super-Scaling On-Robot Learning with Parallelized Symmetries and Egocentric-Exocentric Visual Perception29 Jul 2026
- 2026Pix2Act: Image-Space Manipulation Policies with Equivariant Augmentation13 Jul 2026
- 2026Reflex: Reinforcement Learning with Reflection Symmetry Exploitation in State-Based Continuous Control22 May 2026
- 2026Generalized Reduction to the Isotropy for Flexible Equivariant Neural Fields8 Mar 2026
- 2026Cost-Aware Diffusion Active Search23 Feb 2026
- 2026PRISM: Parallel Reward Integration with Symmetry for MORL20 Feb 2026
- 2026Group-Invariant Unsupervised Skill Discovery: Symmetry-aware Skill Representations for Generalizable Behavior20 Jan 2026
- 2025Drawback of Enforcing Equivariance and its Compensation via the Lens of Expressive Power10 Dec 2025
- 2025Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetry5 Dec 2025
- 2025EfficientFlow: Efficient Equivariant Flow Policy Learning for Embodied AI1 Dec 2025
- 2025Partially Equivariant Reinforcement Learning in Symmetry-Breaking Environments30 Nov 2025
12 newest of 58 All 58 (JSON)