vision-language retrieval large-scale-training recommendation-systems mmeb-v2 visual-documents multimodal-embedding wechat
Abstract
Universal multimodal embeddings are becoming a core component of modern AI systems, enabling heterogeneous content to be represented in a shared space for applications such as retrieval, recommendation, classification, and agentic systems. In this report, we present WeMM-Embedding, a family of universal multimodal embedding models supporting text, images, videos, visual documents, and arbitrarily interleaved multimodal inputs with flexible output dimensions. The family comprises 2B, 4B, and 9B variants and is trained in two stages: a large-scale multimodal alignment stage, followed by a refinement stage using curated data, fine-grained relevance supervision, and cross-scale knowledge transfer. Across extensive evaluations, WeMM-Embedding achieves leading performance on multiple public benchmarks. Notably, the 2B variant already surpasses the previously leading 8B open-source baseline on MMEB-v2, while the 9B variant further achieves a new state-of-the-art overall score of 80.6. WeMM-Embedding also demonstrates strong practical performance across WeChat applications, with substantial gains on a 26-task in-house benchmark and consistent improvements across 14 online A/B tests. It has been deployed at scale across recommendation and search applications, including WeChat Channels, Official Accounts, Moments, and e-commerce services. We have released the model weights and code to facilitate future research at https://github.com/Tencent/WeMM-Embedding.
한국어 요약
한 줄 요약
WeMM-Embedding은 2B, 4B, 9B 규모의 다중 모달 임베딩 모델로, MMEB-v2에서 80.6 점을 달성하며 실무 성능도 입증했다.
핵심 기여도
- 2B 모델이 MMEB-v2에서 기존 8B 오픈소스 베이스라인을 초과함.
- 9B 모델이 MMEB-v2에서 80.6 점의 최신 최고 성능(SOTA) 달성.
- 26-task 내부 벤치마크에서 실질적인 성능 향상과 14개 온라인 A/B 테스트에서 일관된 개선.
- WeChat 채널, 공식 계정, 모먼츠, 전자상거래 등 대규모 실무 배포.
핵심 아이디어
WeMM-Embedding은 텍스트, 이미지, 동영상, 시각 문서, 임의로 교차된 다중 모달 입력을 처리할 수 있는 유니버설 임베딩 모델이다. 기존 CLIP 스타일 모델은 모달별 인코더를 사용해 교차 모달 입력을 처리하지 못했지만, WeMM-Embedding은 MLLM 기반 접근법을 채택해 임의의 모달 조합을 지원한다. 핵심 아이디어는 두 단계 훈련 전략: 대규모 다중 모달 정렬 단계와, 정제된 데이터, 세부 관련성 감독, 크로스-스케일 지식 전달을 통한 정제 단계이다. 이는 모델이 넓은 범위의 입력을 처리하면서도 세부적인 관련성 모델링을 가능하게 한다.
기술적 접근법
- **모델 규모**: 2B, 4B, 9B 파생 모델.
- **훈련 단계**:
1. **대규모 다중 모달 정렬 단계**: 수억 개의 이질적인 모달 쌍으로 훈련.
2. **정제 단계**: 정제된 데이터, 더 높은 품질의 음성 샘플, 세부 관련성 감독, 크로스-스케일 지식 전달을 사용.
- **입력 지원**: 텍스트, 이미지, 동영상, 시각 문서, 임의로 교차된 입력.
- **출력**: 유연한 출력 차원 지원.
주요 결과
- **MMEB-v2**: 2B 모델이 기존 8B 오픈소스 베이스라인을 초과.
- **MMEB-v2**: 9B 모델이 80.6 점의 최신 최고 성능(SOTA) 달성.
- **내부 벤치마크**: 26-task에서 실질적인 성능 향상.
- **온라인 A/B 테스트**: 14개 테스트에서 일관된 개선.
- **WeChat 서비스**: 채널, 공식 계정, 모먼츠, 전자상거래 등 대규모 실무 배포.
의의 및 한계
WeMM-Embedding은 유니버설 다중 모달 임베딩의 성능-효율성 경계를 새롭게 정의하며, 대규모 실무 적용 가능성을 입증했다. 특히, 2B 모델의 높은 파라미터 효율성은 소규모 모델에서도 뛰어난 성능을 보여준다. 그러나, 모델 확장성과 데이터 큐레이션의 한계는 여전히 개선이 필요한 부분이다. 또한, 모든 모델이 동일한 성능을 보장하지는 않으며, 특정 도메인에서는 추가 훈련이 필요할 수 있다.
실용적 활용
WeMM-Embedding은 WeChat의 추천 및 검색 시스템에 대규모로 배포되어 있으며, 채널, 공식 계정, 모먼츠, 전자상거래 서비스 등 다양한 애플리케이션에서 활용 가능하다. 또한, 오픈소스로 배포되어 연구 및 산업적 활용이 가능하다.