跳到主要内容

使用 mmap
公测版

Mmap 允许在不将磁盘上的文件加载到内存的情况下通过内存访问这些文件。通过配置 mmap,Zilliz Cloud 可以根据访问频次的不同将索引和数据分别存放到内存或磁盘上,不仅优化了数据加载行为,扩大了 Collection 的容量,也不会给搜索性能带来负面影响。本文将帮助您理解 Zilliz Cloud 如何利用 mmap 实现快速高效的数据存储和检索能力及使用该能力需要注意的相关事项。

📘说明

该特性当前处于公测阶段。如果您在使用过程中遇到任何问题,可以联系Zilliz Cloud 支持.

概述

Zilliz Cloud 使用 Collection 来组织向量数据及其元数据,并将它们按照列的形式组织成一张表格。表格中的每一行就代表一个 Entity。如下图左侧所示,在一个 Entity 中,向量列用来存放向量数据,而标量列用来存放与该向量相关的元数据信息。当您在向量列上创建索引并加载 Collection 后,Zilliz Cloud 就会将 Collection 中所有标量列中存放的原始数据及您在向量列上创建的索引文件加载到内存。

TH6ow6TUghJK0WbjLnqc6r74nRh

Zilliz Cloud 是一款内存密集型的数据库系统。可用内存的大小决定了 Collection 的容量。如果加载的列中存放的数据大于内存大小时,可能会遇到加载失败的情况。而加载大体积数据这种情况在 AI 驱动的应用中是比较常见的情况。

为了解决此类矛盾,Zilliz Cloud 在 Collection 中默认启用了 mmap 能力。如上图右侧所示,当您在 Zilliz Cloud 中加载 Collection 时,Zilliz Cloud 会使用 mmap 将所有列的原始数据映射到内存并加载向量列的索引文件到内存。

通过对比上图左右两侧的数据加载情况,您应该可以发现在未启用 mmap 能力时,内存压力较大,当需要加载的数据大于内存大小时,就会导致加载失败。而在启用 mmap 能力后,所有字段的原始数据都以映射的方式加载到内存。用户通过内存上的指针访问的仍旧是存放在磁盘上的数据,大大节约了内存资源的使用。

全局 mmap 策略

下表罗列了使用不同配置的集群对应的全局 mmap 策略。

Mmap 对象

Dedicated 集群

Free & Serverless 集群

性能型 CU

容量型 CU

标量字段原始数据

默认关闭,可修改

默认开启,可修改

默认开启,不可修改

标量字段索引

默认关闭,可修改

默认开启,可修改

默认开启,不可修改

向量字段原始数据

默认开启,可修改

默认开启,可修改

默认开启,不可修改

向量字段索引

默认关闭,不可修改

默认关闭,不可修改

默认开启,不可修改

对于使用性能型 CU 的 Dedicated 集群而言, Zilliz Cloud 只在向量字段原始数据上启用了 mmap,并在加载 Collection 时将标量字段的原始数据及所有字段的索引都加载到内存。建议您保持全局配置,确保在搜索和查询过程中元数据过滤和检索的性能。另外,您可以考虑为不参与元数据过滤的标量字段开启 mmap,以减少内存开销,扩大 Collection 容量。

对于使用容量型 CU 的 Dedicated 集群而言,Zilliz Cloud 只在向量字段索引上关闭了 mmap 以保证索引性能,并在加载 Collection 时将标量字段索引和所有字段的原始数据都通过 mmap 的方式转存在磁盘上,从而保证了 Collection 容量的最大化。另外,您可以考虑在参与元数据过滤的标量字段和在搜索和查询请求的输出字段列表中引用的原始数据量过大的标量字段上关闭 mmap,以提升搜索和查询的响应速度,减少网络抖动,提升查询性能。

在 Collection 中设置 mmap

在修改 mmap 设置前,需要释放 Collection。修改完成后,需要再次加载 Collection。您可以为某个字段、某个字段的索引或某个 Collection 分别设置 mmap 策略

📘说明

请谨慎修改 mmap 设置。不恰当的 mmap 策略可能会导致如下问题:

  • 对于使用性能型 CU 的 Dedicated 集群而言,在加载 Collection 时,所有标量字段的原始数据和向量字段索引默认会被加载到内存以保证在搜索和查询中对标量字段的快速访问。修改默认设置可能会导致性能下降。

  • 对于使用容量型 CU 的 Dedicated 集群而言,在加载 Collection时,只有向量字段索引被默认加载到内存以保障 Collection 容量的最大化。修改默认设置可能会因 Collection 缩小而出现内存不足(OOM)问题。

在字段上配置 mmap 策略

如果您的集群使用的是小规格的性能型 CU,但是某个标量字段的原始数据体积较大,可以考虑在这个标量字段上开启 mmap。

如下示例假设您连接到了一个使用性能型 CU 的 Dedicated 集群,并演示了如何在名为 doc_chunk 的 VARCHAR 字段上启用 mmap。

from pymilvus import MilvusClient

CLUSTER_ENDPOINT="YOUR_CLUSTER_ENDPOINT"
TOKEN="YOUR_TOKEN"

client = MilvusClient(
uri=CLUSTER_ENDPOINT,
token=TOKEN
)

schema = MilvusClient.create_schema()

# Disable mmap on a field upon creating the schema for a collection
schema.add_field(
field_name="doc_chunk",
datatype=DataType.INT64,
max_length=512,
# highlight-next-line
mmap_enabled=False,
)

# Disable mmap on an existing field
# The following assumes that you have a collection named `my_collection`
client.alter_collection_field(
collection="my_collection",
field_name="doc_chunk",
properties={"mmap.enable": True}
)

当加载使用上述 Schema 的 Collection 时,Zilliz Cloud 会将 doc_field 字段的原始数据转移到硬盘上。对于修改已有字段的 mmap 设置前,需要先释放该字段所在的 Collection,并在修改完成后再加载该 Collection。

在标量索引上配置 mmap 策略

对于参与元数据过滤或在搜索或查询请求的输入字段列表中引用的标量字段,可以考虑在加载 Collection 时将这些字段加载到内存。

如下示例假设您连接到一个使用了容量型 CU 的 Dedicated 集群,并演示了如果在名为 title 的 VARCHAR 字段的索引上关闭 mmap 来加速对该字段的访问。

# Add a varchar field
schema.add_field(
field_name="title",
datatype=DataType.VARCHAR,
max_length=512
)

index_params = MilvusClient.prepare_index_params()

# Create index on the varchar field with mmap settings
index_params.add_index(
field_name="title",
index_type="INVERTED"
# highlight-next-line
params={ "mmap.enabled": "false" }
)

# Change mmap settings for an index
# The following assumes that you have a collection named `my_collection`
client.alter_index_properties(
collection_name="my_collection",
index_name="title",
properties={"mmap.enabled": True}
)

在加载使用上述索引参数的 Collection 时,Zilliz Cloud 会将 title 字段的索引加载到内存。对于修改已有字段索引的 mmap 设置前,需要先释放该字段所在的 Collection,并在修改完成后再加载该 Collection。