Knowledge Diversion for Efficient Morphology Control and Policy Transfer

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Feng, Fu, Shi, Ruixiao, Xie, Yucheng, Shen, Jianlu, Wang, Jing, Geng, Xin
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909955084255232
author Feng, Fu
Shi, Ruixiao
Xie, Yucheng
Shen, Jianlu
Wang, Jing
Geng, Xin
author_facet Feng, Fu
Shi, Ruixiao
Xie, Yucheng
Shen, Jianlu
Wang, Jing
Geng, Xin
contents Universal morphology control aims to learn a universal policy that generalizes across heterogeneous agent morphologies, with Transformer-based controllers emerging as a popular choice. However, such architectures incur substantial computational costs, resulting in high deployment overhead, and existing methods exhibit limited cross-task generalization, necessitating training from scratch for each new task. To this end, we propose \textbf{DivMorph}, a modular training paradigm that leverages knowledge diversion to learn decomposable controllers. DivMorph factorizes randomly initialized Transformer weights into factor units via SVD prior to training and employs dynamic soft gating to modulate these units based on task and morphology embeddings, separating them into shared \textit{learngenes} and morphology- and task-specific \textit{tailors}, thereby achieving knowledge disentanglement. By selectively activating relevant components, DivMorph enables scalable and efficient policy deployment while supporting effective policy transfer to novel tasks. Extensive experiments demonstrate that DivMorph achieves state-of-the-art performance, achieving a 3$\times$ improvement in sample efficiency over direct finetuning for cross-task transfer and a 17$\times$ reduction in model size for single-agent deployment.
format Preprint
id arxiv_https___arxiv_org_abs_2512_09796
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Knowledge Diversion for Efficient Morphology Control and Policy Transfer
Feng, Fu
Shi, Ruixiao
Xie, Yucheng
Shen, Jianlu
Wang, Jing
Geng, Xin
Machine Learning
Universal morphology control aims to learn a universal policy that generalizes across heterogeneous agent morphologies, with Transformer-based controllers emerging as a popular choice. However, such architectures incur substantial computational costs, resulting in high deployment overhead, and existing methods exhibit limited cross-task generalization, necessitating training from scratch for each new task. To this end, we propose \textbf{DivMorph}, a modular training paradigm that leverages knowledge diversion to learn decomposable controllers. DivMorph factorizes randomly initialized Transformer weights into factor units via SVD prior to training and employs dynamic soft gating to modulate these units based on task and morphology embeddings, separating them into shared \textit{learngenes} and morphology- and task-specific \textit{tailors}, thereby achieving knowledge disentanglement. By selectively activating relevant components, DivMorph enables scalable and efficient policy deployment while supporting effective policy transfer to novel tasks. Extensive experiments demonstrate that DivMorph achieves state-of-the-art performance, achieving a 3$\times$ improvement in sample efficiency over direct finetuning for cross-task transfer and a 17$\times$ reduction in model size for single-agent deployment.
title Knowledge Diversion for Efficient Morphology Control and Policy Transfer
topic Machine Learning
url https://arxiv.org/abs/2512.09796