Invariance Through Latent Alignment
Not on the map. Why?
Abstract (arXiv)
A robot's deployment environment often involves perceptual changes that differ from what it has experienced during training. Standard practices such as data augmentation attempt to bridge this gap by augmenting source images in an effort to extend the support of the training distribution to better cover what the agent might experience at test time. In many cases, however, it is impossible to know test-time distribution-shift a priori, making these schemes infeasible. In this paper, we introduce a general approach, called Invariance Through Latent Alignment (ILA), that improves the test-time performance of a visuomotor control policy in deployment environments with unknown perceptual variations. ILA performs unsupervised adaptation at deployment-time by matching the distribution of latent features on the target domain to the agent's prior experience, without relying on paired data. Although simple, we show that this idea leads to surprising improvements on a variety of challenging adaptation scenarios, including changes in lighting conditions, the content in the scene, and camera poses. We present results on calibrated control benchmarks in simulation -- the distractor control suite -- and a physical robot under a sim-to-real setup.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 27 of its 37 references that are in the map. First posted on arXiv 15 Dec 2021.
Builds on
27 of 37 references.
References · 27
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2019Invariant Risk Minimizationcited by 1,832
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,401
- 2015Learning Transferable Features with Deep Adaptation Networkscited by 1,323
- 2018Soft Actor-Critic Algorithms and Applicationscited by 1,179
- 2017CyCADA: Cycle-Consistent Adversarial Domain Adaptationcited by 1,079
- 2021Isaac Gym: High Performance GPU-Based Physics Simulation For Robot Learningcited by 882
- 2018Learning Dexterous In-Hand Manipulationcited by 643
- 2020Learning Quadrupedal Locomotion over Challenging Terraincited by 494
- 2019Leveraging Procedural Generation to Benchmark Reinforcement Learningcited by 478
- 2020Reinforcement Learning with Augmented Datacited by 452
15 more
- 2019Learning agile and dynamic motor skills for legged robotscited by 402
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
- 2016FCNs in the Wild: Pixel-level Adversarial and Constraint-based Adaptationcited by 242
- 2019Active Domain Randomizationcited by 161
- 2020Generalization in Reinforcement Learning by Soft Data Augmentationcited by 132
- 2020Self-Supervised Policy Adaptation during Deploymentcited by 121
- 2019Unsupervised Domain Adaptation through Self-Supervisioncited by 116
- 2020Invariant Causal Prediction for Block MDPscited by 104
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
- 2021Learning High-Speed Flight in the Wildcited by 98
- 2021The Distracting Control Suite -- A Challenging Benchmark for Reinforcement Learning from Pixelscited by 97
- 2019Learning to Manipulate Deformable Objects without Demonstrationscited by 76
- 2020Deep Drone Acrobaticscited by 66
- 2018Deep Drone Racing: Learning Agile Flight in Dynamic Environmentscited by 31
- 2020Tailoring: encoding inductive biases by optimizing unsupervised objectives at prediction timecited by 13
Built on by
Citing papers per year: 2022: 1; 2023: 2; 2024: 2; 2025: 2; 2026: 1.
8 papers.
- 2023What Makes Pre-Trained Visual Representations Successful for Robust Manipulation?cited by 38
- 2023RePo: Resilient Model-Based Reinforcement Learning by Regularizing Posterior Predictabilitycited by 17
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignmentcited by 17
- 2022Learning Representations that Enable Generalization in Assistive Taskscited by 11
- 2024R3L: Relative Representations for Reinforcement Learningcited by 2
- 2025Mapping representations in Reinforcement Learning via Semantic Alignment for Zero-Shot Stitchingcited by 0
- 2025State-Conditional Adversarial Learning: An Off-Policy Visual Domain Transfer Method for End-to-End Imitation Learningcited by 0
- 2026Is Success All You Need? Investigating the Impact of Input Perturbations on VLA Behaviour in Tabletop Manipulation Taskscited by 0
- 2026Is Success All You Need? Investigating the Impact of Input Perturbations on VLA Behaviour in Tabletop Manipulation Tasks1 Oct 2026
- 2025State-Conditional Adversarial Learning: An Off-Policy Visual Domain Transfer Method for End-to-End Imitation Learning5 Dec 2025
- 2025Mapping representations in Reinforcement Learning via Semantic Alignment for Zero-Shot Stitching26 Feb 2025
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignment4 Jun 2024
- 2024R3L: Relative Representations for Reinforcement Learning19 Apr 2024
- 2023What Makes Pre-Trained Visual Representations Successful for Robust Manipulation?3 Nov 2023
- 2023RePo: Resilient Model-Based Reinforcement Learning by Regularizing Posterior Predictability31 Aug 2023
- 2022Learning Representations that Enable Generalization in Assistive Tasks5 Dec 2022