Grounding Spatial Relations in a Compact World Model: Instruction Leakage and a Goal-Free Dynamics Fix

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems · Depth: Expert, quick

Summary

Compact world models designed to ground spatial relations using language goals, such as "put the red block left of the blue block," are susceptible to "instruction leakage." Researchers found that a goal-conditioned predictor achieving 0.90 relation-readout accuracy often relies on instruction transcription rather than genuine perception. Evidence includes accuracy collapsing to 0.27 when the goal is withheld, and 94.5% of predictions following false instructions compared to 2.3% for the true scene. This leakage occurs when the scored quantity is directly transcribable from the instruction, independent of other inputs. While tabletop and BabyAI benchmarks exhibit this, a Language-Table forward-dynamics model only leaks if the instruction names the direction. The proposed fix involves removing the goal from the model's dynamics, treating it as a planner's cost, and supervising the "read" path, which restores instruction-independent grounding to 0.88. This detection and remedy protocol is applicable to any goal-conditioned world model where the instruction names the scored quantity.

Key takeaway

For AI Scientists and Machine Learning Engineers developing goal-conditioned world models, you must rigorously test for instruction leakage. If your model's high accuracy on spatial relations like "left of" stems from transcribing the instruction rather than genuine perception, your system lacks robust grounding. You should implement the proposed fix: keep the language goal out of the model's dynamics, treating it solely as a planner's cost, and focus supervision on the "read" path to ensure instruction-independent grounding.

Key insights

Goal-conditioned world models can suffer from "instruction leakage," where high accuracy stems from transcribing instructions, not true perception.

Principles

Method

Detect leakage by testing goal-conditioned models with and without the goal, and with counterfactual instructions. Remedy by removing the goal from dynamics and supervising the "read" path.

In practice

Topics

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.