Vector Search

Vector Search is a method to search objects using a digital representation of both the query and the objects - easier explained by example:

Consider a user searching for the flower "Dandelion". A regular text search will match this Wikipedia article, as the "dandelion" term is in the text:

Taraxacum


From Wikipedia, the free encyclopedia (Redirected from Dandelion)

"Dandelion" redirects here. It may refer to any species of the genus Taraxacum or specifically to Taraxacum officinale. For similar plants, see False dandelion. For other uses, see Dandelion (disambiguation)

Taraxacum (/təˈræksəkʊm/) is a large genus of flowering plants in the family Asteraceae, which consists of species commonly known as dandelions.

...


This document is about dandelions, for sure. However, if the user searches for "blowball" (the mature spherical seed head of a dandelion), the text above will not match, although the document is a good match for "blowball". Example images of a dandelion:

DandelionFlower.jpg DandelionFlower.jpg

Photos by Greg Hume

This gets more complicated with more content types. It is hard to use text search to match the images without any text. The same goes for videos, podcasts, songs, TV shows, and so on - this is often called multi-modal search. A workaround is to search the object’s textual metadata, like a song’s title, the podcast episode summary, and the image’s alt text on the webpage. Such metadata is only sometimes available and is often too short/imprecise - this does not solve the root problem.

The query can be an image itself or a song, where the users want more like this, in the context of the current song or an image being viewed. This is a problem found in recommender systems.

Digital representations

A solution to the synonym or multi-modal problem is to change from matching in the textual to the digital domain. This means transforming text, images, and songs into a set of numbers that indicate what it is about.

Examples of different objects, and their vector representation:

DandelionFlower.jpg [0.560, 0.001, 0.223, ...]
Dandelion bu Anna of the North [0.0, 0.011, 0.0, ...]
"Taraxacum is a large genus of flowering plants in the family Asteraceae, which consists of species commonly known as dandelions." [0.002, 0.001, 0.411, ...]
dandelions [0.002, 0.021, 0.355, ...]

The digital representation of the object is hence a sequence of numbers, called a vector, also called an embedding.

By converting all the objects searched for and the query to vectors (digital representations), the search problem is changed into finding similar vectors - i.e., vector search.

How vectors are created

Creating vectors from the objects can be done in multiple ways. Machine learning is often used; there are many ready-made models to get you started.

The quality of the vectors will decide the quality of the search, so this is where organizations will want to spend their effort improving search.

A vector has a dimension (length) and type (type of each cell).

The cost/quality tradeoff is essential - given a vector, it can be represented with lower precision or shortened, keeping the most relevant dimensions. This reduces search precision, but cuts costs into a fraction. A 75% cost reduction might reduce precision, but acceptable for the use case. Vespa supports four types, with an 8x difference in memory cost:

Int88 bits, 1 byte per dimension
bfloat16  16 bits, 2 bytes per dimension
float32 bits, 4 bytes per dimension
double64 bits, 8 bytes per dimension

Read more on selecting the optimal type:

Vectors are often some hundred numbers long, like 768 or 384. A longer vector can hold more information, but some dimensions are more information-rich than others. If most of the values in a dimension are equal or very close to each other, the dimension has little value and can be eliminated:

A better approach is often to use an ML model with the optimal size from the start for the use case, once found.

"dandelion" matches "dandelion", but "rose" does not. Text search has a binary nature. However, both are flowers and considered more similar to each other than "airplane". Unlike text matching, a vector search will not match items exactly. In the following example, we are using a very short 3-dim vector for simplicity and using the dimensions as coordinates in a 3D space:

Nearest neighbor scatter plot

We see that [0.000, 0.497, 0.110] is not equal to [0.101, 0.560, 0.093], but quite close - closer than [0.611, 0.000, 0.217].

In other words, if the values for each dimension are close, the vectors are similar - and this can be used in search to find the nearest neighbor or NN. There are multiple ways to calculate vector similarity using a distance metric:

Approximate for speed

Calculating vector similarity can be thought of as multiplying the number of dimensions with the number of vectors. If you are a Canadian, represented by a 768-dim vector, this is 768 x 39,000,000 = 29,952,000,000, or 29 billion calculations to find the other Canadian most similar to you.

There are ways to approximate this by doing fewer calculations, and still finding the closest vector with high probability - this is called Approximate Nearest Neighbor search, or ANN search. Vespa supports both exact nearest neighbor search and ANN search:

Note that ANNs require some kind of indexing to speed up search, so inserts (adding a new vector) are more expensive (uses more CPU) and takes more space (memory and disk). When evaluating different kinds of ANN indexing, consider if your use case requires updates, including deletes, to the vectors - Vespa supports all.