Modality Forcing for Scalable Spatial Generation

Multimodal Intelligence Vision-Language Pre-training GenAI Diffusion TDIG CV Monocular or Binocular Depth Estimation and Occlusion Handling
文本到图像(T2I)模型蕴含着丰富的空间先验知识。要合成逼真且场景杂乱的图像,需具备对几何关系的理解能力,包括透视关系与相对尺度。此前的研究尝试将T2I模型加以改造,以利用该空间先验实现深度预测,但这类方法依赖密集的深度标注数据,且训练流程复杂。我们提出“模态强制”(Modality Forcing)方法——一种简洁、可扩展的后训练策略,仅需一个在稀疏深度数据上训练的DiT(Diffusion Transformer)模型,即可实现图像与深度图的联合生成。该方法通过为不同模态(图像与深度)分别设定独立的噪声调度水平,支持以任意顺序进行条件生成或联合生成(例如:给定文本生成图像与深度图;给定图像生成深度图;或给定深度图生成图像)。各模态专属的解码器结构,使我们得以直接在真实世界中采集的稀疏深度数据上进行训练,并获得强泛化能力与高精度的深度预测结果。进一步实验表明,“模态强制”完整继承了T2I预训练的可扩展性:我们从零开始训练了一系列不同规模的T2I模型(参数量从3.7亿至33亿),发现模型规模越大、所用图像训练数据越多,其深度预测精度越高。我们性能最强的模型,在单目深度估计任务上可与当前最优方法相媲美,并相较现有联合图像-深度生成模型将绝对相对误差(AbsRel)降低了57%。上述结果有力证明:图像生成可作为一种高效、可扩展的预训练范式,用于提升模型的空间感知能力。https://modality-forcing.github.io/
Text-to-image (T2I) models contain rich spatial priors. Synthesizing photorealistic, cluttered scenes requires an understanding of geometry, including perspective and relative scale. Prior works adapt T2I models to leverage this prior for depth prediction, but they require dense depth data and involve complex recipes. We propose Modality Forcing, a simple, scalable post-training recipe for joint image-depth generation using a single DiT trained on sparse depth data. Modality Forcing enables conditional and joint generation of image and depth in any permutation by assigning separate noise levels per modality. Per-modality decoders let us train on sparse, real-world depth and achieve strong, generalizable depth prediction. We further show that Modality Forcing inherits the scalability of T2I pre-training: by training a set of T2I models from scratch (370M to 3.3B parameters), we find that larger models trained on more image data produce more accurate depth. Our strongest model is competitive with state-of-the-art monocular depth estimators and reduces AbsRel by 57% relative to existing joint image-depth generative models. These results provide strong evidence that image generation is a scalable pre-training objective for spatial perception. https://modality-forcing.github.io/
许愿