OmniStream: Mastering Perception, Reconstruction and Action in Continuous Streams
About
Modern visual agents require representations that are general, causal, and physically structured to operate in real-time streaming environments. However, current vision foundation models remain fragmented, specializing narrowly in image semantic perception, offline temporal modeling, or spatial geometry. This paper introduces OmniStream, a unified streaming visual backbone that effectively perceives, reconstructs, and acts from diverse visual inputs. By incorporating causal spatiotemporal attention and 3D rotary positional embeddings (3D-RoPE), our model supports efficient, frame-by-frame online processing of video streams via a persistent KV-cache. We pre-train OmniStream using a synergistic multi-task framework coupling static and temporal representation learning, streaming geometric reconstruction, and vision-language alignment on 29 datasets. Extensive evaluations show that, even with a strictly frozen backbone, OmniStream achieves consistently competitive performance with specialized experts across image and video probing, streaming geometric reconstruction, complex video and spatial reasoning, as well as robotic manipulation (unseen at training). Rather than pursuing benchmark-specific dominance, our work demonstrates the viability of training a single, versatile vision backbone that generalizes across semantic, spatial, and temporal reasoning, i.e., a more meaningful step toward general-purpose visual understanding for interactive and embodied agents.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | ADE20K | mIoU49.1 | 1024 | |
| Image Classification | ImageNet | Top-1 Accuracy84.7 | 431 | |
| Video Question Answering | VideoMME | Accuracy60.7 | 210 | |
| Video Depth Estimation | Sintel | Delta Threshold Accuracy (1.25)58.3 | 193 | |
| Camera pose estimation | Sintel | ATE0.227 | 192 | |
| Video Question Answering | EgoSchema | Accuracy60.9 | 161 | |
| Monocular Depth Estimation | NYU V2 | -- | 131 | |
| Video Depth Estimation | KITTI | Abs Rel0.136 | 126 | |
| Video Question Answering | VideoMMMU | Accuracy40 | 124 | |
| Camera pose estimation | ScanNet | RPE (t)0.017 | 119 |