Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ASPECT:Analogical Semantic Policy Execution via Language Conditioned Transfer

About

Reinforcement Learning (RL) agents often struggle to generalize knowledge to new tasks, even those structurally similar to ones they have mastered. Although recent approaches have attempted to mitigate this issue via zero-shot transfer, they are often constrained by predefined, discrete class systems, limiting their adaptability to novel or compositional task variations. We propose a significantly more generalized approach, replacing discrete latent variables with natural language conditioning via a text-conditioned Variational Autoencoder (VAE). Our core innovation utilizes a Large Language Model (LLM) as a dynamic \textit{semantic operator} at test time. Rather than relying on rigid rules, our agent queries the LLM to semantically remap the description of the current observation to align with the source task. This source-aligned caption conditions the VAE to generate an imagined state compatible with the agent's original training, enabling direct policy reuse. By harnessing the flexible reasoning capabilities of LLMs, our approach achieves zero-shot transfer across a broad spectrum of complex and truly novel analogous tasks, moving beyond the limitations of fixed category mappings. Code and videos are available \href{https://anonymous.4open.science/r/ASPECT-85C3/}{here}.

Ajsal Shereef Palattuparambil, Thommen George Karimpanal, Santu Rana• 2026

Related benchmarks

TaskDatasetResultRank
Combined GeneralizationMiniWorld (Case 2)
TGT Score (Case 2)8.8
6
Combined GeneralizationMiniGrid (Case 2)
TGT Score8.8
6
Generalization to Unseen ObjectsMiniWorld Case 1: Unseen Objects v1 (test)
Target Score8.4
6
Generalization to Unseen ObjectsMiniGrid Case 1: Unseen Objects v1 (test)
Target Generalization Score9.4
6
Language Conditioned TransferMiniWorld Reverse Task Case 3 (target)
Target Success Count (Case 3)84
6
Language Conditioned TransferMiniGrid Reverse Task Case 3 (target)
Target Success Count (Case 3)90
6
Generalization to Unseen ObjectsFragile Object Manipulation (MANIP.) v1 (test)
Lift Score9.6
4
Showing 7 of 7 rows

Other info

Follow for update