Meet Prometheus

A non-generative Joint-Embedding Predictive Architecture learning the physical dynamics of the world from continuous video streams.

Abstract

Traditional computer vision relies on static human labels or wastes computational capacity predicting noisy pixels. Prometheus abandons pixel reconstruction entirely. By learning purely in latent feature space across continuous video streams, it masters intuitive physics, geometric continuity, and spatial reasoning from scratch.

The Joint-Embedding Paradigm Shift

Predicting in latent space bypasses the high-frequency pixel noise that cripples generative models, focusing 100% of capacity on structural invariants.

Comparison: Generative pixel reconstruction (left) spends capacity on unpredictable detail. Prometheus JEPA (right) optimizes for semantic physical representations.

Foundational Architecture Pillars

Every component of Prometheus is designed for real-time spatiotemporal comprehension and zero-collapse stability.

Tubelet Tokenization

Unified Conv2D (16x16) and Conv3D (2x16x16) tubelet embeddings with dedicated learnable modality tokens.

Dynamic 3D RoPE

3-axis Rotary Position Embeddings allocate head dimensions across Depth (time), Height, and Width with GPT-J interleaved phases.

SIGReg Regularization

Constrains latent distributions to an isotropic Gaussian via sliced Epps-Pulley test, eliminating dimensional collapse without heuristics.

MultiModal Tubelet Tokenization & Dynamic 3D RoPE

Video volumes are transformed into 3D tubelet tokens and augmented with rotary phases along all spatial and temporal dimensions.

Modality-specific projection pipelines for single images (Conv2D) and multi-frame videos (Conv3D) producing uniform transformer tokens.

3D Rotary Position Embeddings Architecture

Decomposing attention head dimensions into orthogonal temporal depth, height, and width subspaces enables seamless resolution generalization.

Time Depth (t) 4 / 16
Height (y) 3 / 14
Width (x) 5 / 14

Interactive explorer: Manipulate temporal depth and spatial coordinates to inspect live 3D phase rotation and transformation matrices.

Spatiotemporal Multi-Block Masking

Hierarchical short-range and long-range block masking forces the model to predict complex trajectories across time rather than interpolating local pixels.

Masking Ratio (Tubelet Occlusion) 75%

Interactive simulator: Adjust masking ratio and compare 3D continuous tubelet occlusion with standard 2D independent random noise.

Dual-Stream Self-Supervised Training Loop

The student encoder processes visible context tokens while the EMA teacher produces stable target embeddings across the full video volume.

Complete end-to-end dataflow: Student encoder, Predictor with mask tokens, EMA Teacher target, Dense Predictive Loss, and SIGReg.

Dense Predictive Loss & Distance Weighting

Visible context tokens are supervised with a square-root Euclidean distance weight to the nearest masked patch, maintaining local context fidelity.

Mathematical formulation of the Dense Predictive Loss and context distance curve w_j = sqrt(d_min).

SIGReg: Anti-Collapse via Isotropic Gaussian Geometry

By testing sliced 1D projections against an isotropic Gaussian distribution, SIGReg prevents dimensional collapse without stop-gradients or BatchNorm.

SIGReg Regularizer Strength (λ) λ = 0.20

Interactive inspector: Adjust regularization weight λ to observe the phase transition from representation collapse to an isotropic Gaussian N(0, I).

Optimization Dynamics: Muon + WSD Schedule

Muon orthogonalized momentum updates combined with a Warmup-Stable-Decay schedule accelerate convergence and enable maximal update parameter scaling (muP).

Pretraining convergence curves comparing Muon + WSD against standard AdamW with cosine annealing.

Downstream Transfer via Spatial Tuning Adapter (STA)

Prometheus seamlessly powers real-time detection by fusing single-scale ViT tokens with a lightweight convolutional detail stem into multi-scale feature pyramids (C3, C4, C5).

Spatial Tuning Adapter (STA) multi-scale pyramid integration feeding the D-FINE NMS-free detector in Argus.

Prometheus-AC: Embodied World Model for Robotics

Autoregressively predicts future physical states from actions for robotic trajectory planning.

Block-causal action-conditioned predictor architecture for robotic trajectory simulation and real-time MPC control.

Prometheus Technical Specifications

Key architectural parameters, pretraining configuration, and inference profile.

Backbone Dimension 768 D 12 layers, 12 attention heads
Predictor Dimension 384 D 12 layers, lightweight latent projection
Spatiotemporal Patch 2 × 16 × 16 2 temporal frames, 16x16 spatial pixels
Positional Encoding 3D RoPE Dynamic D/H/W GPT-J rotation
Anti-Collapse Loss SIGReg Sliced Isotropic Gaussian Regularization
Optimizer & Schedule Muon + WSD Matrix Newton-Schulz with warmup-stable-decay
Context Retention ~25% Multi-block masking (8 short + 2 long)
Downstream Adapter STA Pyramid C3, C4, C5 strides 8, 16, 32 for D-FINE

Learn the world

Explore Argus Detection Engine