Distilling Morphology-Conditioned Hypernetworks for Efficient Universal Morphology Control
Not on the map. Why?
Abstract (arXiv)
Learning a universal policy across different robot morphologies can significantly improve learning efficiency and enable zero-shot generalization to unseen morphologies. However, learning a highly performant universal policy requires sophisticated architectures like transformers (TF) that have larger memory and computational cost than simpler multi-layer perceptrons (MLP). To achieve both good performance like TF and high efficiency like MLP at inference time, we propose HyperDistill, which consists of: (1) A morphology-conditioned hypernetwork (HN) that generates robot-wise MLP policies, and (2) A policy distillation approach that is essential for successful training. We show that on UNIMAL, a benchmark with hundreds of diverse morphologies, HyperDistill performs as well as a universal TF teacher policy on both training and unseen test robots, but reduces model size by 6-14 times, and computational cost by 67-160 times in different environments. Our analysis attributes the efficiency advantage of HyperDistill at inference time to knowledge decoupling, i.e., the ability to decouple inter-task and intra-task knowledge, a general principle that could also be applied to improve inference efficiency in other domains.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 46 of its 58 references that are in the map. First posted on arXiv 9 Feb 2024.
Builds on
46 of 58 references.
References · 46
- 2017Attention Is All You Needcited by 59,932
- 2023GPT-4 Technical Reportcited by 18,132
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 2,993
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
- 2022RT-1: Robotics Transformer for Real-World Control at Scalecited by 2,007
- 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,401
- 2016HyperNetworkscited by 1,003
- 2023Open X-Embodiment: Robotic Learning Datasets and RT-X Modelscited by 849
- 2015End-to-End Training of Deep Visuomotor Policiescited by 847
- 2021Scaling Vision with Sparse Mixture of Expertscited by 740
- 2018QT-Opt: Scalable Deep Reinforcement Learning for Vision-Based Robotic Manipulationcited by 730
34 more
- 2018Learning Dexterous In-Hand Manipulationcited by 643
- 2017Sim-to-Real Transfer of Robotic Control with Dynamics Randomizationcited by 616
- 2019Measuring and Relieving the Over-smoothing Problem for Graph Neural Networks from the Topological Viewcited by 544
- 2020Learning Quadrupedal Locomotion over Challenging Terraincited by 494
- 2015Policy Distillationcited by 414
- 2020Curriculum Learning for Reinforcement Learning Domains: A Framework and Surveycited by 350
- 2018Quantifying Generalization in Reinforcement Learningcited by 345
- 2018Learning to Adapt in Dynamic, Real-World Environments Through Meta-Reinforcement Learningcited by 270
- 2020Emergent Complexity and Zero-shot Transfer via Unsupervised Environment Designcited by 260
- 2015Contextual Markov Decision Processescited by 213
- 2015Actor-Mimic: Deep Multitask and Transfer Reinforcement Learningcited by 176
- 2021A System for General In-Hand Object Re-Orientationcited by 155
- 2021Embodied Intelligence via Learning and Evolutioncited by 154
- 2020One Policy to Control Them All: Shared Modular Policies for Agent-Agnostic Controlcited by 114
- 2023UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-aware Curriculum and Iterative Generalist-Specialist Learningcited by 101
- 2023Scaling Vision-Language Models with Sparse Mixture of Expertscited by 88
- 2019Distilling Policy Distillationcited by 81
- 2022MetaMorph: Learning Universal Controllers with Transformerscited by 78
- 2023One-for-All: Bridge the Gap Between Heterogeneous Architectures in Knowledge Distillationcited by 77
- 2020Transient Non-Stationarity and Generalisation in Deep Reinforcement Learningcited by 75
- 2019Learning to Control Self-Assembling Morphologies: A Study of Generalization via Modularitycited by 68
- 2020My Body is a Cage: the Role of Morphology in Graph-Based Incompatible Controlcited by 68
- 2022GenLoco: Generalized Locomotion Controllers for Quadrupedal Robotscited by 36
- 2021Recomposing the Reinforcement Learning Building Blocks with Hypernetworkscited by 33
- 2022AnyMorph: Learning Transferable Polices By Inferring Agent Morphologycited by 25
- 2021Bayesian Meta-Learning for Few-Shot Policy Adaptation Across Robotic Platformscited by 22
- 2023Universal Morphology Control via Contextual Modulationcited by 21
- 2023SARA-RT: Scaling up Robotics Transformers with Self-Adaptive Robust Attentioncited by 20
- 2022Hypernetworks for Zero-shot Transfer in Reinforcement Learningcited by 18
- 2022Low-Rank Modular Reinforcement Learning via Muscle Synergycited by 16
- 2023Recurrent Hypernetworks are Surprisingly Strong in Meta-RLcited by 15
- 2022A System for Morphology-Task Generalization via Unified Representation and Behavior Distillationcited by 12
- 2023Subequivariant Graph Reinforcement Learning in 3D Environmentscited by 10
- 2021Linear Representation Meta-Reinforcement Learning for Instant Adaptationcited by 9
Built on by
Citing papers per year: 2025: 4; 2026: 7.
11 papers.
- 2025HyperVLA: Efficient Inference in Vision-Language-Action Models via Hypernetworkscited by 6
- 2025Shared Modular Recurrence in Contextual MDPs for Universal Morphology Controlcited by 3
- 2026Learning Diffusion Policy from Primitive Skills for Robot Manipulationcited by 3
- 2025Knowledge Diversion for Efficient Morphology Control and Policy Transfercited by 1
- 2025Efficient Morphology-Aware Policy Transfer to New Embodimentscited by 0
- 2026TeNet: Text-to-Network for Compact Policy Synthesiscited by 0
- 2026Constraint-based Pre-training: From Structured Constraints to Scalable Model Initializationcited by 0
- 2026The Embodiment Gap in Robot Foundation Modelscited by 0
- 2026Bridging Body and Brain: Gene-Driven Morphology--Control Co-Designcited by 0
- 2026Beam Search as Test-Time Self-Distillation via Counterfactual Contextscited by 0
- 2026The Weakest Link: Distilling LLM Reasoning with Worst-Case Constrained Reinforcement Learningcited by 0
- 2026The Weakest Link: Distilling LLM Reasoning with Worst-Case Constrained Reinforcement Learning29 Sep 2026
- 2026Beam Search as Test-Time Self-Distillation via Counterfactual Contexts29 Sep 2026
- 2026Bridging Body and Brain: Gene-Driven Morphology--Control Co-Design25 Sep 2026
- 2026The Embodiment Gap in Robot Foundation Models19 Aug 2026
- 2026Constraint-based Pre-training: From Structured Constraints to Scalable Model Initialization16 Apr 2026
- 2026TeNet: Text-to-Network for Compact Policy Synthesis22 Jan 2026
- 2026Learning Diffusion Policy from Primitive Skills for Robot Manipulation5 Jan 2026
- 2025Knowledge Diversion for Efficient Morphology Control and Policy Transfer10 Dec 2025
- 2025HyperVLA: Efficient Inference in Vision-Language-Action Models via Hypernetworks6 Oct 2025
- 2025Efficient Morphology-Aware Policy Transfer to New Embodiments5 Aug 2025
- 2025Shared Modular Recurrence in Contextual MDPs for Universal Morphology Control10 Jun 2025