Modality Forcing

NEW

Key Features

Turns a pretrained text-to-image model into a joint image-depth DiT.
Uses a simple scalable post-training recipe on sparse depth data.
Assigns separate noise levels per modality for image and depth.
Uses per-modality decoders to support sparse real-world depth training.
Supports conditional and joint generation of image and depth.
Shows scaling behavior across 300M to 3B parameter T2I models.
Reports strong depth prediction versus existing joint image-depth generators.
Links to arXiv, GitHub code, and a Hugging Face Space.

The method trains a single DiT on sparse depth data with separate noise levels per modality and per-modality decoders. This lets the model perform conditional and joint generation of image and depth in different permutations, while learning from sparse real-world depth supervision.


Modality Forcing is useful for spatial generation, monocular depth estimation, and image-depth synthesis workflows. The project reports scaling behavior from 300M to 3B parameter T2I models and states that its strongest model is competitive with state-of-the-art monocular depth estimators.

Get more likes & reach the top of search results by adding this button on your site!

Embed button preview - Light theme
Embed button preview - Dark theme
TurboType Banner
Zero to AI Engineer Program

Zero to AI Engineer

Skip the degree. Learn real-world AI skills used by AI researchers and engineers. Get certified in 8 weeks or less. No experience required.

Subscribe to the AI Search Newsletter

Get top updates in AI to your inbox every weekend. It's free!