Learning Object-Centric Spatial Reasoning for Sequential Manipulation in Cluttered Environments

TL;DR

This paper introduces Unveiler, a modular framework for robotic manipulation in cluttered environments that combines a transformer-based spatial reasoning encoder with an action decoder, achieving high success rates and transferability.

Contribution

The work presents a novel decoupled architecture with a transformer-based spatial relationship encoder and an action decoder, improving efficiency and performance over end-to-end models.

Findings

Achieves up to 97.6% success in simulation scenarios.

Outperforms classic and large-model baselines in clutter retrieval tasks.

Zero-shot transfer of spatial reasoning to real scenes.

Abstract

Robotic manipulation in cluttered environments presents a critical challenge for automation. Recent large-scale, end-to-end models demonstrate impressive capabilities but often lack the data efficiency and modularity required for retrieving objects in dense clutter. In this work, we argue for a paradigm of specialized, decoupled systems and present Unveiler, a framework that explicitly separates high-level spatial reasoning from low-level action execution. Unveiler's core is a lightweight, transformer-based Spatial Relationship Encoder (SRE) that sequentially identifies the most critical obstacle for removal. This discrete decision is then passed to a rotation-invariant Action Decoder for execution. We demonstrate that this decoupled architecture is not only more computationally efficient in terms of parameter count and inference time, but also significantly outperforms both classic end-to-end policies and modern, large-model-based baselines in retrieving targets from dense clutter. The SRE is trained in two stages: imitation learning from heuristic demonstrations provides sample-efficient initialization, after which PPO fine-tuning enables the policy to discover removal strategies that surpass the heuristic in dense clutter. Our results, achieving up to 97.6\% success in partially occluded and 90.0\% in fully occluded scenarios in simulation, make a case for the power of specialized, object-centric reasoning in complex manipulation tasks. Additionally, we demonstrate that the SRE's spatial reasoning transfers zero-shot to real scenes, and validate the full system on a physical robot requiring only geometric workspace calibration; no learned components are retrained.