Scale Up Strategically: Learning Compositional Generalization via Bias-Aware Evaluation and Data Collection for Robotic Manipulation

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Robotics & Autonomous Systems, Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

A new diagnostic framework and bias-aware data collection strategy addresses compositional generalization failures in robotic manipulation policies. This framework localizes policy shortcuts to individual "instruction factors" like color, verb, object, size, and spatial attribute, which are reusable semantic components. It formalizes "instruction factor bias"—the tendency of fine-tuned policies to over-rely on dominant factors—and quantifies it using Factor Dominance Rate (FDR) and Factor Dominance Hierarchy (FDH). Evaluation across six foundation policies revealed a consistent ordering: color ≥ object ≥ spatial ≥ verb ≥ size, with color being dominant and verb/size under-grounded. The proposed bias-aware data collection strategy reallocates a fixed budget towards these under-grounded factors, outperforming baselines in simulation and on a real robot using half the demonstrations, enabling more sample-efficient and generalizable policy learning.

Key takeaway

For Robotics Engineers developing policies for diverse instructions, understanding and mitigating instruction factor bias is crucial. You should diagnose your pretrained policies using metrics like Factor Dominance Rate to identify over-relied-upon semantic components. By strategically reallocating your data collection budget towards under-grounded factors, you can achieve more sample-efficient and generalizable policy learning, potentially halving demonstration requirements and improving real-world robot performance.

Key insights

Instruction factor bias, where policies over-rely on dominant semantic components, hinders compositional generalization in robotic manipulation.

Principles

Method

A diagnostic framework quantifies instruction factor bias using Factor Dominance Rate (FDR) and Factor Dominance Hierarchy (FDH) to identify under-grounded factors, guiding bias-aware data collection.

In practice

Topics

Best for: Research Scientist, AI Scientist, Robotics Engineer, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.