We are excited to announce a new embedding model which is significantly more capable, cost effective, and simpler to use. The new model, text-embedding-ada-002
, replaces five separate models for text search, text similarity, and code search, and outperforms our previous most capable model, Davinci, at most tasks, while being priced 99.8% lower.
Embeddings are numerical representations of concepts converted to number sequences, which make it easy for computers to understand the relationships between those concepts. Since the initial launch of the OpenAI /embeddings endpoint, many applications have incorporated embeddings to personalize, recommend, and search content.
You can query the /embeddings endpoint for the new model with two lines of code using our OpenAI Python Library, just like you could with previous models:
import openai
response = openai.Embedding.create(
input="porcine pals say",
engine="text-embedding-ada-002"
)
print(response)
{
"data": [
{
"embedding": [
-0.0108,
-0.0107,
0.0323,
...
-0.0114
],
"index": 0,
"object": "embedding"
}
],
"model": "text-embedding-ada-002",
"object": "list"
}
Model Improvements
Stronger performance. text-embedding-ada-002
outperforms all the old embedding models on text search, code search, and sentence similarity tasks and gets comparable performance on text classification. For each task category, we evaluate the models on the datasets used in old embeddings.
Unification of capabilities. We have significantly simplified the interface of the /embeddings endpoint by merging the five separate models shown above (text-similarity
, text-search-query
, text-search-doc
, code-search-text
and code-search-code
) into a single new model. This single representation performs better than our previous embedding models across a diverse set of text search, sentence similarity, and code search benchmarks.
Longer context. The context length of the new model is increased by a factor of four, from 2048 to 8192, making it more convenient to work with long documents.
Smaller embedding size. The new embeddings have only 1536 dimensions, one-eighth the size of davinci-001
embeddings, making the new embeddings more cost effective in working with vector databases.
Reduced price. We have reduced the price of new embedding models by 90% compared to old models of the same size. The new model achieves better or similar performance as the old Davinci models at a 99.8% lower price.
Overall, the new embedding model is a much more powerful tool for natural language processing and code tasks. We are excited to see how our customers will use it to create even more capable applications in their respective fields.
Limitations
The new text-embedding-ada-002
model is not outperforming text-similarity-davinci-001
on the SentEval linear probing classification benchmark. For tasks that require training a light-weighted linear layer on top of embedding vectors for classification prediction, we suggest comparing the new model to text-similarity-davinci-001
and choosing whichever model gives optimal performance.
Check the Limitations & Risks section in the embeddings documentation for general limitations of our embedding models.
Examples of Embeddings API in Action
Kalendar AI is a sales outreach product that uses embeddings to match the right sales pitch to the right customers out of a dataset containing 340M profiles. This automation relies on similarity between embeddings of customer profiles and sale pitches to rank up most suitable matches, eliminating 40–56% of unwanted targeting compared to their old approach.
<!–
*Caption: The interface of the marketing tool by Kalendar AI. With the new embedding model, it is able to filter and select only a small subset of the audience out of all 56k audience, tightly matching the pitch defined by user inputs.*
–>
Notion, the online workspace company, will use OpenAI’s new embeddings to improve Notion search beyond today’s keyword matching systems.
OpenAI