码农之家

专注优质代码开发,为软件行业发展贡献力量

RAG架构与代码之向量空间

一、RAG核心技术架构:三阶处理流程

核心组件作用:

文本分块:平衡信息完整性与检索效率(建议长度512-1024 token)

向量嵌入:将文本映射为高维空间中的数学表示(如1536维向量)

混合检索:结合语义向量与关键词BM25,召回率提升30%

二、向量空间本质:数学视角解析

2.1 向量运算的语义意义

import numpy as np
# 示例:词向量关系推理
king = np.array([1.2, 0.8, -0.5])
queen = np.array([1.0, 0.9, -0.6])
man = np.array([0.9, 0.7, -0.3])
woman = king - man + queen  # 结果 ≈ [1.1, 0.8, -0.4]

关键特性:

余弦相似度:cosθ = A·B / (||A||·||B||),衡量语义相关性(-1到1)

 聚类特性:相似主题文档在向量空间中聚集

三、嵌入模型选型:MTEB榜单权威指南

     在当前主流的文本嵌入模型中,各技术方案在性能表现、语言支持与经济成本方面呈现出显著差异。Gemini-Embed 模型在 MTEB 基准测试中得分为 68.37,具备良好的多语言支持能力,每百万 token 成本为 0.15 美元,适合跨语言检索场景,是构建多语言搜索系统的理想选择。

     若聚焦于中文应用场景,Yuan-EB 模型展现出明显优势,其 MTEB 得分高达 78.41,专门针对中文语境进行了优化,且为开源模型,无使用成本限制,特别适合构建企业级中文知识库系统。

     相比之下,OpenAI text-3 模型在英语任务中表现稳定(MTEB 58.93),但语言支持以英语为主,每百万 token 成本为 0.20 美元,更适用于以英文内容为主的应用环境。

     BGE-M3 模型同样为多语言场景提供了有力支持,MTEB 得分为 76.20,性能接近 Yuan-EB,且为开源模型,具备良好的综合性价比,在需要平衡多语言能力与部署成本的场景中表现突出。

     总体而言,模型选择应紧密结合实际业务需求:若以中文环境为核心,Yuan-EB 优势显著;若系统需处理多语言混合内容,Gemini-Embed 与 BGE-M3 是更优选择;而 OpenAI text-3 则适用于以英文为主的轻量化应用场景。

四、向量生成与匹配原理

4.1 嵌入生成流程(以BERT为例)

from transformers import AutoTokenizer, AutoModel
import torch
# 1. 文本分词
tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
inputs = tokenizer("Quantum computing principles", return_tensors="pt")
# 2. 模型前向传播
model = AutoModel.from_pretrained("bert-base-uncased")
with torch.no_grad():
    outputs = model(**inputs)
# 3. 向量池化(均值策略)
embeddings = torch.mean(outputs.last_hidden_state, dim=1).squeeze()
print(embeddings.shape)  # 输出:torch.Size([768])

4.2 相似度匹配优化

问题:原始余弦相似度在长尾分布中表现不稳定

解决方案:

# 添加平滑系数与归一化
def enhanced_cosine_sim(vec1, vec2, epsilon=1e-6):
    norm1 = vec1 / (np.linalg.norm(vec1) + epsilon)
    norm2 = vec2 / (np.linalg.norm(vec2) + epsilon)
    return np.dot(norm1, norm2)

五、向量数据库选型:五大维度

选型建议:

初创项目 → Chroma(轻量级本地部署)

企业生产 → Milvus(分布式+高吞吐)

云原生方案 → DashVector(免运维)

六、Chroma实战:从部署到查询

6.1 Docker生产环境部署

# 启动带持久化的Chroma服务
docker run -d \
  --name chromadb \
  -p 8000:8000 \
  -v /data/chroma:/data \
  chromadb/chroma:latest \
  chroma run --path /data

6.2 Python客户端操作全流程

import chromadb
from chromadb.utils.embedding_functions import OpenAIEmbeddingFunction
# 1. 连接服务端
client = chromadb.HttpClient(host="localhost", port=8000)
# 2. 创建集合(使用OpenAI嵌入)
embed_fn = OpenAIEmbeddingFunction(api_key="sk-...", model_name="text-embedding-3-small")
collection = client.get_or_create_collection("tech_docs", embedding_function=embed_fn)
# 3. 批量写入文档
documents = [
    "量子计算利用量子比特实现并行运算",
    "Transformer架构通过自注意力机制提升序列建模能力"
]
metadatas = [{"category": "quantum"}, {"category": "nlp"}]
ids = ["doc1", "doc2"]
collection.add(documents=documents, metadatas=metadatas, ids=ids)
# 4. 混合查询(语义+元数据过滤)
results = collection.query(
    query_texts=["神经网络的最新进展"],
    n_results=2,
    where={"category": "nlp"},  # 元数据过滤
    where_document={"$contains": "架构"}  # 文本内容过滤
)
print(results["documents"][0])

6.3 高级优化技巧

分层索引:对高频数据启用内存缓存

collection = client.get_collection("hot_data", embedding_function=embed_fn, caching=True)

量化压缩:减少75%存储空间

collection.configure(quantization="fp16")  # 半精度浮点

多模态支持:集成CLIP模型处理图像

from chromadb.utils.embedding_functions import OpenCLIPEmbeddingFunction
clip_fn = OpenCLIPEmbeddingFunction()

七、企业级方案:RAG检索精度优化

7.1 三阶段精度提升策略

预处理阶段:

动态分块:Late-Chunking技术解决代词指代问题

元数据增强:添加文档来源/更新时间/置信度标签

检索阶段:

# 重排序提升Top1命中率
from sentence_transformers import CrossEncoder
reranker = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
reranked = reranker.rank(query, candidates)

生成阶段:

提示工程注入检索置信度:

请基于以下内容(可信度{score})回答问题:{context}


注:所有代码测试环境 Python 3.10 + Chroma 0.5.0,需配置OPENAI_API_KEY


luodong

0 评论数