Understanding Embeddings An embedding is a numerical representation of a piece of information, for example, text, documents, images, audio, etc. The representation captures the semantic meaning of what is being embedded, making it robust for many industry applications. Given the text "What is the main benefit of voting?", an embedding of the sentence could be represented in a vector space, for example, with a list of 384 numbers (for example, [0.84, 0.42, ..., 0.02]). Since this list captures the meaning, we can do exciting things, like calculating the distance between different embeddings to determine how well the meaning of two sentences matches. Embeddings are not limited to text! You can also create an embedding of an image (for example, a list of 384 numbers) and compare it with a text embedding to determine if a sentence describes the image. This concept is under powerful systems for image search, classification, description, and more! How are embeddings generated? The open-source library called allows you to create state-of-the-art embeddings from images and text for free. This blog shows an example of this library. Sentence Transformers What are Embeddings for? "[...] once you understand this ML multitool (embedding), you'll be able to build everything from search engines to recommendation systems to chatbots and a whole lot more. You don't have to be a data scientist with ML expertise to use them, nor do you need a huge labeled dataset." - . Dale Markowitz, Google Cloud Once a piece of information (a sentence, a document, an image) is embedded, the creativity starts; several interesting industrial applications use embeddings. E.g., Google Search uses embeddings to ; Snapchat uses them to " "; and Meta (Facebook) uses them for . match text to text and text to images serve the right ad to the right user at the right time their social search Before they could get intelligence from embeddings, these companies had to embed their pieces of information. An embedded dataset allows algorithms to search quickly, sort, group, and more. However, it can be expensive and technically complicated. In this post, we use simple open-source tools to show how easy it can be to embed and analyze a dataset. Getting started with embeddings We will create a small Frequently Asked Questions (FAQs) engine: receive a query from a user and identify which FAQ is the most similar. We will use the . US Social Security Medicare FAQs But first, we need to embed our dataset (other texts use the terms encode and embed interchangeably). The Hugging Face Inference API allows us to embed a dataset using a quick POST call easily. Since the embeddings capture the semantic meaning of the questions, it is possible to compare different embeddings and see how different or similar they are. Thanks to this, you can get the most similar embedding to a query, which is equivalent to finding the most similar FAQ. Check out our for a more detailed explanation of how this mechanism works. semantic search tutorial In a nutshell, we will: Embed Medicare's FAQs using the Inference API. Upload the embedded questions to the Hub for free hosting. Compare a customer's query to the embedded dataset to identify which is the most similar FAQ. 1. Embedding a dataset The first step is selecting an existing pre-trained model for creating the embeddings. We can choose a model from the . In this case, let's use the because it's a small but powerful model. In a future post, we will examine other models and their trade-offs. Sentence Transformers library "sentence-transformers/all-MiniLM-L6-v2" Log in to the Hub. You must create a write token in your . We will store the write token in . Account Settings hf_token model_id = "sentence-transformers/all-MiniLM-L6-v2"
hf_token = "get your token in http://hf.co/settings/tokens" To generate the embeddings you can use the endpoint with the headers . Here is a function that receives a dictionary with the texts and returns a list with embeddings. https://api-inference.huggingface.co/pipeline/feature-extraction/{model_id} {"Authorization": f"Bearer {hf_token}"} import requests

api_url = f"https://api-inference.huggingface.co/pipeline/feature-extraction/{model_id}"
headers = {"Authorization": f"Bearer {hf_token}"} The first time you generate the embeddings, it may take a while (approximately 20 seconds) for the API to return them. We use the decorator (install with ) so that if on the first try, doesn't work, wait 10 seconds and try three times again. This happens because, on the first request, the model needs to be downloaded and installed on the server, but subsequent calls are much faster. retry pip install retry output = query(dict(inputs = texts)) def query(texts):
    response = requests.post(api_url, headers=headers, json={"inputs": texts, "options":{"wait_for_model":True}})
    return response.json() The current API does not enforce strict rate limitations. Instead, Hugging Face balances the loads evenly between all our available resources and favors steady flows of requests. If you need to embed several texts or images, the would speed the inference and let you choose between using a CPU or GPU. Hugging Face Accelerated Inference API texts = ["How do I get a replacement Medicare card?",
        "What is the monthly premium for Medicare Part B?",
        "How do I terminate my Medicare Part B (medical insurance)?",
        "How do I sign up for Medicare?",
        "Can I sign up for Medicare Part B if I am working and have health insurance through an employer?",
        "How do I sign up for Medicare Part B if I already have Part A?",
        "What are Medicare late enrollment penalties?",
        "What is Medicare and who can get it?",
        "How can I get help with my Medicare Part A and Part B premiums?",
        "What are the different parts of Medicare?",
        "Will my Medicare premiums be higher because of my higher income?",
        "What is TRICARE ?",
        "Should I sign up for Medicare Part B if I have Veterans' Benefits?"]

output = query(texts) As a response, you get back a list of lists. Each list contains the embedding of a FAQ. The model, , is encoding the input questions to 13 embeddings of size 384 each. Let's convert the list to a Pandas of shape (13x384). "sentence-transformers/all-MiniLM-L6-v2" DataFrame import pandas as pd
embeddings = pd.DataFrame(output) It looks similar to this matrix: [[-0.02388945  0.05525852 -0.01165488 ...  0.00577787  0.03409787  -0.0068891 ]
 [-0.0126876   0.04687412 -0.01050217 ... -0.02310316 -0.00278466   0.01047371]
 [ 0.00049438  0.11941205  0.00522949 ...  0.01687654 -0.02386115   0.00526433]
 ...
 [-0.03900796 -0.01060951 -0.00738271 ... -0.08390449  0.03768405   0.00231361]
 [-0.09598278 -0.06301168 -0.11690582 ...  0.00549841  0.1528919   0.02472013]
 [-0.01162949  0.05961934  0.01650903 ... -0.02821241 -0.00116556   0.0010672 ]] 2. Host embeddings for free on the Hugging Face Hub 🤗 Datasets is a library for quickly accessing and sharing datasets. Let's host the embeddings dataset in the Hub using the user interface (UI). Then, anyone can load it with a single line of code. You can also use the terminal to share datasets; see for the steps. In the of this entry, you will be able to use the terminal to share the dataset. If you want to skip this section, check out the repo with the embedded FAQs. the documentation notebook companion ITESM/embedded_faqs_medicare First, we export our embeddings from a Pandas to a CSV. You can save your dataset in any way you prefer, e.g., zip or pickle; you don't need to use Pandas or CSV. Since our embeddings file is not large, we can store it in a CSV, which is easily inferred by the function we will employ in the next section (see the ), i.e., we don't need to create a loading script. We will save the embeddings with the name . DataFrame datasets.load_dataset() Datasets documentation embeddings.csv embeddings.to_csv("embeddings.csv", index=False) Follow the next steps to host in the Hub. embeddings.csv Click on your user in the top right corner of the . Hub UI Create a dataset with "New dataset." Choose the Owner (organization or individual), name, and license of the dataset. Select if you want it to be private or public. Create the dataset. Go to the "Files" tab (screenshot below) and click "Add file" and "Upload file." Finally, drag or upload the dataset, and commit the changes. Now the dataset is hosted on the Hub for free. You (or whoever you want to share the embeddings with) can quickly load them. Let's see how. 3. Get the most similar Frequently Asked Questions to a query Suppose a Medicare customer asks, "How can Medicare help me?". We will which of our FAQs could best answer our user query. We will create an embedding of the query that can represent its semantic meaning. We then compare it to each embedding in our FAQ dataset to identify which is closest to the query in vector space. find Install the 🤗 Datasets library with . Then, load the embedded dataset from the Hub and convert it to a PyTorch . Note that this is not the only way to operate on a ; for example, you could use NumPy, Tensorflow, or SciPy (refer to the ). If you want to practice with a real dataset, the repo contains the embedded FAQs, or you can use the to this blog. pip install datasets FloatTensor Dataset Documentation ITESM/embedded_faqs_medicare companion notebook import torch
from datasets import load_dataset

faqs_embeddings = load_dataset('namespace/repo_name')
dataset_embeddings = torch.from_numpy(faqs_embeddings["train"].to_pandas().to_numpy()).to(torch.float) We use the query function we defined before to embed the customer's question and convert it to a PyTorch to operate over it efficiently. Note that after the embedded dataset is loaded, we could use the and methods of a to identify the closest FAQ to an embedded query using the . Here is a . FloatTensor add_faiss_index search Dataset faiss library nice tutorial of the alternative question = ["How can Medicare help me?"]
output = query(question)

query_embeddings = torch.FloatTensor(output) You can use the function in the Sentence Transformers library to identify which of the FAQs are closest (most similar) to the user's query. This function uses cosine similarity as the default function to determine the proximity of the embeddings. However, you could also use other functions that measure the distance between two points in a vector space, for example, the dot product. util.semantic_search Install with , and search for the five most similar FAQs to the query. sentence-transformers pip install -U sentence-transformers from sentence_transformers.util import semantic_search

hits = semantic_search(query_embeddings, dataset_embeddings, top_k=5) identifies how close each of the 13 FAQs is to the customer query and returns a list of dictionaries with the top FAQs. looks like this: util.semantic_search top_k hits [{'corpus_id': 8, 'score': 0.75653076171875},
 {'corpus_id': 7, 'score': 0.7418993711471558},
 {'corpus_id': 3, 'score': 0.7252674102783203},
 {'corpus_id': 9, 'score': 0.6735571622848511},
 {'corpus_id': 10, 'score': 0.6505177617073059}] The values ​​in allow us to index the list of we defined in the first section and get the five most similar FAQs: corpus_id texts print([texts[hits[0][i]['corpus_id']] for i in range(len(hits[0]))]) Here are the 5 FAQs that come closest to the customer's query: ['How can I get help with my Medicare Part A and Part B premiums?',
 'What is Medicare and who can get it?',
 'How do I sign up for Medicare?',
 'What are the different parts of Medicare?',
 'Will my Medicare premiums be higher because of my higher income?'] This list represents the 5 FAQs closest to the customer's query. Nice! We used here PyTorch and Sentence Transformers as our main numerical tools. However, we could have defined the cosine similarity and ranking functions by ourselves using tools such as NumPy and SciPy. Additional resources to keep learning If you want to know more about the Sentence Transformers library: The for all the new models and instructions on how to download models. Hub Organization The comparing Sentence Transformer models with GPT-3 Embeddings. Spoiler alert: the Sentence Transformers are awesome! Nils Reimers tweet The , Sentence Transformers documentation on recent research. Nima's thread Thanks for reading! Also Published Here

Facebook

Google

Twitter

A Hitchhiker's Guide to Restaking and Its Risks

Testing and failing

Check my Twitter account!

Technical Developer Advocate

Machine Learning Engineer

Too Long; Didn't Read

How to Get Started With Embeddings

How to Get Started With Embeddings

About Author

Comments

TOPICS

THIS ARTICLE WAS FEATURED IN

Related Stories

Untitled Story

A Hitchhiker's Guide to Restaking and Its Risks

The Noonification: Use This 7-Step McKinsey Framework to Solve Any Problem (1/10/2023)

The Noonification: A Taxonomy of Inclusiveness (1/11/2024)

The Noonification: What is the InfiniteNature-Zero AI Model? (11/19/2022)

10 Ways AI Has Changed Our Lives

A Hitchhiker's Guide to Restaking and Its Risks

The Noonification: Use This 7-Step McKinsey Framework to Solve Any Problem (1/10/2023)

The Noonification: A Taxonomy of Inclusiveness (1/11/2024)

The Noonification: What is the InfiniteNature-Zero AI Model? (11/19/2022)

10 Ways AI Has Changed Our Lives

Light-Mode

Classic

Newspaper

Dark-Mode

Neon Noir

Minty

HN StartUps