Improving Text Embeddings with Large Language Models

By Liang Wang, Nan Yang, Nan Yang, Xiaolong Huang, Lin-Jun Yang, Linjun Yang, Rangan Majumder, Fu-Ru Wei, Furu Wei. Annual Meeting of the Association for Computational Linguistics, 2023

400 citations recorded by the source corpus.

In this paper, we introduce a novel and simple method for obtaining high-quality text embeddings using only synthetic data and less than 1k training steps. Unlike existing methods that often depend on multi-stage intermediate pre-training with billions of weakly-supervised text pairs, followed by fine-tuning with a few labeled datasets, our method does not require building complex training pipelines or relying on manually collected datasets that are often constrained by task diversity and language coverage. We leverage proprietary LLMs to generate diverse synthetic data for hundreds of thousands of text embedding tasks across 93 languages. We then fine-tune open-source decoder-only LLMs on the synthetic data using standard contrastive loss. Experiments demonstrate that our method achieves strong performance on highly competitive text embedding benchmarks without using any labeled data. Furthermore, when fine-tuned with a mixture of synthetic and labeled data, our model sets new state-of-the-art results on the BEIR and MTEB benchmarks.

Subfields: Artificial Intelligence

View Paper: https://www.semanticscholar.org/paper/88ca4ee548d07263386ca8e4effc4a001bb2716f

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Wei Ping
  2. Bryan Catanzaro
  3. Hong-Liu Cao
  4. Mohammad Shoeybi
  5. Liang Wang
  6. Chankyu Lee
  7. Nan Yang
  8. Fu-Ru Wei

Citations in the corpus