Cosmos-Reason1: From Physical Common Sense To Embodied Reasoning

  • 2025-05-19 18:59:19
  • NVIDIA, :, Alisson Azzolini, Junjie Bai, Hannah Brandon, Jiaxin Cao, Prithvijit Chattopadhyay, Huayu Chen, Jinju Chu, Yin Cui, Jenna Diamond, Yifan Ding, Liang Feng, Francesco Ferroni, Rama Govindaraju, Jinwei Gu, Siddharth Gururani, Imad El Hanafi, Zekun Hao, Jacob Huffman, Jingyi Jin, Brendan Johnson, Rizwan Khan, George Kurian, Elena Lantz, Nayeon Lee, Zhaoshuo Li, Xuan Li, Maosheng Liao, Tsung-Yi Lin, Yen-Chen Lin, Ming-Yu Liu, Xiangyu Lu, Alice Luo, Andrew Mathau, Yun Ni, Lindsey Pavao, Wei Ping, David W. Romero, Misha Smelyanskiy, Shuran Song, Lyne Tchapmi, Andrew Z. Wang, Boxin Wang, Haoxiang Wang, Fangyin Wei, Jiashu Xu, Yao Xu, Dinghao Yang, Xiaodong Yang, Zhuolin Yang, Jingxu Zhang, Xiaohui Zeng, Zhe Zhang
  • 0

Abstract

Physical AI systems need to perceive, understand, and perform complex actionsin the physical world. In this paper, we present the Cosmos-Reason1 models thatcan understand the physical world and generate appropriate embodied decisions(e.g., next step action) in natural language through long chain-of-thoughtreasoning processes. We begin by defining key capabilities for Physical AIreasoning, with a focus on physical common sense and embodied reasoning. Torepresent physical common sense, we use a hierarchical ontology that capturesfundamental knowledge about space, time, and physics. For embodied reasoning,we rely on a two-dimensional ontology that generalizes across differentphysical embodiments. Building on these capabilities, we develop two multimodallarge language models, Cosmos-Reason1-7B and Cosmos-Reason1-56B. We curate dataand train our models in two stages: Physical AI supervised fine-tuning (SFT)and Physical AI reinforcement learning (RL). To evaluate our models, we buildcomprehensive benchmarks for physical common sense and embodied reasoningaccording to our ontologies. Evaluation results show that Physical AI SFT andRL bring significant improvements. To facilitate the development of PhysicalAI, we make our code and pre-trained models available under the NVIDIA OpenModel License at https://github.com/nvidia-cosmos/cosmos-reason1.

 

Quick Read (beta)

loading the full paper ...