Demystifying Tokens in LLMs
Transformer Embeddings and Tokenization
Blog written by José Thomaz
Follow the post here
With the rise of AI, vector databases are becoming more popular. But what exactly is a vector database and when should you use it?
What is a vector database?
Traditional search engines use full-text search, but NLPs like ChatGPT and Bing AI use semantic search or similarity search, a type of search that considers not only the characters match, but also the meaning of the words.
This feature of semantic search is powered by vector databases.
- Recently Google also started using semantic search
Full-text search VS Semantic search
Let's compare these two types of search.
Full-text search
- Search for a word or phrase in large amounts of text
- The search engine returns a list of documents that contain the search term
Semantic search
- Search for a word or phrase in large amounts of text
- The search engine will return a list of documents that contain the search term or have a similar meaning
How does vectorization work?
Vectorization is the fundamental of vector databases, so first we need to understand this technique and its advantages.
NLP Example
-
A NLP receives a text as input and returns another text as output. This involves several processing steps, one of them is vectorization.
-
In the vectorization process, the tokens of the input text are converted into vectors using linear algebra operations. Simplifying:
A text becomes an array with numeric values, for example:
-
The result of this "translation" is called a vector embedding. The vector embedding is a numerical representation of the input text.
-
It works not only for text, but also for images, audio, and video. Each of these types of data use a different algorithm to generate the vector embedding.
Vectorization algorithms
Images: vgg, resnet, inception, mobilenet, googlenet, etc.
Text: word2vec, glove, fasttext, BERT, node2vec, etc.
Audio: wav2vec, mxnet, etc.
Full-text search VS Semantic search
-
Now we have a lot of vectors, but these vectors are basically numbers, matrices, they don't mean anything by themselves. So the vector database is used to index these vectors.
-
The vector database will index the vectors, leaving the vectors with similar values close to each other, this makes it possible to perform a similarity search in the database.
How the vector database knows which vectors are similar?
-
Querying the database using math formulas to find the closest vectors in a high-dimensional space
- Euclidean distance
- Cosine similarity
-
Machine learning algorithms
- K-nearest neighbors (KNN)
- Approximate nearest neighbors (ANN)
-
Indexing
Examples
Example with football players ⚽
Let's suppose you want to search for the following text:
You will very likely get results talking about Messi, because currently (2023) he was elected the best football player in the world. But you will also get results talking about Cristiano Ronaldo, Mbappé, Neymar, etc. as they are also very good football players and considered "the best" by many fans.
Performing a semantic search in a dataset
-
In this code, we are performing a semantic search on the terms "politics", "global", and "economy".
-
As result, we get the 10 most semantically close terms, with the term and the numerical proximity coefficient.
Complete code: semantic-search.py
