Back Home

機器人與世界模型

Cosmos-H-Dreams Advances Surgical World Models Toward Real-Time Closed-Loop Operation, Generating About 160 FPS on a Single RTX PRO 6000

NVIDIA has unveiled a generative simulator for surgical robotics built with few-step diffusion, self-forcing distillation, and a streaming KV cache. The system can receive robot motion data in real time and synthesize subsequent video, but visual realism remains no substitute for physical correctness and clinical validation.

Strubbl · CC BY-SA 4.0 · Image source
zh-Hant

On July 27, NVIDIA unveiled Cosmos-H-Dreams, distilling the Cosmos-H-Surgical-Simulator—originally designed for offline rollouts—into an interactive causal world model. Starting from an initial RGB frame and a continuous stream of robot kinematics data, the model autoregressively generates the next segment of surgical video. NVIDIA reports about 160 FPS on a single RTX PRO 6000, compared with roughly 10 FPS for the standard teacher model—fast enough for a human operator or robot policy to form a closed loop within the generated environment.

The training pipeline first maps the displacement, rotation, and gripper states of a dual-arm dVRK into a unified 44-dimensional action representation. The teacher model is then fine-tuned on data covering successful operations as well as dropped needles, missed passes, and knot-tying failures. The training context is gradually extended from 12 to 72 frames. The student model initially imitates pre-cached teacher denoising trajectories while learning causal attention and a streaming KV cache. Subsequent self-forcing distillation allows the student to continue rollouts using its own generated frames, while the frozen teacher provides distribution-matching supervision. This narrows the gap between training—where the model sees only clean histories—and deployment, where errors in its own outputs can repeatedly accumulate. Each latent frame requires as few as two denoising steps.

For inference, FlashDreams combines a streaming KV cache, model compilation, and CUDA Graphs, while WebRTC and WebXR connect browser keyboards or Meta Quest controllers. The code, models, and recipes for fine-tuning a teacher model on proprietary data and performing distillation have all been released, allowing engineering teams to test synthetic-data generation, policy training, and failure-scenario replay.

The key limitation is that 160 FPS measures generation speed only. NVIDIA has not yet demonstrated that tool-tip positions, gripper cycles, tissue responses, long-horizon drift, or simulated policy rankings match real-world systems. The current model is also specialized for desktop suturing with the dVRK. It is an R&D simulator—not a diagnostic tool, an intraoperative imaging system, or a physical robot-control system. The next benchmark should be closed-loop physical consistency and sim-to-real validation, rather than video quality alone.

Sources

  1. NVIDIA Cosmos-H-Dreams: Bringing Real-Time Generative Simulation to Surgical Robotics
  2. Cosmos-Surg-dVRK: World Foundation Model-based Automated Online Evaluation of Surgical Robot Policy Learning