Geometry-conditioned action decoding for transparent objects and insertion tasks
Robot policy teams working on fine manipulation should add a geometry path to the action decoder and test it on tasks where semantic object recognition is already good but control fails: transparent bottles, ring-shaped objects, tight insertions, stable release, and contact-heavy alignment.
GeoAlign gives a concrete implementation pattern. It post-trains a Depth Anything V2-Small branch on robot RGB-D data, discards the depth head at rollout, and uses RGB-derived geometry features. The robot state creates query slots that select local geometry from the feature grid, then compact geometry tokens condition the flow-matching action decoder. This keeps rollout inputs to RGB, language, and proprioception while still giving the action head local shape cues.
The cheap validation is a task slice with geometry-sensitive objects. GeoAlign reports 78.8% average success on eight real ALOHA tasks, compared with 65.0% for a matched RGB-only baseline, with transparent-bottle success rising from 35.0% to 75.0% and tape-roll insertion from 40.0% to 65.0%. GeoSem-WAM points in the same direction for world action models: future geometry and semantic supervision are used during training, dense heads are removed at deployment, and real Franka success rises from 88.9% for Fast-WAM to 95.4%.