Work

Scaling Sentence Embeddings with Large Language Models

Year: 2024
Type: conference-paper
Abstract: Large Language Models (LLMs) have recently gained significant interest due to their impressive results in various natural language tasks.However, their application to sentence embeddings is still under active research.In this work, we introduce PromptEOL, a simple and efficient method designed to enhance LLM performance on sentence embeddings with a one-word limitation.We further integrate PromptEOL with in-context learning and alignment to leverage LLMs in two settings: without fine-tuning and with fine-tuning.Our extensive experiments show that PromptEOL enables LLMs to generate superior sentence embeddings without fine-tuning, outperforming contrastive learning methods.Additionally, with fine-tuning, a 2.7B parameter model using PromptEOL surpasses the performance of a 4.8B parameter model from previous methods.We also analyze how scaling model parameters, from 125 million to 66 billion, impacts sentence embedding performance.Our code and model is available at https:// github.com/kongds/scaling_sentemb.
Institution: Beihang University
Language: English
Open Access status: gold
OpenAlex ID: W4404782013
FWCI: 17.91
Cites:
Cited by: 61
Related to: 10