cs.ROJul 12, 2026

Artificial Foveated Perception for Mitigating Shortcut Learning in Robotic Foundation Models

Authors: Xiatao SunYuan ZhuangMateo Sanchez Lopez NegreteMatei-Victor ColdeaChen LiangHaoyang ZhangChe LiuZiyao Zeng+4 more

Organizations: 1Yale University · University of Connecticut · 3Peking University · 5Digients · 4Imperial College London

Abstract

Robotic foundation models still need task-specific fine-tuning before deployment, and the fine-tuned policies often break under modest changes in scene layout, lighting, or nearby distractors. We trace this brittleness to \textit{shortcut learning}: fine-tuning supervises actions but not the visual evidence the policy uses, so the policy can settle on scene-level correlations that predict the demonstrations without causing success. We propose Artificial Foveated Perception (AFP), a lightweight, policy-agnostic module that takes the same vision and language inputs as existing Vision-Language-Action and World Action Model pipelines and predicts task-conditioned masks over the relevant objects, the robot, and other action-critical regions. During fine-tuning the masks serve as an auxiliary grounding signal that aligns the policy's visual attention with task-relevant regions; the policy architecture is unchanged, and at inference the policy runs on the original observation stream with no AFP call in the control loop. In simulation with four robotic foundation models and on a real robot with π0.5π_{0.5}, AFP improves generalization under environmental perturbations, reduces overfitting, and shortens fine-tuning. Ablations over mask quality and grounding-loss design show that these gains come from directing policy learning toward task-relevant visual evidence. Code, data, and videos are available at https://apollo-lab-yale.github.io/26-CoRL-AFP-website/.

Explore similar work

CardsList
  1. LIBERO-VPro: Benchmarking Closed-Loop Visual Robustness of Robotic Foundation Models

    Sep 21, 2026Huiqiong Li, Zhiting Mei, Anirudha Majumdar +3