AI Concepts
Learn how embeddings are used to build AI applications
Embeddings are an essential component in building AI applications. This topic describes embeddings and how they are used, generated, and stored in POstgres.
What are embeddings?
When working with unstructured data, a common objective is to transform it into a more structured format that is easier to analyze and retrieve. This transformation can be achieved through the use of 'embeddings', which are vectors containing an array of floating-point numbers that represent the features or dimensions of your data. For example, a sentence like "The cow jumped over the moon" might be represented by an embedding that looks like this: [0.5, 0.3, 0.1].
The advantage of embeddings is that they allow us to measure the similarity between different pieces of text. By calculating the distance between two embeddings, we can assess their relatedness - the smaller the distance, the greater the similarity, and vice versa. This quality is particularly useful as it enables embeddings to capture the underlying meaning of the text.
Take the following three sentences, for example:
- Sentence 1: "The cow jumped over the moon."
- Sentence 2: "The bovine leaped above the celestial body."
- Sentence 3: "I enjoy eating pancakes."
You can determine the most similar sentences by following these steps:
-
Generate embeddings for each sentence. For illustrative purposes, assume these values represent actual embeddings:
- Embedding for sentence 1 → [0.5, 0.3, 0.1]
- Embedding for sentence 2 → [0.6, 0.29, 0.12]
- Embedding for sentence 3 → [0.1, -0.2, 0.4]
-
Compute the distance between all pairs of embeddings (1 & 2, 2 & 3, and 1 & 3).
-
Identify the pair of embeddings with the shortest distance between them.
When we apply this process, it is likely that sentences 1 and 2, both of which involve jumping cattle, will emerge as the most related according to a distance calculation.
Vector similarity search
Transforming data into embeddings and computing similarities between one or more items is referred to as vector search or similarity search. This process has a wide range of applications, including:
- Information retrieval: By representing user queries as vectors, we can perform more accurate searches based on the meaning behind the queries, allowing us to retrieve more relevant information.
- Natural language processing: Embeddings capture the essence of the text, making them excellent tools for tasks such as text classification and sentiment analysis.
- Recommendation systems: Using vector similarity, we can recommend items similar to a given item, whether they be movies, products, books, or otherwise. This technique allows us to create more personalized and relevant recommendations.
- Anomaly detection: By determining the similarity between items within a dataset, we can identify outliers or anomalies—items that don't quite fit the pattern. This can be crucial in many fields, from cybersecurity to quality control.
Distance metrics
Vector similarity search computes similarities (the distance) between data points. Calculating how far apart data points are helps us understand the relationship between them. Distance can be computed in different ways using different metrics. Some popular distance metrics include:
- Euclidean (L2): Often referred to as the "ordinary" distance you'd measure with a ruler.
- Manhattan (L1): Also known as "taxicab" or "city block" distance.
- Cosine: This calculates the cosine of the angle between two vectors.
Other distance metrics supported by the pgvector
extension include Hamming distance and [Jaccard distance]https://en.wikipedia.org/wiki/Jaccard_index).
Different distance metrics can be more appropriate for different tasks, depending on the nature of the data and the specific relationships you're interested in. For instance, cosine similarity is often used in text analysis.
Generating embeddings
A common approach to generating embeddings is to use an LLM API, such as OpenAI’s Embeddings API. This API allows you to input a text string into an API endpoint, which then returns the corresponding embedding. The "cow jumped over the moon" is a simplistic example with 3 dimensions. Most embedding models generate embeddings with a much larger number of dimensions. OpenAI's newest and most performant embedding models, text-embedding-3-small
and text-embedding-3-large
, generate embeddings with 1536 and 3072 dimensions by default, respectively.
Here's an example of how to use OpenAI's text-embedding-3-small
model to generate an embedding:
note
Running the command above requires an OpenAI API key, which must be obtained from OpenAI.
Upon successful execution, you'll receive a response similar to the following:
To learn more about OpenAI's embeddings, see Embeddings. Here, you'll find an example of obtaining embeddings from an Amazon fine-food reviews dataset supplied as a CSV file. See Obtaining the embeddings.
There are many embedding models you can use, such as those provided by Mistral AI, Cohere, Hugging Face, etc. AI tools like LanngChain provide interfaces and integrations for working with a variety of models. See LangChain: Text embedding models. You'll also find a Neon Postgres guide on the LangChain site and Class NeonPostgres, which provides an interface for working with a Neon Postgres database.
Storing vector embeddings in Postgres
Neon supports the pgvector Postgres extension, which enables the storage and retrieval of vector embeddings directly within your Postgres database. When building AI applications, installing this extension eliminates the need to extend your architecture to include a separate vector store. Installing the pgvector
extension simply requires running the following CREATE EXTENSION
statement from the Neon SQL Editor or any SQL client connected to your Neon Postgres database.
After installing the pgvector
extension, you can create a table to store your embeddings. For example, you might define a table similar to the following to store your embeddings:
To add embeddings to the table, you would insert the data as shown:
For detailed information about using pgvector
, refer to our guide: The pgvector extension.
Last updated on