惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

D
DataBreaches.Net
罗磊的独立博客
M
MIT News - Artificial intelligence
G
Google Developers Blog
V
V2EX
D
Docker
博客园_首页
The Cloudflare Blog
人人都是产品经理
人人都是产品经理
Y
Y Combinator Blog
WordPress大学
WordPress大学
T
Tailwind CSS Blog
博客园 - 司徒正美
J
Java Code Geeks
L
LangChain Blog
博客园 - 三生石上(FineUI控件)
B
Blog RSS Feed
博客园 - 【当耐特】
小众软件
小众软件
Apple Machine Learning Research
Apple Machine Learning Research
大猫的无限游戏
大猫的无限游戏
P
Proofpoint News Feed
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
博客园 - Franky

DEV Community

Authentication Security Deep Dive: From Brute Force to Salted Hashing (With Java Examples) Why AI Systems Don’t Fail — They Drift Spilling beans for how i learn for exam😁"Reinforcement Learning Cheat Sheet" I Replaced Chrome with Safari for AI Browser Automation. Here's What Broke (and What Finally Worked) How Python Borrows Other People's Work The $40 Architecture: Processing 1 Billion API Requests with 99.99% Uptime Vibe Coding: A Workflow Guide (From Zero to SaaS) Most webhook security guides protect the wrong side. The scary part is delivery. Headless CMS for TanStack Start: Build a Blog with Cosmic EU Age Verification App "Hacked in 2 Minutes" — What Actually Happened Comfy Cloud’s delete function does not actually remove files Running AI Models on GPU Cloud Servers: A Beginner Guide Event-driven media intelligence with AWS Step Functions and Bedrock I scored 500 AI prompts across 8 quality dimensions — here's what broke How to Call Google Gemini API from Next.js (Free Tier, No Backend Needed) The Portal Protocol: Reclaiming Human Connection in the Age of AI How to Fix Your Team's Scattered Knowledge Problem With a Self-Hosted Forum Intro to tc Cloud Functors: A Graph-First Mental Model for the Modern Cloud Designing Multi-Tenant Backends With Both Ownership and Team Access I Built a Neumorphic CSS Library with 77+ Components — Here's What I Learned PostgreSQL Performance Optimization: Why Connection Pooling Is Critical at Scale Cómo construí un SaaS multi-rubro para gestionar expensas en Argentina con FastAPI + Vue 3 🚀 I Built an Ethical Hacking Scanner Tool – Open Source Project I Replaced /usage and /context in Claude Code With a Single Statusline A Pythonic Way to Handle Emails (IMAP/SMTP) with Auto-Discovery and AI-Ready Design I Collected 8.9 Million Polymarket Price Points — Here's What I Found About How Markets Really Move EcoTrack AI — Carbon Footprint Tracker & Dashboard Everyone's Using AI. No One Agrees How. 5 self-hosted ebook managers worth trying in 2026 Building Your First AI Agent with LangChain: From Chatbot to Autonomous Assistant
Understanding Text Similarity with Embeddings and Cosine ...
Venu171 · 2026-05-01 · via DEV Community

Venu171

How to measure semantic similarity between sentences using modern NLP techniques


Introduction

Have you ever wondered how search engines or chatbots understand that "Machine Learning affects all areas of life" is much more similar to "Artificial intelligence is transforming the world" than "Maradona was one of the best football players in history"?

This isn't magic — it's embeddings + cosine similarity.

In this blog post, we'll break down exactly how this works, starting from the mathematical foundation and ending with real, runnable Python code using Hugging Face Transformers.

By the end, you'll understand:

  • What text embeddings actually are
  • Why cosine similarity is the go-to metric
  • How to implement semantic text similarity from scratch
  • Real-world results using the BART model

Let's dive in!


What Are Text Embeddings?

Embeddings are numerical vectors that capture the meaning of text in a high-dimensional space.

Instead of treating words as isolated tokens, modern transformer models (like BERT, BART, or GPT) convert entire sentences into dense vectors (typically 768 or 1024 dimensions).

Key property: Semantically similar texts end up close to each other in this vector space.

  • "King" and "Queen" → close vectors
  • "King" and "Apple" → far apart vectors
  • "Artificial Intelligence" and "Machine Learning" → very close

This is the foundation of semantic search, RAG systems, recommendation engines, and plagiarism detection.


The Magic Metric: Cosine Similarity

Once we have two embedding vectors, how do we quantify how similar they are?

We use cosine similarity — the cosine of the angle between two vectors.

The Formula

Interpretation:

  • 1.0 → Identical direction (very similar meaning)
  • 0.0 → Orthogonal (unrelated)
  • -1.0 → Opposite direction (opposite meaning)

Why cosine and not Euclidean distance?

Cosine similarity is magnitude invariant. It only cares about the direction (i.e., the semantic orientation), not the length of the vectors. This makes it perfect for comparing texts of different lengths.


A Concrete Example (With Numbers!)

To make this crystal clear, let's use simplified 4-dimensional vectors (real models use 768-D, but the math is identical).

Our Sentences

Source: "Artificial intelligence is transforming the world."

Embedding: [0.85, 0.65, 0.12, 0.25]

Candidate 1 (sports): "Maradona was one of the best football players in history."

Embedding: [0.15, 0.08, 0.92, 0.30]

Candidate 2 (tech): "Machine Learning affects all areas of life."

Embedding: [0.78, 0.58, 0.18, 0.22]

Step-by-Step Calculation (Candidate 2)

1. Dot Product

0.85 × 0.78 + 0.65 × 0.58 + 0.12 × 0.18 + 0.25 × 0.22
= 0.663 + 0.377 + 0.0216 + 0.055 = 1.1166

Enter fullscreen mode Exit fullscreen mode

2. Vector Magnitudes

  • Source: √(0.85² + 0.65² + 0.12² + 0.25²) ≈ 1.1054
  • Candidate 2: √(0.78² + 0.58² + 0.18² + 0.22²) ≈ 1.0127

3. Final Cosine Similarity

cos(θ) = 1.1166 / (1.1054 × 1.0127) ≈ **0.997**

Enter fullscreen mode Exit fullscreen mode

Result:

  • Candidate 1 (sports): 0.336
  • Candidate 2 (tech): 0.997

The model correctly identifies that the second sentence is almost semantically identical to the source!


Real-World Implementation with BART

Now let's see how this works with actual transformer embeddings.

Here's the complete, production-ready code:

from transformers import pipeline
import torch

# Load feature extraction pipeline
feature_extractor = pipeline(
    "feature-extraction",
    model="facebook/bart-base"
)

def get_sentence_embedding(text):
    """Convert text to averaged embedding vector."""
    embeddings = feature_extractor(text)
    tensor = torch.tensor(embeddings).squeeze(0)  # Remove batch dim
    return tensor.mean(dim=0, keepdim=True)       # Average over tokens

def text_similarity(text1, text2):
    """Compute cosine similarity between two sentences."""
    emb1 = get_sentence_embedding(text1)
    emb2 = get_sentence_embedding(text2)
    return torch.nn.functional.cosine_similarity(emb1, emb2).item()

# Example usage
source = "Artificial intelligence is transforming the world."
candidates = [
    "Maradona was one of the best football players in history.",
    "Machine Learning affects all areas of life."
]

print(f"Source: {source}\n")
for cand in candidates:
    score = text_similarity(source, cand)
    print(f"{cand}")
    print(f"   Similarity: {score:.4f}\n")

Enter fullscreen mode Exit fullscreen mode

Output (actual run):

Source: Artificial intelligence is transforming the world.

→ Maradona was one of the best football players in history.
   Similarity: 0.4625

→ Machine Learning affects all areas of life.
   Similarity: 0.7117

Enter fullscreen mode Exit fullscreen mode

Beautiful! The model gives us exactly the expected behavior.


Why This Technique is So Powerful

This simple pattern powers many modern AI applications:

Application How Cosine Similarity Helps
Semantic Search Find documents with similar meaning, not just keywords
RAG Systems Retrieve the most relevant context for LLMs
Duplicate Detection Identify paraphrased content
Recommendation Suggest similar articles, products, or movies
Clustering Group documents by topic automatically

Key Takeaways

  1. Embeddings turn text into numbers that capture meaning.
  2. Averaging token embeddings gives you a robust sentence vector.
  3. Cosine similarity is the standard way to compare these vectors.
  4. You don't need massive models — even facebook/bart-base (535M params) works surprisingly well for this task.
  5. This technique is foundational to almost every modern NLP application.

Try It Yourself

Want to experiment?

  1. Install the dependencies:
   pip install transformers torch

Enter fullscreen mode Exit fullscreen mode

  1. Run the code above (first run will download ~535MB model).

  2. Try your own sentences!


Conclusion

Text similarity using embeddings and cosine similarity is one of those "simple but incredibly powerful" techniques in NLP. Once you understand the vector space intuition and the math behind cosine similarity, a whole world of applications opens up — from building smarter search engines to improving RAG pipelines.

The best part? You now have the complete mental model and the working code to start building with it today.


What will you build with this technique?

Drop your ideas in the comments!


Further Reading


Thanks for reading! If you found this helpful, consider sharing it with your network.

Written with ❤️ for the NLP community

May 2026