开源模子应用落地-工具使用篇-向量数据库（三）

一、媒介

    通过学习"开源模子应用落地"系列文章，我们成功地创建了一个完备可实行的AI交付流程。如今，我们要引入向量数据库，作为我们AI服务的二级缓存。本文将具体介绍怎样使用Milvus Lite来为我们的AI服务摆设一个前置缓存。

---

二、术语

2.1、向量数据库

    向量数据库是一种专门用于存储和处理高维向量数据的数据库系统。与传统的关系型数据库或文档数据库不同，向量数据库的设计目的是高效地支持向量数据的索引和相似性搜索。
    在传统数据库中，数据通常是以结构化的表格情势存储，每个记录都有预界说的字段。但是，对于包含大量高维向量的数据，如图像、音频、文本等，传统的数据库模子往往无法有效地处理。向量数据库通过引入特定的数据结构和索引算法，答应高效地存储和查询向量数据。
    向量数据库的核心概念是向量索引。它使用一种称为向量空间模子的方法，将向量映射到多维空间中的点，并利用这种映射关系构建索引结构。如许，当需要搜索相似向量时，可以通过盘算向量之间的间隔或相似度来快速定位相似的向量。
2.2、向量数据库的使用场景

     向量数据库在很多范畴中都有广泛的应用场景，特别是涉及到高维向量数据存储和相似性搜索的使命。以下是一些常见的使用场景：

• 目的识别和图像搜索：向量数据库可用于存储图像特性向量，以支持快速的相似图像搜索和目的识别。它在图像搜索引擎、人脸识别和视频监控等范畴具有重要作用。
  
• 保举系统：向量数据库可以存储用户和物品的特性向量，用于个性化保举。基于相似性搜索，可以找到与用户兴趣相似的物品，提供个性化的保举结果。
  
• 自然语言处理：在文本处理使命中，可以使用向量数据库存储文本向量，如词向量、句向量等。基于相似性搜索，可以进行文本匹配、语义相似度盘算等操作。
  
• 数据聚类和分类：向量数据库可用于高维向量数据的聚类和分类分析。它可以帮助发现数据会合的聚类模式和类别，用于数据发掘和机器学习使命。
  
• 检索与保举系统：在电子商务和商品搜索中，向量数据库可以存储商品特性向量，以支持相似商品的搜索和保举。它可以提供更准确和个性化的搜索结果。
  
• 医疗和生物信息学：向量数据库可用于存储基因表达向量、蛋白质特性向量等生物信息学数据。它可以在基因组学、药物研发等范畴中帮助进行数据分析和研究。
  
• 视频内容分析：向量数据库可用于存储视频特性向量，如视频帧特性、视频片断特性等。它可以用于视频内容搜索、视频剪辑和视频保举等应用。
  

2.3、向量相似度检索

    相似度检索是指将目的对象与数据库中数据进行比对，并召回最相似的结果。同理，向量相似度检索返回的是最相似的向量数据。
2.4、向量相似度检索算法

• 余弦相似度（Cosine Similarity）：余弦相似度是一种常用的衡量向量相似性的方法。它通过盘算两个向量之间的夹角余弦值来度量它们的相似水平。余弦相似度范围在[-1, 1]之间，值越接近1表现两个向量越相似。
  
• 欧氏间隔（Euclidean Distance）：欧氏间隔是盘算向量之间间隔的一种常见方法。它衡量了两个向量之间的多少间隔，即两个向量之间的直线间隔。欧氏间隔越小表现两个向量越相似。
  
• 曼哈顿间隔（Manhattan Distance）：曼哈顿间隔是盘算向量之间间隔的一种度量方式。它衡量了两个向量之间的城市街区间隔，即通过水平和垂直方向移动所需的步数之和。
  
• Jaccard相似度（Jaccard Similarity）：Jaccard相似度通常用于盘算集合之间的相似性，但也可以应用于特性向量的相似性盘算。它通过盘算两个向量的交集与并集之间的比值来度量它们的相似水平。
  
• 汉明间隔（Hamming Distance）：汉明间隔通常用于盘算两个等长字符串之间的间隔，但也可应用于二进制向量的相似性盘算。它衡量了两个向量之间在相应位置上不同位的数目。
  
• 近来邻搜索算法（Nearest Neighbor Search）：近来邻搜索算法通过盘算向量之间的相似度或间隔，找到与目的向量最相似的邻居向量。常用的近来邻搜索算法包罗暴力搜索、KD树、球树、LSH（局部敏感哈希）等。
  

2.5、Milvus

    是一个开源的向量数据库引擎，专门用于存储和处理大规模高维向量数据。它提供了高效的向量索引和相似性搜索功能，使用户可以大概快速地进行向量数据的存储、查询和分析。
    Milvus的设计目的是为了满意现代应用中对大规模向量数据的需求，例如人脸识别、图像搜索、保举系统等。它接纳了向量空间模子和多种索引算法，包罗倒排索引、近似近来邻（Approximate Nearest Neighbor，ANN）等，以支持高效的相似性搜索。
    Milvus提供了易于使用的编程接口和丰富的功能，使用户可以方便地插入、查询和分析向量数据。它支持多种数据范例的向量，包罗浮点型、整型等，也支持多种间隔度量方法，如欧氏间隔、余弦相似度等。
    Milvus还提供了分布式摆设和横向扩展的本领，可以在多台机器上构建高可用性和高性能的向量数据库集群。它支持数据的分片和负载均衡，可以处理大规模数据集和高并发查询。
2.6、Milvus Lite

    是Milvus向量数据库的一个轻量级版本。旨在提供在资源受限的情况中快速、高效地进行向量存储和相似性搜索的本领。
    与完备版的Milvus相比，它具有以下特点：

• 轻量级：Milvus Lite具有较小的存储占用和内存斲丧，得当在资源受限的装备上摆设和运行。
  
• 快速摆设：Milvus Lite提供了简化的摆设和设置过程，使其更易于在嵌入式装备和边沿服务器上进行摆设和集成。
  
• 高效的向量索引和搜索：只管是轻量级版本，Milvus Lite仍然提供了高效的向量索引和相似性搜索功能，以支持快速的向量数据查询。
  
• 离线模式：Milvus Lite支持在离线模式下进行向量索引和搜索，无需实时连接到远程服务器。
  

2.7、Attu

    是Milvus 的高效开源管理工具。 它具有直观的图形用户界面（GUI），使您可以轻松地与数据库进行交互。
2.8、归一化

    是一种数据处理技术，用于将不同标准或范围的数据转换为同一的标准范围，通常是0到1之间大概是-1到1之间。它是数据预处理的常见步骤之一，旨在消除不同特性之间的标准差异，以便更好地比力和分析数据。
2.9、标准化

    是一种数据处理技术，用于将数据转换为具有零均值和单位方差的标准分布。它是数据预处理的一种常见方法，旨在消除不同特性之间的标准差异，使得数据更得当进行比力和分析。

---

三、使用方式

3.1、架构示意图

这里的Milvus Lite摆设在内网，位于业务服务和AI服务的中间，作为AI服务的二级缓存（一级缓存为Redis），为AI服务减缓负载压力。

3.2、安装Milvus Lite

     1. 创建虚拟情况
         conda create --name milvus python=3.10
     2. 激活虚拟情况
        conda activate milvus
     3.  安装milvus包
        pip install milvus

3.3、编写Milvus Lite服务端代码

   此处将Milvus Lite作为 Python 模块启动

1. from milvus import default_server
3. def start_server():
4.     default_server.start()
6. def stop_server():
7.     default_server.stop()
9. if __name__ == '__main__':
10.     with default_server:
11.         start_server()
13.         input("按下任意键继续...")

复制代码

3.4、启动Milvus Lite服务

    python -u 上述代码的文件名
    

    启动完成后，会监听19530端口
3.5、安装Milvus客户端管理工具

    下载地址：Releases · zilliztech/attu · GitHub
3.6、登录Milvus Lite服务端

    留意根据现实情况调整IP和端口，默认端口为19530

登录进去，就能看到如下信息：

    
四、业务整合

业务数据需要提前初始化到向量数据库中
4.1、连接milvus服务端

1. # 1. 连接向量数据库 Milvus
2. def connect_db(host):
3.     logging.info("start connecting to Milvus")
5.     # Milvus Lite has already started, use default_server here.
6.     connections.connect(host=host, port=19530,user='root',password='123456',)
8.     logging.info("connected to Milvus")
9.    

复制代码

4.2、创建集合

1. dim = 256
2. alias = 'default'
3. nlist = 65536
5. # 2. 创建数据集合 collection
6. def create_collection(collection_name):
7.     isExist = has_collection(collection_name=collection_name, using=alias)
9.     if isExist:
10.         print(f'集合{collection_name}已经存在')
11.         return
13.     fields = [
14.         FieldSchema(name="id", dtype=DataType.INT64, is_primary=True, auto_id=True),
15.         FieldSchema(name="keyword", dtype=DataType.FLOAT_VECTOR, dim=dim),
16.         FieldSchema(name="content", dtype=DataType.VARCHAR, max_length=4096)
17.     ]
18.     schema = CollectionSchema(fields, "the collection of tb_test")
20.     logging.info(f"Create collection {collection_name}")
21.     tb_test = Collection(collection_name, schema, consistency_level="Strong")
22.     return tb_test

复制代码

PS：keyword的向量维度是256，高维向量可以提供更丰富的信息表现本领，可以大概捕获更多的特性和关系，从而提高模子的表达本领。
创建成功后：

4.3、插入数据

1. # 3. 插入数据实体 entities
2. def insert_data(collection_name,narray,content,isNormalize = True):
3.     res = None
4.     try:
5.         # 获得已存在集合对象
6.         collection = Collection(name=collection_name, using=alias)
7.         logging.info("Start inserting entities")
9.         if isNormalize == True:
10.             # 标准化处理
11.             normalize_narray = pretreatment(narray)
12.             print(normalize_narray)
13.             # 补齐向量长度
14.             padded_vector = narray_pad(normalize_narray, dim)
15.             print(padded_vector)
16.         else:
17.             # 补齐向量长度
18.             padded_vector = narray_pad(narray, dim)
20.         # 打印补齐后的向量
21.         logging.info(padded_vector.shape)
23.         entities = [
24.             padded_vector,  # title
25.             [content],  # content
26.         ]
28.         res = collection.insert(entities)
29.         collection.flush()
30.     finally:
31.         return res
36. # 插入数据
37. narray = np.array([100220.0,102247.0,31905.0,40814.0,101009.0, 87335.0,8863.0,20.0,15.0,15.0,104745.0,100354.0,43815.0,103010.0,102233.0,100351.0,102482.0])
38. content = "标题：青春之光青春是一首歌，悠扬而激昂；青春是一幅画，斑斓而生动；青春是一部电影，感人至深。而在我心中，青春更是那个在奥运赛场上奔跑的刘翔。记得那是2004年的雅典奥运会，刘翔以12.91秒的成绩打破了世界纪录，成为中国田径历史上第一位获得奥运金牌的运动员。那一刻，我被他的坚韧和毅力深深打动，也深深地感受到了青春的力量。刘翔的青春，充满了挑战和奋斗。他曾经因为伤病困扰，一度想要放弃，但他没有。他知道，只有坚持下去，才能实现自己的梦想。于是，他在痛苦中挣扎，用汗水和泪水浇灌着自己的青春。终于，他成功了，他站在了奥运的最高领奖台上，成为了全中国的骄傲。刘翔的青春，充满了激情和活力。他是中国田径的一颗璀璨明星，他的每一次起跑都充满力量，他的每一次跨栏都充满速度。他的青春，就像一道闪电，照亮了整个赛场，也照亮了我们的心灵。刘翔的青春，充满了希望和梦想。他的梦想是成为最好的自己，他的希望是为中国赢得更多的荣誉。他的青春，就像一盏明灯，指引着他前进的方向，也激励着我们去追求自己的梦想。青春，就是要有梦想，有希望，有勇气去追逐。刘翔的青春，就是这样，充满了梦想、希望和勇气。他的青春，是我们所有人的青春，是我们所有人追求梦想的动力。青春，是一场无悔的旅程，无论前方有多少困难和挫折，只要我们有梦想，有希望，有勇气，就一定能够到达我们的目的地。让我们一起，像刘翔一样，用自己的青春，去创造属于我们自己的辉煌！"
39. # 调用方法
40. insert_data(collection_name,narray,content,True)

复制代码

插入成功后：

PS：受限于本篇内容较长，怎样获取文本向量以及怎样进行标准化或归一化处理，将在别的的文章中说明。
4.4、创建索引

1. # 4. 创建索引 index
2. def create_index(collection_name,index_column,index_name):
3.     try:
4.         collection = Collection(name=collection_name, using=alias)
5.         # 给向量字段构建索引，并指定索引类型，以及相似度度量方式
6.         # nlist 表示簇的个数，该参数可以将向量划分成多个区域，有利于加快搜索
7.         index_params = {
8.             "metric_type": "IP",
9.             "index_type": "IVF_FLAT",
10.             "params": {"nlist": nlist}
11.         }
12.         collection.create_index(field_name=index_column, index_name=index_name, index_params=index_params)
13.     finally:
14.         if collection:
15.             collection.release()
18. #  创建索引 index
19. collection_name = "tb_test"
20. index_column='keyword'
21. index_name='idx_keyword'
22. # 调用
23. create_index(collection_name, index_column, index_name)

复制代码

创建成功后：

4.5、加载数据至缓存

1. def load_in_memory(collection_name):
2.     collection = Collection(name=collection_name, using=alias)
3.     # 将整个 collection 加载到内存中，也可以只加载某个 Partition
4.     collection.load()
6.     return collection
9. # 加载集合到缓存
10. collection_name = "tb_test"
11. # 调用
12. load_in_memory(collection_name)
13.    

复制代码

---

五、附带说明

5.1、目前市面上成熟的向量数据库产物有很多，联合现实效果，这里选择Milvus作为项目解决方案。
5.2、其他Milvus Lite启动方式
    # 通过 CLI 来启动 Milvus Lite，实行命令：milvus-server
    # 通过 CLI 以调试模式来启动 Milvus Lite，实行命令：milvus-server  --debug
5.3、milvus_cli管理工具
    # 安装
      pip install milvus-cli
   # 登录
     milvus_cli
   # 连接数据库
     connect -uri http://127.0.0.1:19530
5.4、milvus_cli常用命令

1. 创建数据库
2.         create database -db test
3.                        
4. 使用数据库
5.         use database -db test
6.        
7. 查看数据库
8.         list databases
10. 删除数据库
11.         delete database -db test
12.        
13. 创建collection
14.         create collection -c car -f id:INT64:primary_field
15.                 -f vector:FLOAT_VECTOR:128
16.                 -f color:INT64:color
17.                 -f brand:INT64:brand
18.                 -p id -a
19.                 -level Strong
21.         Options:
22.           -c, --collection-name TEXT              Collection name to specify alias.
23.           -p, --schema-primary-field TEXT        Primary field name.
24.           -a, --schema-auto-id                    [Optional, Flag] - Enable auto id.
25.           -desc, --schema-description TEXT        [Optional] - Description details.
26.           -d, --is-dynamic TEXT                   [Optional] - Collection schema supportsdynamic fields or not.
27.           -level, --consistency-level TEXT         [Optional] - Consistency level:Bounded,Session,Strong, Eventual .
28.           -f, --schema-field TEXT                 [Multiple] - FieldSchema. Usage is "<Name>:<DataType>:<Dim(if vector) or Description>"
29.           -s, --shards-num INTEGER                [Optional] - Shards number
30.        
31. 查看collection
32.         list collections
34. 删除collection
35.         delete collection -c tb_test
36.        
37. 创建index
38.         create index
39.         Collection name (tb_test): tb_test
40.         The name of the field to create an index for (vector): vector
41.         Index name: vectorIndex
42.         Default is ''
43.         Index type (FLAT, IVF_FLAT, IVF_SQ8, IVF_PQ, RNSG, HNSW, ANNOY, AUTOINDEX, DISKANN, ) []:  IVF_FLAT  
44.         Default is ''
45.         Index metric type (L2, IP, HAMMING, TANIMOTO,): L2
46.         Index params nlist: 2
48. 查看index
49.         list indexes -c tb_test
51. 查看partitions
52.         list partitions -c tb_test
53.        
54. 删除partitions
55.         delete partition -c tb_test -p new_partition
57. 删除entities
58.         delete entities -c tb_test
59.        
60. 加载
61.         load collection -c tb_test
62.         Options:
63.           -c, --collection TEXT  The name of collection to load.
64.           -p, --partition TEXT   [Optional, Multiple] - The name of partition to load.
65.           --help                 Show this message and exit.
67. 释放
68.         release collection -c tb_test
69.         Options:
70.           -c, --collection TEXT  The name of collection to load.
71.           -p, --partition TEXT   [Optional, Multiple] - The name of partition to load.
72.           --help  
74. show
75.         show collection -c tb_test
76.         show index -c tb_test
77.         show index_progress -c tb_test
78.         show loading_progress -c tb_test
79.         show partition -c tb_test
80.        
81. 创建别名
82.         create alias -c tb_test -A -a tb_test11
84.         Options:
85.           -c, --collection-name TEXT  Collection name to be specified alias.
86.           -a, --alias-name TEXT       The alias of the collection.
87.           -A, --alter                 [Optional, Flag] - Change an existing alias to
88.                                                                   current collection.
89.        
90. 创建用户
91.         create user -u root -p 123456
92.        
93.         -u, --username TEXT  The username of milvus user.
94.         -p, --password TEXT  The password of milvus user.
96. 查看用户
97.         list users
98.        
99.        
100. 查看版本
101.         version
102.        

复制代码

5.5、高维向量表现的优劣
优势：

• 表现本领加强：高维向量可以提供更丰富的信息表现本领，可以大概捕获更多的特性和关系，从而提高模子的表达本领。
  
• 解决冗余信息：在高维空间中，冗余特性大概会被稀疏化，使得模子更轻易识别和利用有效的特性。
  
• 处理复杂标题：某些复杂标题大概需要更高维度的向量来表现，以便更好地捕获标题的复杂性和多样性。
  

劣势

• 维度灾难：高维度数据大概导致维度灾难标题，即数据稀疏性增加，对于有限的训练数据而言，模子的泛化本领大概会受到影响。
  
• 盘算复杂性增加：高维度数据需要更多的盘算资源和时间来处理和分析，大概会增加盘算的复杂性和开销。
  
• 数据稀疏性：在高维空间中，数据点之间的间隔变得更远，大概会导致数据稀疏性增加，从而影响模子的准确性和可靠性。
  

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。