Abstract
DME is a two-stage multimodal embedding model that merges efficient contrastive pre-training with training-only reasoning and reconstruction mechanisms to achieve fine-grained discrimination at billion-scale serving cost.
Multimodal representation learning is a cornerstone of modern AI. By encoding multimodal queries and targets into vectors, it powers industrial search and recommendation and underpins modern agents. Real-world platforms with complex modalities and massive-scale content, such as Douyin, Xiaohongshu, and YouTube, demand both efficiency under billion-scale indexing and fine-grained discrimination for hard matching. Existing MLLM embedding models rarely satisfy both. Contrastive models are efficient but rely on pair-level supervision too coarse for fine-grained distinctions, while CoT-based models improve discrimination through explicit generation impractical to serve online. We present Douyin Multimodal Embedding (DME), a model trained in two stages to combine both strengths. Stage 1 performs large-scale contrastive pre-training that establishes a unified multimodal embedding space with broad modality and task coverage. Stage 2 supplements semantic sufficiency, the property that an embedding is grounded in retrieval-relevant evidence and preserves fine-grained counterpart-side semantics, via two mechanisms. Evidence-Grounded Typed Latent Reasoning organizes retrieval evidence through hidden-space latent reasoning, and Cross-Conditional Reconstruction enforces counterpart-side semantics through cross-directional autoregressive reconstruction. Both act only during training and add only marginal query-side overhead, so DME serves as efficiently as a standard contrastive encoder. On MMEB-v2, DME reaches state-of-the-art results at comparable scales for its 2B and 9B variants (74.8 and 78.4), with especially strong video and visual-document tasks. In production, DME delivers a 2.92% relative gain on Douyin's in-house offline evaluation set, is deployed across Douyin scenarios such as generative, image, and AI search, and yields a 0.1% Lifetime (LT) gain in online A/B testing on Douyin search.
Community
Introducing DME, a two-stage framework for industrial multimodal embeddings! ๐
Stage 1 scales heterogeneous contrastive pre-training across text, image, video, and visual documents.
Stage 2 goes beyond pairwise alignment with Evidence-Grounded Typed Latent Reasoning and Cross-Conditional Self-Decoding via NTP/MTP.
DME achieves the current SOTA on MMEB-v2, and delivers +2.92% offline and +0.1% online LT gains in a real Douyin search scenario.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- ReLoop-UME: Recurrent Depth with Learnable Retrieval Registers for Universal Multimodal Embedding (2026)
- The Case Against Generation for Retrieval: Discriminative Language Models as Effective Retrievers (2026)
- Learning from Failures: Retrieval-Centric CoT via Hard Negatives for Unified Multimodal Retrieval (2026)
- UEmbed: Unified Sparse and Dense Multimodal Embeddings (2026)
- GALA: Generative Aligned Learning for Adaptive Multimodal Representation in the Taobao Shangou Recommender System (2026)
- LightSTAR: Efficient Visual Document Retrieval via Lightweight Selection with Vision-Adaptive Refinement (2026)
- Illuminating Visual Identity in Universal Multimodal Embeddings (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper