Embodied-R: Collaborative Framework for Activating Embodied Spatial Reasoning in Foundation Models via Reinforcement Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhao, Baining, Wang, Ziyou, Fang, Jianjie, Gao, Chen, Man, Fanhang, Cui, Jinqiang, Wang, Xin, Chen, Xinlei, Li, Yong, Zhu, Wenwu
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916693741142016
author Zhao, Baining
Wang, Ziyou
Fang, Jianjie
Gao, Chen
Man, Fanhang
Cui, Jinqiang
Wang, Xin
Chen, Xinlei
Li, Yong
Zhu, Wenwu
author_facet Zhao, Baining
Wang, Ziyou
Fang, Jianjie
Gao, Chen
Man, Fanhang
Cui, Jinqiang
Wang, Xin
Chen, Xinlei
Li, Yong
Zhu, Wenwu
contents Humans can perceive and reason about spatial relationships from sequential visual observations, such as egocentric video streams. However, how pretrained models acquire such abilities, especially high-level reasoning, remains unclear. This paper introduces Embodied-R, a collaborative framework combining large-scale Vision-Language Models (VLMs) for perception and small-scale Language Models (LMs) for reasoning. Using Reinforcement Learning (RL) with a novel reward system considering think-answer logical consistency, the model achieves slow-thinking capabilities with limited computational resources. After training on only 5k embodied video samples, Embodied-R with a 3B LM matches state-of-the-art multimodal reasoning models (OpenAI-o1, Gemini-2.5-pro) on both in-distribution and out-of-distribution embodied spatial reasoning tasks. Embodied-R also exhibits emergent thinking patterns such as systematic analysis and contextual integration. We further explore research questions including response length, training on VLM, strategies for reward design, and differences in model generalization after SFT (Supervised Fine-Tuning) and RL training.
format Preprint
id arxiv_https___arxiv_org_abs_2504_12680
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Embodied-R: Collaborative Framework for Activating Embodied Spatial Reasoning in Foundation Models via Reinforcement Learning
Zhao, Baining
Wang, Ziyou
Fang, Jianjie
Gao, Chen
Man, Fanhang
Cui, Jinqiang
Wang, Xin
Chen, Xinlei
Li, Yong
Zhu, Wenwu
Artificial Intelligence
Computer Vision and Pattern Recognition
Humans can perceive and reason about spatial relationships from sequential visual observations, such as egocentric video streams. However, how pretrained models acquire such abilities, especially high-level reasoning, remains unclear. This paper introduces Embodied-R, a collaborative framework combining large-scale Vision-Language Models (VLMs) for perception and small-scale Language Models (LMs) for reasoning. Using Reinforcement Learning (RL) with a novel reward system considering think-answer logical consistency, the model achieves slow-thinking capabilities with limited computational resources. After training on only 5k embodied video samples, Embodied-R with a 3B LM matches state-of-the-art multimodal reasoning models (OpenAI-o1, Gemini-2.5-pro) on both in-distribution and out-of-distribution embodied spatial reasoning tasks. Embodied-R also exhibits emergent thinking patterns such as systematic analysis and contextual integration. We further explore research questions including response length, training on VLM, strategies for reward design, and differences in model generalization after SFT (Supervised Fine-Tuning) and RL training.
title Embodied-R: Collaborative Framework for Activating Embodied Spatial Reasoning in Foundation Models via Reinforcement Learning
topic Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2504.12680