Knowledge Diversion for Efficient Morphology Control and Policy Transfer

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Feng, Fu, Shi, Ruixiao, Xie, Yucheng, Shen, Jianlu, Wang, Jing, Geng, Xin
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_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