Spotlighting Task-Relevant Features: Object-Centric Representations for Better Generalization in Robotic Manipulation

🎉 Accepted at IROS 2026 🤖

1Ecole Centrale de Lyon, CNRS, Universite Claude Bernard Lyon 1, INSA Lyon, Université Lumière Lyon 2, LIRIS, UMR5205, 69130 Ecully, France
Visual representation overview showing Global, Dense, and Object-centric architectures

Object-centric representations, unlike global or dense features, naturally separate task-relevant objects from irrelevant background noise, enabling robust generalization.

Abstract

Training robotic policies that reliably generalize to novel environments remains a persistent challenge. While state-of-the-art models leverage powerful global or dense visual features, these representations struggle to separate critical task-specific signals from background noise, causing failures under visual shifts.

In this work, we investigate Slot-based Object-Centric Representations (SOCRs) [1] as a structural solution that decomposes scenes into discrete, actionable entities without supervision. Through a large-scale diagnostic study across simulated (METAWORLD [4], LIBERO [5]) and real-world tasks, we systematically evaluate SOCRs against dominant baselines.

We find that structured abstraction drives inherent robustness: SOCRs drastically outperform standard models under severe lighting, texture, and clutter shifts without task-specific fine-tuning, and they scale effectively with in-domain video pre-training. However, we also identify a critical vulnerability: a structural capacity trade-off that leads to slot merging under high clutter. By mapping out both the strengths and the bottleneck limits of SOCRs, this work provides a clear roadmap for integrating structured visual abstractions into next-generation, generalizable robotic systems.

Video Overview

Methodology & Visual Focus Analysis

Our method extracts an object-centric representation from dense features using a Slot Attention mechanism [1]. Given an input image, it is first encoded by a DINOv2 visual backbone [2] into dense feature tokens. A Transformer layer handles temporal consistency before iteratively attending to features to extract a compact set of object representations (slots). We integrate this visual module, dubbed DINOSAUR* [3], into a multi-task policy training framework to evaluate manipulation generalization.

Policy Model Architecture

Figure 2: Overview of our robotic manipulation policy architecture. The pre-trained visual model extracts features that are combined with text instructions and proprioception in the observation trunk before passing to the action head.

To understand how these features aggregate information, we conducted a model focus analysis (Figure 1: Grad-CAM for CNNs, Attention-Rollout for ViTs). We found that while standard models tend to have broad or easily distracted focus areas, our slot-based representations split dense features into multiple slots that specialize in different objects. This explicitly separates task-relevant information (like the gripper and target object) from background noise.

The Impact of Robotic Pre-Training

While object-centric models excel at structured scene decomposition, they are primarily trained on in-the-wild datasets (like COCO). To bridge the domain gap for robotic manipulation, we introduced a domain-specific pretraining stage. Our DINOSAUR-Rob* model is pre-trained on a diverse, large-scale mixture of real-world robotic datasets including BridgeV2 [7], Fractal [8], and DROID [9]. We found that this diverse data mixture significantly outperforms models trained on single-source robotic datasets or standard vision baselines, particularly in real-world scenarios.

Pre-training data MetaWorld LIBERO Real-Robot
COCO (Standard Baseline) 0.73 0.75 0.48
BridgeV2 (B) [7] 0.75 0.73 0.45
Fractal (F) [8] 0.72 0.58 0.38
DROID (D) [9] 0.74 0.72 0.26
DINOSAUR-Rob* (D+B+F) 0.76 0.77 0.56

Table 3: Pre-training data evaluation. Final global performance for DINOSAUR*.

Experimental Results

We evaluated DINOSAUR* and DINOSAUR-Rob* against dominant global (VC-1, R3M), dense (DINOv2, Theia, ResNet-50), and segmentation-driven (SAM+DINOv2) baselines. Our evaluation spanned simulated environments (MetaWorld, LIBERO) and real-world Franka deployments.

Overview of the real-world robotic setup using a Franka arm.

Figure 3: Overview of the real-world robotic setup using a Franka arm across four tabletop tasks: Stacking bowls, Screwdriver Drawer, Cans to bin, and Plates to stash.

Generalization Under Visual Distribution Shifts

Object-centric models generalize substantially better than dense and global representations under realistic out-of-distribution (OOD) scenarios.

Diversity of generalization scenes in simulation
Generalization Distractors
Generalization Textures

Figure 4: Overview of generalization scenarios including novel distractors, unseen textures, and lighting changes applied to simulated environments (top) and real-world setups (bottom).

Model Distractors Textures Lighting Overall OOD
ResNet-500.040.000.220.10
DINOv20.110.030.390.18
VC10.060.00.230.10
Theia0.650.280.480.47
DINOSAUR* (Ours)0.210.480.710.46
DINOSAUR-Rob* (Ours)0.460.360.650.49

Table 1: Out-of-Distribution Success Rates in MetaWorld.

Model Distractors Textures Overall OOD
ResNet-500.150.100.12
DINOv20.060.080.07
VC10.030.020.03
Theia0.060.080.07
DINOSAUR* (Ours)0.270.290.28
DINOSAUR-Rob* (Ours)0.370.440.41

Table 2: Out-of-Distribution Success Rates on the Real Robot.

Failure Modes & The Capacity Trade-off

Through qualitative analysis, we identified slot merging as the primary bottleneck for object-centric policies. When objects outnumber the available slots (K), features from novel distractors leak into task-relevant slots, polluting the state representation.

  • Under-segmentation (K < 10): Reducing slot capacity forces harmful merging, dropping OOD success rates dramatically (down to 34% at K=4).
  • Increased Capacity (K > 10): Increasing the slot budget (e.g., K=20) allows the encoder to isolate novel distractors, improving OOD robustness up to 60%. However, this introduces a trade-off: flooding the policy with a noisier, higher-dimensional state space degrades In-Domain performance.
Slot merging analysis under varying distractors

Figure 5: Qualitative analysis showing slot merging failures under hard, medium, and easy distractor levels (K=10).

Slot count ablation chart

Figure 6: The capacity trade-off: Impact of varying slot numbers (K) on In-Domain vs. Out-of-Domain success rates.

Real-World Scene Diversity

To rigorously test our policies under real-world conditions, we evaluated the Franka arm across a highly diverse set of configurations. The gallery below demonstrates the variance in object starting positions.

Screwdriver Task Diversity
Plate Task Diversity
Bowl Task Diversity
Can Task Diversity

Figure 7: Real-world inference scene diversity showcasing task executions under varying spatial configurations and novel distractors.

References

  1. F. Locatello, D. Weissenborn, T. Unterthiner, A. Mahendran, G. Heigold, J. Uszkoreit, A. Dosovitskiy, and T. Kipf. "Object-centric learning with slot attention," 2020.
  2. M. Oquab, T. Darcet, T. Moutakanni, et al., "Dinov2: Learning robust visual features without supervision," 2024.
  3. M. Seitzer, M. Horn, A. Zadaianchuk, et al., "Bridging the gap to real-world object-centric learning," 2023.
  4. T. Yu, D. Quillen, Z. He, R. Julian, A. Narayan, et al., "Meta-world: A benchmark and evaluation for multi-task and meta reinforcement learning," 2021.
  5. B. Liu, Y. Zhu, C. Gao, Y. Feng, Q. Liu, Y. Zhu, and P. Stone, "Libero: Benchmarking knowledge transfer for lifelong robot learning," 2023.
  6. J. Shang, K. Schmeckpeper, B. B. May, M. V. Minniti, T. Kelestemur, D. Watkins, and L. Herlant, "Theia: Distilling diverse vision foundation models for robot learning," 2024.
  7. Homer et al., "BridgeData V2: A Dataset for Robot Learning at Scale," 2023.
  8. Brohan et al., "RT-1: Robotics Transformer for Real-World Control at Scale," 2022.
  9. Khansari et al., "DROID: A Large-Scale In-The-Wild Robot Manipulation Dataset," 2024.

BibTeX

@misc{chapin2026spotlightingtaskrelevantfeaturesobjectcentric,
      title={Spotlighting Task-Relevant Features: Object-Centric Representations for Better Generalization in Robotic Manipulation}, 
      author={Alexandre Chapin and Bruno Machado and Emmanuel Dellandréa and Liming Chen},
      year={2026},
      eprint={2601.21416},
      archivePrefix={arXiv},
      primaryClass={cs.RO},
      url={https://arxiv.org/abs/2601.21416}, 
}