← all repositories
wangyuxinwhy/uniem

A Chinese embedding model that claims to beat OpenAI's ada-002

uniem trains, fine-tunes, and benchmarks Chinese text embeddings with the explicit goal of outperforming OpenAI's ada-002 on native-language tasks.

876 stars Python RAG · SearchML Frameworks
uniem
Not currently ranked — collecting fresh signals.
star history

What it does

uniem is a Python toolkit for training, fine-tuning, and evaluating Chinese text embedding models. It provides a FineTuner class that wraps several architectures—sentence_transformers, text2vec, and GPT-style models via SGPT—and releases the M3E model family on HuggingFace. The authors claim M3E outperforms OpenAI’s text-embedding-ada-002 on Chinese classification and retrieval tasks, and they back it with an open-source benchmark suite.

The interesting bit

Most embedding leaderboards treat Chinese as an afterthought; MTEB-zh is a deliberate attempt to build a native-language equivalent, benchmarking six model families across news, e-commerce, and financial text. The M3E models are drop-in replacements for sentence-transformers, so they slot into existing vector pipelines without architectural rewrites.

Key highlights

  • M3E models are compatible with the sentence-transformers ecosystem (Chroma, semantic-kernel, etc.)
  • FineTuner supports multiple backbones including GPT-style models via SGPT and Prefix Tuning
  • MTEB-zh evaluates 6 model types across 4 task categories on 9 Chinese datasets
  • Published benchmarks show M3E-base outperforming OpenAI ada-002 on retrieval (map@10, mrr@10, ndcg@10) and classification averages
  • Apache-2.0 licensed

Caveats

  • The README notes a breaking API change in FineTuner between versions 0.2.0 and 0.3.0
  • Retrieval benchmarks use only the first 10,000 documents of T2Ranking due to cost and time constraints with OpenAI’s API
  • Documentation and interface text are primarily in Chinese, which may limit accessibility for non-Chinese speakers

Verdict

Worth a look if you need Chinese-language embeddings or want a training framework that competes with OpenAI’s baseline on native text. Skip it if your workload is English-only and you are already happy with your current retrieval stack.

Frequently asked

What is wangyuxinwhy/uniem?
uniem trains, fine-tunes, and benchmarks Chinese text embeddings with the explicit goal of outperforming OpenAI's ada-002 on native-language tasks.
Is uniem open source?
Yes — wangyuxinwhy/uniem is open source, released under the Apache-2.0 license.
What language is uniem written in?
wangyuxinwhy/uniem is primarily written in Python.
How popular is uniem?
wangyuxinwhy/uniem has 876 stars on GitHub.
Where can I find uniem?
wangyuxinwhy/uniem is on GitHub at https://github.com/wangyuxinwhy/uniem.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.