Search papers, labs, and topics across Lattice.
This paper introduces UEmbed, a decoder-only multimodal embedding model that integrates both sparse lexical and dense representations in a single causal forward pass, overcoming limitations of existing Learned Sparse Retrieval (LSR) methods that rely on bidirectional architectures and auxiliary modules. By appending learnable tokens and partitioning the vocabulary into disjoint subsets, UEmbed achieves competitive performance on benchmarks like MMEB-v2 and BEIR, demonstrating its effectiveness across various applications. The key result shows that UEmbed-9B outperforms other multimodal embedding models while maintaining efficiency, marking a significant advancement in the integration of sparse retrieval with multimodal inputs.
UEmbed achieves a groundbreaking unification of dense and sparse embeddings, outperforming existing models in multimodal tasks while streamlining the retrieval process.
Sparse retrieval underpins modern search systems, from web search to retrieval-augmented generation. Existing work has introduced Learned Sparse Retrieval (LSR) to push beyond exact lexical matching toward richer semantics. Yet LSR has so far remained tied to encoder-style bidirectional architectures, and its extension to multimodal settings still relies heavily on auxiliary cross-modal modules. To address these limitations, we introduce UEmbed (Unified Embedding), a decoder-only multimodal embedding model that produces both sparse lexical and dense representations in one causal forward pass. UEmbed appends N learnable special tokens to the input and partitions the vocabulary into N disjoint subsets. Each token's causal hidden state predicts sparse weights over its assigned subset, and the N subsets are concatenated into the full sparse vector. Trained on public data, we release UEmbed at 2B, 4B, and 9B scales. UEmbed-9B reaches 71.8 (dense) and 71.0 (sparse) on MMEB-v2, outperforming multimodal embedding models trained on publicly available data (e.g., RzenEmbed). On BEIR, UEmbed also remains competitive with strong dense and sparse baselines. Furthermore, we demonstrate the practical utility of UEmbed across three dimensions: effectiveness, efficiency, and agentic applications. Overall, UEmbed offers a new paradigm: it unifies dense and sparse embeddings in one model, while further extending sparse retrieval to unify text and multimodal inputs.
