Search papers, labs, and topics across Lattice.
This paper introduces a novel approach to robot learning by structurally factoring policies into world and task components, which enhances generalization across diverse constraints and environments. The authors formalize this factorization using Bayesian model evidence, demonstrating that it aligns with the data-generating process and reduces complexity in task parameters. Experimental results show that their framework, which combines AICON with a learned policy, outperforms traditional methods and achieves zero-shot generalization to new configurations, even transferring to real hardware without the need for retraining.
Separating world factors from task factors enables robots to generalize across diverse environments and tasks, achieving zero-shot performance and hardware transfer without retraining.
Robot learning must produce policies that generalize to new combinations of constraints, teammates, and environments. To achieve this, we must structurally factor the policy, which is a choice that dictates what generalizes, what requires retraining, and what remains entangled. Existing methods span a wide spectrum, from expecting structure to emerge from data scaling, to hand-designing it via hierarchies, skill libraries or learned specializations. In this paper, we study what we argue is the most fundamental factorization in robotics: separating the world from the task. We investigate the conditions under which this factorization is principled. World factors are properties of the embodied system and the environment; they exist independently of intent. Task factors are defined by the task's logic over what the world admits. We formalize this asymmetry through Bayesian model evidence: it aligns with the data-generating process, maintains high likelihood through an analytical world model, and reduces the Occam razor's penalty on task parameters. We instantiate this factorization by pairing AICON, a differentiable graph of recursive estimators and interconnections that is compositional, operates without task-specific data, and propagates cost gradients to actuators, with a compact, learned policy that modulates gradient paths. Gradients serve as the interface between the two factors: they carry world structure through the graph and task structure through costs, enabling low-dimensional learning while preserving structural generalization. We test the world/task factorization across three problems that encompass heterogeneous robots, environments, task logic and sensorimotor modalities. Our framework outperforms end-to-end baselines and analytical heuristics in all settings, generalizes zero-shot to out-of-distribution configurations, and transfers to real hardware without retraining.
