EgoSim: Egocentric Exploration in Virtual Worlds with Multi-modal Conditioning

Wei Yu, Songheng Yin, Steve Easterbrook, Animesh Garg

ICLR 2025 · OpenReview

Versions

TL;DR

EgoSim is a novel framework that transforms video diffusion models into controllable "world simulators" for egocentric exploration. By introducing geometric constraints via a Spacetime Epipolar Attention Layer (SEAL) and a smart gating mechanism called the CI2V-adapter, the model can precisely follow camera trajectories and text prompts without the "hallucinations" or control drift common in previous methods.


The Quest for Playable World Models

The ultimate goal of "world models" in AI is to create a playable simulation—an environment where an agent can move, interact, and plan. While current video diffusion models (like Sora or SVD) generate stunning visuals, they lack the fine-grained control needed for true simulation.

Extracted figure from page 2 of EgoSim: Egocentric Exploration in Virtual Worlds with Multi-modal Conditioning

The authors identify two primary bottlenecks preventing current models from acting as reliable simulators:

  1. Control Interference: When guiding a model with camera poses, the model's internal "priors" often override the instructions. For example, if a model hasn't seen a camera move forward in a specific kitchen layout during training, it might refuse to move or glitch, even if the input command is valid.
  2. Textual Intrusion: When combining a starting image (context frame) with a text prompt, the text often "leaks" into parts of the scene that have already been observed. This causes realistic rooms to suddenly morph into "fantasy" versions of themselves as the video progresses.

Architectural Innovations: SEAL and CI2V-adapter

EgoSim introduces two key architectural innovations to solve these interference problems:

1. Spacetime Epipolar Attention Layer (SEAL)

Instead of just feeding camera coordinates into the model as a flat vector, SEAL uses Epipolar Geometry. In 3D vision, if you know how the camera moved between two frames, any point in Frame A must appear along a specific line (the epipolar line) in Frame B.

How it works: SEAL restricts the model's attention mechanism so that patches in one frame only "look" at patches in other frames that lie along these geometric lines. This enforces a rigid 3D constraint, ensuring the generated video strictly follows the requested camera motion rather than relying solely on learned priors.

2. CI2V-adapter

This module acts as a "traffic cop" for multi-modal inputs. It decides, on a patch-by-patch basis, whether to prioritize the visual context (what we've already seen) or the text prompt (what we need to imagine).

Extracted figure from page 2 of EgoSim: Egocentric Exploration in Virtual Worlds with Multi-modal Conditioning

The Gating Mechanism: It uses a sigmoid gating function influenced by camera motion. If the camera is moving into an area already visible in the first frame, it prioritizes visual embeddings. If the camera moves into "uncharted territory," it dials up the text embeddings to guide the imagination of new structures. This mutual exclusivity prevents the "fantasy" transitions that plague other models.

Datasets and Evaluation

The researchers evaluated EgoSim on two main benchmarks:

  • RealEstate: A standard dataset of indoor house tours, primarily featuring static environments.
  • Epic-Field: A significantly more challenging dataset of egocentric (first-person) videos involving dynamic human actions (e.g., cooking). The authors repurposed this dataset by re-annotating it with high-quality text labels using CogVLM2 and applying NAFNET for motion deblurring to handle the rapid head movements typical of first-person video.

Key Results

EgoSim significantly outperformed baselines like CameraCtrl and MotionCtrl across the board:

  • Better Camera Alignment: On the RealEstate dataset, EgoSim reduced Rotation Error (RotErr) from 1.23 to 0.80 and Translation Error (TransErr) from 9.67 to 8.01.
  • Higher Visual Fidelity: It achieved a much lower Frechet Video Distance (FVD) of 223.8 (lower is better) compared to 318.3 for the nearest competitor on Image-to-Video tasks.
  • Qualitative Success: In side-by-side comparisons, EgoSim was the only model capable of generating "out-of-distribution" movements (like moving toward a corner the training data might consider a "dead end") without the video reversing or freezing.

Limitations and Future Work

A current limitation involves 3D VAE Compression. Many modern video models use 3D Variational Autoencoders to compress video in both space and time. This makes it difficult to map 2D epipolar lines to the compressed latent patches, which can slightly degrade the precision of the attention mechanism.

Looking forward, the authors suggest using EgoSim for Inverse Dynamics—training robots to plan their actions by "imagining" the physical consequences of their movements in this simulated world.