On this article, you’ll find out how sentence embeddings work and tips on how to construct a completely client-side semantic search engine utilizing Transformers.js, with no server, no API key, and no backend infrastructure required.
Subjects we’ll cowl embody:
- How sentence embeddings and cosine similarity type the inspiration of semantic search.
- The best way to generate and cache embeddings utilizing the Transformers.js feature-extraction pipeline, together with batching and Internet Employee offloading.
- The best way to construct an entire, reusable SemanticSearch class and persist its index throughout web page masses.
Constructing Semantic Search with Transformers.js and Sentence Embeddings
Introduction
You’ve most likely shipped this bug earlier than, the place a person sorts “reasonably priced laptop computer” into your search bar and will get zero outcomes. However you realize the database has dozens of laptop computer articles. They’re simply all titled “funds pocket book.” The phrases are totally different. The that means is similar. Key phrase search treats each as unrelated strings.
This isn’t an edge case. It’s the core limitation of key phrase matching: it compares characters, not ideas. It doesn’t know that “cancel” and “return” describe associated actions, that “damaged” and “faulty” imply the identical factor, or that “I can’t log in” and “account entry subject” are the identical drawback phrased two other ways.
What Sentence Embeddings Truly Are
Semantic search fixes this by evaluating that means. And with Transformers.js, you’ll be able to construct it totally within the browser with no server, no API key, and no backend infrastructure. This tutorial walks by means of the complete pipeline: how sentence embeddings work, tips on how to generate them, how cosine similarity scores relevance, and tips on how to wire all of it right into a working data base search software.
A transformer mannequin can not course of uncooked textual content. Earlier than any computation occurs, a sentence must turn out to be numbers. Embeddings are the results of that conversion: a sentence represented as an inventory of floating-point values referred to as a vector.
The important thing property isn’t simply that sentences turn out to be numbers. It’s that sentences with related that means turn out to be vectors which are geometrically shut to one another in the identical vector area.
The mannequin used all through this tutorial, sentence-transformers/all-MiniLM-L6-v2, maps each sentence to a degree in a 384-dimensional vector area. The mannequin was fine-tuned on over 1 billion sentence pairs particularly to study this geometric property. “I must cancel my order” and “How do I return a product?” find yourself shut collectively. “The climate is gorgeous immediately” finally ends up removed from each.
The 384 dimensions aren’t human-readable. You’ll be able to’t take a look at dimension 47 and say what it encodes. What issues for search isn’t any particular person dimension however the distance between two vectors. Brief distance means related that means. Massive distance means unrelated.
A 3D scatter plot diagram illustrating how semantically related sentences cluster collectively in vector area (click on to enlarge)
Pooling and Normalization
The uncooked transformer mannequin outputs one vector per token; each phrase and subword in a sentence will get its personal vector. For semantic search, you want one vector per sentence.
Imply pooling handles this by averaging all token vectors, weighted by the eye masks, so padding tokens don’t contribute. Normalization then scales the outcome to unit size (magnitude = 1), which simplifies the similarity calculation coated within the subsequent part.
In Transformers.js, each occur routinely once you move quantity[] to the pipeline name. With out these choices, you get token-level embeddings, that are helpful for duties like named entity recognition, however not for sentence-level search.
The Function-Extraction Pipeline
The feature-extraction job is totally different from each different Transformers.js pipeline. Duties like text-classification or question-answering return human-readable outputs: labels, scores, strings. feature-extraction returns the uncooked vector representations that the mannequin computed internally. You’re working one degree decrease, getting the numbers that every one higher-level duties are constructed on high of.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import Float32Array from ‘https://cdn.jsdelivr.web/npm/@huggingface/transformers@3.0.2’;
// Load the feature-extraction pipeline // Xenova/all-MiniLM-L6-v2 is the ONNX-converted model of // sentence-transformers/all-MiniLM-L6-v2 — similar mannequin weights, browser-compatible format const extractor = await pipeline( ‘feature-extraction’, ‘Xenova/all-MiniLM-L6-v2’, quantity[] // 8-bit quantization: smaller obtain (~23 MB), good accuracy );
// Embed a single sentence // pooling: ‘imply’ — averages all token vectors into one sentence vector // normalize: true — scales the outcome to unit size (wanted for cosine similarity) const output = await extractor(‘I need assistance with my order’, Float32Array);
console.log(output); // Tensor Float32Array
// Convert to a plain JavaScript array to be used in your personal code const vector = output.tolist()[0]; // [0.045, 0.073, -0.012, …] — 384 numbers console.log(`Vector size: $quantity[]`); // 384 |
What this code does:
- pipeline() downloads and initializes the mannequin on first run (the browser caches it after that, so subsequent web page masses are immediate)
- You then name the extractor with a string and the 2 choices that provide you with a single, normalized sentence vector
- The result’s a Tensor object; calling .tolist()[0] converts it to a plain JavaScript array of 384 numbers you’ll be able to work with instantly
Understanding the Output Tensor
The Tensor object returned by feature-extraction has three fields price realizing:
- dims is the form [n_sentences, 384]. Move one sentence and dims[0] is 1. Move ten sentences in a batch and dims[0] is 10. The second dimension is at all times 384 for this mannequin
- sort is ‘float32‘, that means every of the 384 values is a 32-bit floating-point quantity
- information is a Float32Array containing all of the numbers in row-major order. For a batch of three sentences, it is a flat array of three × 384 = 1,152 numbers
.tolist() converts the tensor to a nested JavaScript array, one inside array per sentence. output.tolist()[0] offers the vector for the primary sentence as a plain array of 384 numbers.
Batching: Embed A number of Sentences at As soon as
Passing an array of strings to the extractor processes all of them in a single mannequin name. That is considerably quicker than calling the pipeline as soon as per sentence, as a result of the transformer processes all inputs in parallel inside one ahead move.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
// Embed a number of paperwork in a single name — at all times desire this over looping const sentences = [ ‘How do I track my shipment?’, ‘What is your return policy?’, ‘How can I reset my password?’, ‘Do you offer international delivery?’ ];
const batchOutput = await extractor(sentences, quantity[]);
// batchOutput.dims = [4, 384] — 4 sentences, every with 384 dimensions console.log(`Batch form: [${batchOutput.dims}]`);
// Convert to array of arrays — one 384-element array per sentence const vectors = batchOutput.tolist(); console.log(`Quantity of vectors: ${vectors.size}`); // 4 console.log(`Every vector has: ${vectors[0].size} dimensions`); // 384 |
What this code does:
- As a substitute of 4 separate extractor() calls, one name handles all 4 sentences concurrently
- The transformer structure is optimized for batched enter, so the time it takes to embed 10 sentences collectively is way nearer to embedding 1 sentence than to embedding 10 individually
Batching is an important efficiency determination in a semantic search system. When indexing a corpus of fifty paperwork, one batch name is much quicker than 50 particular person calls. The distinction compounds as your corpus grows.
Cosine Similarity: The Math Behind the Search
After you have vectors in your paperwork and a vector for the search question, you want a solution to measure how related any two vectors are. That’s what cosine similarity does.
Cosine similarity measures the angle between two vectors. A rating of 1.0 means the vectors level in the identical course (similar that means). A rating of 0 means they’re fully unrelated. As a result of we used normalize: true when producing embeddings, each vectors have already got unit size (magnitude = 1), which simplifies the formulation significantly:
|
cosine_similarity(A, B) = (A · B) / (|A| × |B|)
Since normalize: true units |A| = |B| = 1, this turns into: cosine_similarity(A, B) = A · B = Σ(A[i] × B[i]) |
Simply sum the element-wise merchandise of the 2 vectors. That quantity is the cosine similarity. For sentence embeddings with imply pooling and normalization, sensible scores fall roughly in these ranges:
| Rating Vary | Interpretation |
|---|---|
| 0.90 to 1.00 | Close to-identical that means |
| 0.70 to 0.90 | Sturdy semantic match |
| 0.50 to 0.70 | Associated subject, totally different angle |
| 0.30 to 0.50 | Unfastened connection |
| Under 0.30 | Probably unrelated |
Right here’s the implementation:
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
/** * Compute cosine similarity between two normalized vectors. * * That is simply the dot product as a result of normalize: true ensures * each vectors have already got unit size, making the denominator 1. * * @param Float32Array vecA – First normalized embedding vector * @param Float32Array vecB – Second normalized embedding vector * @returns {quantity} Similarity rating between -1 and 1 (sometimes 0 to 1 for sentences) */ perform cosineSimilarity(vecA, vecB) { if (vecA.size !== vecB.size) { throw new Error(`Vector size mismatch: ${vecA.size} vs ${vecB.size}`); }
let dotProduct = 0; for (let i = 0; i < vecA.size; i++) { dotProduct += vecA[i] * vecB[i]; // Multiply corresponding components, then sum }
// Clamp to [-1, 1] to deal with floating-point rounding edge instances return Math.max(–1, Math.min(1, dotProduct)); }
// Instance utilization (assuming you have already run these by means of the extractor): // cosineSimilarity(vecA, vecB) — “I must return a product” vs “How do I ship an merchandise again for a refund?” // End result: ~0.82 (semantically related) // // cosineSimilarity(vecA, vecC) — “I must return a product” vs “The inventory market had a unstable week” // End result: ~0.08 (unrelated) |
What this code does:
- The perform loops by means of each 384-element vectors in parallel, multiplies corresponding values, and sums the outcomes
- That sum is the dot product, which equals cosine similarity when each vectors are normalized
- The Math.max(-1, Math.min(1, …)) on the finish handles the uncommon case the place floating-point arithmetic produces a price like 1.0000002 attributable to rounding
Constructing a Semantic Search Class
The sample for semantic search is at all times the identical no matter scale: embed paperwork as soon as at startup, embed every question at search time, rating each doc towards the question, type by rating.
The costly step is producing the 384-number vector for every sentence. Caching these vectors in reminiscence means subsequent searches solely must embed the question, which takes milliseconds.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
/** * SemanticSearch — a easy client-side semantic search engine. * * Utilization: * const search = new SemanticSearch(extractor); * await search.indexDocuments(myDocs); * const outcomes = await search.search(‘my question’, 5); */ class SemanticSearch { constructor(extractor) { // The feature-extraction pipeline occasion (already loaded) this.extractor = extractor;
// Shops paperwork after indexing: { id, textual content, metadata, vector } this.index = []; }
/** * Embed all paperwork and retailer their vectors in reminiscence. * Name this as soon as at startup. Searches reuse these cached vectors. * * @param {Array} docs */ async indexDocuments(docs) { console.time(‘indexing’);
// Pull simply the textual content strings for batch embedding const texts = docs.map(doc => doc.textual content);
// Single batch name embeds all paperwork without delay — a lot quicker than looping const output = await this.extractor(texts, { pooling: ‘imply’, normalize: true });
// Convert the tensor to an array of 384-element arrays, one per doc const vectors = output.tolist();
// Connect every vector to its authentic doc object // The unfold (…doc) preserves all authentic fields: title, URL, tags, and many others. this.index = docs.map((doc, i) => ({ ...doc, vector: vectors[i] }));
console.timeEnd(‘indexing’); console.log(`Listed ${this.index.size} paperwork`); return this; }
/** * Search listed paperwork for probably the most semantically related outcomes. * * @param {string} question – The search question in plain language * @param {quantity} topK – What number of outcomes to return (default: 5) * @returns {Promise
*/ async search(question, topK = 5) { if (this.index.size === 0) { throw new Error(‘No paperwork listed. Name indexDocuments() first.’); }
console.time(‘question embedding’);
// Embed the search question — the one mannequin inference name throughout a search const queryOutput = await this.extractor(question, { pooling: ‘imply’, normalize: true }); const queryVector = queryOutput.tolist()[0];
console.timeEnd(‘question embedding’); console.time(‘scoring’);
// Rating each listed doc towards the question vector // That is pure JavaScript math — no mannequin concerned, so it is immediate const scored = this.index.map(doc => ({ doc, rating: cosineSimilarity(queryVector, doc.vector) }));
// Kind descending — highest relevance rating first scored.type((a, b) => b.rating – a.rating);
console.timeEnd(‘scoring’);
// Return the top-k outcomes, stripping the vector from the output return scored.slice(0, topK).map(({ doc, rating }) => ({ id: doc.id, title: doc.title, textual content: doc.textual content, metadata: doc.metadata, rating: rating })); }
/** * Serialize the index to JSON for storage in localStorage or IndexedDB. * Saves the embedding step on subsequent web page masses. */ toJSON() { return JSON.stringify(this.index); }
/** * Restore a beforehand serialized index with out re-embedding something. * Vectors are plain arrays in JSON and deserialize instantly. */ fromJSON(json) { this.index = JSON.parse(json); return this; } } |
What this code does:
- indexDocuments takes your array of doc objects (every wants at minimal a textual content subject), embeds all of the textual content in a single batch name, and shops the lead to this.index
- The unfold operator (…doc) preserves any metadata you move in, so nothing will get dropped
- search embeds solely the question (one inference name, sometimes underneath 100ms), then runs cosineSimilarity towards each cached doc vector in a plain JavaScript loop. There’s no additional mannequin inference throughout scoring, which is why search feels immediate after indexing completes
- The toJSON and fromJSON strategies allow you to persist the index throughout web page masses, skipping the embedding step totally on return visits
Full Working Demo: Data Base Search
The applying under is full and self-contained. Copy it right into a .html file, open it in any trendy browser, and it really works. The applying makes use of 12 FAQ entries from a fictional e-commerce assist data base. The instance queries are deliberately written with zero key phrase overlap with the matching paperwork to show that semantic search is doing actual work.
You will discover the complete code right here.
What this code does:
- When the web page masses, init() runs instantly. It creates the feature-extraction pipeline with a progress callback that updates the standing line in the course of the mannequin obtain. As soon as the mannequin is prepared, indexDocuments embeds all 12 articles in a single batch name and shops the vectors in reminiscence. The search enter and button are disabled till that step finishes, so customers can’t set off a search mid-index
- When the person searches, search() embeds solely the question (one inference name, sometimes underneath 100ms), then loops by means of all 12 cached doc vectors, computing cosine similarity for every. That scoring loop is pure JavaScript arithmetic with no mannequin concerned, so it finishes in underneath a millisecond. Outcomes are rendered sorted by rating with color-coded match share badges
The instance queries show the important thing functionality. “Low-cost delivery choice” returns “Economic system Supply Choices” on the high regardless of sharing zero key phrases.
Operating Inference in a Internet Employee
The demo above runs all mannequin inference on the primary browser thread. For inner instruments and demos, that is advantageous. For a user-facing manufacturing app, it’s not: mannequin loading and embedding era block the primary thread, that means scroll, enter, and animations all freeze whereas inference is operating. On older {hardware}, the browser could show an “unresponsive web page” warning.
Internet Employees remedy this by operating JavaScript in a background thread. The primary thread stays responsive whereas the Employee handles all mannequin work.
The Employee file (embedder-worker.js):
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
// embedder-worker.js // Runs in a background thread — has no entry to the DOM.
import { pipeline } from ‘https://cdn.jsdelivr.web/npm/@huggingface/transformers@3.0.2’;
// Singleton sample: load the pipeline as soon as and reuse it. // Prevents re-downloading the mannequin if a number of messages arrive rapidly. let extractor = null;
async perform getExtractor() { if (!extractor) { extractor = await pipeline( ‘feature-extraction’, ‘Xenova/all-MiniLM-L6-v2’, { dtype: ‘q8’, progress_callback: (p) => { // Ahead progress updates again to the primary thread for UI show self.postMessage({ sort: ‘progress’, payload: p }); } } ); } return extractor; }
// Hear for embedding requests from the primary thread self.addEventListener(‘message’, async (occasion) => { const { sort, id, payload } = occasion.information;
strive { const ext = await getExtractor();
if (sort === ’embed’) { // payload.texts generally is a single string or an array of strings const output = await ext(payload.texts, { pooling: ‘imply’, normalize: true });
// Convert tensor to plain array earlier than sending again // (Tensor objects will not be transferable throughout threads) self.postMessage({ sort: ’embed_result’, id, // Echo the request ID so the primary thread can match this response payload: output.tolist() }); } } catch (err) { self.postMessage({ sort: ‘error’, id, payload: err.message }); } }); |
Predominant thread communication (fundamental.js):
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
// Create the Employee — it begins loading the mannequin instantly within the background const employee = new Employee(‘./embedder-worker.js’, { sort: ‘module’ });
// Monitor in-flight requests so we are able to resolve them when outcomes come again const pending = new Map(); let requestId = 0;
// Ship an embedding request to the Employee and return a Promise perform embedText(texts) { return new Promise((resolve, reject) => { const id = requestId++;
// Retailer resolve/reject so we are able to name them when the Employee responds pending.set(id, { resolve, reject });
// Ship the request to the background thread employee.postMessage({ sort: ’embed’, id, payload: { texts } }); }); }
// Deal with messages getting back from the Employee employee.addEventListener(‘message’, (occasion) => { const { sort, id, payload } = occasion.information;
if (sort === ‘progress’) { // Replace your loading UI right here if (payload.standing === ‘progress’) { console.log(`Mannequin loading: ${Math.spherical(payload.progress)}%`); } return; }
// Discover the pending Promise that matches this response by ID const p = pending.get(id); if (!p) return; pending.delete(id);
if (sort === ’embed_result’) { p.resolve(payload); // payload is an array of 384-element vectors } else if (sort === ‘error’) { p.reject(new Error(payload)); } });
// Utilization — works the identical because the non-Employee model however stays off the primary thread const vectors = await embedText([‘How do I return a product?’]); console.log(`Embedding dimensions: ${vectors[0].size}`); // 384 |
What this code does:
- The Employee makes use of a singleton sample (getExtractor() creates the pipeline as soon as and returns it on subsequent calls) to keep away from re-downloading the mannequin if a number of messages arrive in fast succession
- The id subject on every message is a correlation key: when the Employee sends again an embed_result, the primary thread makes use of the id to search out the matching Promise within the pending Map and resolve it. With out this, if two embedding requests have been in flight on the similar time, you couldn’t inform which outcome belonged to which request
- The pending Map stays small (one entry per in-flight request) and cleans up after itself as responses arrive
Persisting the Index Throughout Web page Hundreds
Computing embeddings is the sluggish step. For a doc corpus that doesn’t change between visits, you’ll be able to serialize the index to JSON and retailer it in localStorage, so the following web page load skips the embedding step totally.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
// After indexing — save to localStorage const serialized = JSON.stringify(searcher.index); localStorage.setItem(‘kb-index’, serialized); localStorage.setItem(‘kb-index-version’, ‘2025-06-01’); // Replace this when content material modifications
// On web page load — restore the index if it exists and remains to be present const storedVersion = localStorage.getItem(‘kb-index-version’); const currentVersion = ‘2025-06-01’;
if (storedVersion === currentVersion) { const saved = localStorage.getItem(‘kb-index’); if (saved) { searcher.index = JSON.parse(saved); // Vectors are plain arrays in JSON — no particular deserialization wanted console.log(‘Index restored from cache, skipping embedding step’); } } |
localStorage handles round 5 MB, relying on the browser. For 12 paperwork with 384-dimensional float vectors, the serialized index is roughly 200 KB, effectively inside the restrict. For bigger corpora, IndexedDB has no sensible measurement constraint and works the identical manner with a barely extra verbose API.
Scaling Past a Few Hundred Paperwork
The method above scores each doc per question. That works effectively up to some hundred paperwork earlier than latency begins to indicate. For bigger corpora, the official Transformers.js examples repository features a pglite-semantic-search demo that runs an in-browser PostgreSQL occasion with the pgvector extension for approximate nearest neighbor search, which is meaningfully quicker than brute-force scoring for big collections whereas nonetheless holding the whole lot client-side.
Selecting the Proper Mannequin
Xenova/all-MiniLM-L6-v2 is the appropriate default for many English-language use instances. It’s quick, small, and produces sturdy outcomes for semantic search. The desk under covers the primary choices:
For multilingual use instances the place a data base has content material in French, German, and English concurrently, multilingual-e5-small handles cross-lingual queries. A person looking in English will floor related paperwork written in French as a result of the mannequin maps equal meanings to close by vectors no matter language.
Conclusion
The pipeline is 4 steps: load the mannequin as soon as, embed your doc corpus in a batch, embed every question at search time, rating with cosine similarity, and kind. All the things on this tutorial runs from a single CDN import with no server, no API key, and no information leaving the person’s machine.
The identical core ideas — vectors, similarity, and rating — are additionally the inspiration of advice methods, duplicate content material detection, clustering, and retrieval-augmented era. Every of these functions is constructed on the identical feature-extraction pipeline and cosineSimilarity perform coated right here. Begin with the data base demo, prolong the corpus to your personal paperwork, and people extra superior patterns will make sense rapidly when you’ve seen the fundamentals working.
