创建向量索引

背景信息

云原生数据仓库AnalyticDB PostgreSQL版向量数据库中的FastANN向量检索引擎实现了主流的HNSW（Hierarchical Small World Graph）算法，它基于PostgreSQL中的段页式存储实现，并且在索引中只存储了指向表中向量列的指针，极大地减少了向量索引的存储空间。同时FastANN向量检索引擎也支持PQ（Product Quantization）量化功能，可以对高维向量进行降维。通过在索引中存储降维后的向量达到减少在向量插入和查询时的回表操作，从而提升了向量检索的性能。

语法

CREATE INDEX [INDEX_NAME]
ON [SCHEMA_NAME].[TABLE_NAME]   
USING ANN(COLUMN_NAME) 
WITH (DIM=<DIMENSION>,
      DISTANCEMEASURE=<MEASURE>,
      HNSW_M=<M>,
      HNSW_EF_CONSTRUCTION=<EF_CONSTURCTION>,
      PQ_ENABLE=<PQ_ENABLE>,
      PQ_SEGMENTS=<PQ_SEGMENTS>,
      PQ_CENTERS=<PQ_CENTERS>,
      EXTERNAL_STORAGE=<EXTERNAL_STORAGE>);

各字段说明：

• INDEX_NAME：索引名。

• SCHEMA_NAME：模式（命名空间）名。

• TABLE_NAME：表名。

• COLUMN_NAME：向量索引列名。

• 其他向量索引参数：

  向量索引参数	是否必填	说明	默认值	取值范围
  DIM	是	向量维度。该参数主要用于向量插入时的检查，当维度不匹配的时候，系统将提示相关错误信息。	0	[1, 8192]
  DISTANCEMEASURE	否	支持的相似度距离度量算法： L2：使用欧氏距离（平方）函数构建索引，通常适用于图片相似度检索场景。 计算公式： IP：使用反内积距离函数构建索引，通常适用于向量归一化之后替代余弦相似度。 计算公式： COSINE：使用余弦距离函数构建索引，通常适用于文本相似度检索场景。 计算公式： 说明 IP和COSINE要求内核版本需大于6.3.10.18及以上，使用时请确保内核版本满足要求。查看及升级内核版本请参见版本升级。 建议对向量数据进行归一化处理之后再入库，然后统一使用IP距离为索引的相似度距离度量算法，以获得最佳的性能。 使用IP距离作为索引的相似度距离度量算法时，也可以在查询时直接获得欧氏距离和余弦相似度，具体使用方法见向量检索。 更多关于向量检索的UDF，请参见相关参考。	L2	(L2, IP, COSINE)
  HNSW_M	否	HNSW算法中的最大邻居数。	50	[1, 1000]
  HNSW_EF_CONSTRUCTION	否	HNSW算法构建索引时的候选集大小。	600	[1, 4000]
  PQ_ENABLE	否	是否开启PQ向量降维的功能。PQ向量降维依赖于存量的向量样本数据进行训练，如果数据量小于50w时，不建议设置此参数。	0	[0, 1]
  PQ_SEGMENTS	否	PQ_ENABLE为1时生效。用于指定PQ向量降维过程中使用的kmeans聚类算法中的分段数。 如果DIM能被8整除，则不需要填写，否则需要手动设置。 PQ_SEGMENTS必须大于0，且DIM必须能被PQ_SEGMENTS整除。	0	[1, 256]
  PQ_CENTERS	否	PQ_ENABLE为1时生效。用于指定PQ向量降维过程中使用的kmeans聚类算法中的中心点数。	2048	[256, 8192]
  EXTERNAL_STORAGE	否	是否使用mmap构建HNSW索引。 为0时，默认会采用段页式存储构建索引，这种模式可以使用PostgreSQL中的shared_buffer做缓存，支持删除和更新等操作。 为1时，该索引会采用mmap构建索引，该模式不支持删除、更新等操作。	0	[0, 1]

示例

假设有一个文本知识库，将文章分割成chunk后，转换为embedding向量，最后存入数据库中，其中切割生成的chunks表包含以下字段：

1. 创建存储向量的数据库表。

   CREATE TABLE chunks (
       id SERIAL PRIMARY KEY,
       chunk VARCHAR(1024),
       intime TIMESTAMP,
       url VARCHAR(1024),
       feature REAL[]
   ) DISTRIBUTED BY (id);

2. 将向量列的存储模式设置为PLAIN，以降低数据行扫描成本，获得更好的性能。

   ALTER TABLE chunks ALTER COLUMN feature SET STORAGE PLAIN;

3. 对向量列建立向量索引。

   -- 创建欧氏距离度量的向量索引。
   CREATE INDEX idx_chunks_feature_l2 ON chunks USING ann(feature) WITH (dim=1536, distancemeasure=l2, hnsw_m=64, pq_enable=1);
   -- 创建内积距离度量的向量索引。
   CREATE INDEX idx_chunks_feature_ip ON chunks USING ann(feature) WITH (dim=1536, distancemeasure=ip, hnsw_m=64, pq_enable=1);
   -- 创建余弦相似度度量的向量索引。
   CREATE INDEX idx_chunks_feature_cosine ON chunks USING ann(feature) WITH (dim=1536, distancemeasure=cosine, hnsw_m=64, pq_enable=1);

   说明

   • 目前一个向量表可以创建多个向量列，每个向量列又可以创建多个向量索引，具体根据实际需要创建对应的向量索引，避免创建无效的索引。

   • 向量查询的方式必须与向量索引的距离度量对应。例如，向量查询中的符号<->对应欧氏距离度量的向量索引，<#>对应内积距离度量的向量索引，<=>对应余弦相似度度量的向量索引。如果没有对应的向量索引，则向量查询会退化为精确搜索（即暴力搜索）。

   • 请使用FastANN向量检索插件提供的ANN访问方法，当实例开启向量检索引擎优化功能时，会自动在该实例中创建FastANN向量检索插件。

   • 若报错You must specify an operator class or define a default operator class for the data type，表示该实例未开启向量检索引擎优化功能，请开启后重试，具体操作请参见开启或关闭向量检索引擎优化。

4. 为常用的结构化列建立索引，提升融合查询速度。

   CREATE INDEX ON chunks(intime);

向量索引的构建方式

向量索引主要分为两种构建方式：

• 流式构建。

  即先创建一个空表，并在空表上建立向量索引，那么在进行向量数据导入时，就会进行流式实时的索引构建。此方式适用于实时向量检索场景，但是会导致数据导入速度过慢。

• 异步构建。

  即在创建一个空表后，在不建向量索引的情况下，先将数据导入，然后再对存量的向量数据构建向量索引。此方式适用于大规模向量数据导入的场景。

例如，在DMS上异步构建索引可以采用如下方式：

-- 假设在8C32GB的实例规格上进行操作。
-- 设置statement_timeout和idle_in_transaction_session_timeout都为0，确保不触发超时中断。
-- 设置fastann.build_parallel_processes为8，确保计算节点的CPU可以用满，这里需要根据segment的不同的规格需要设置不同的值，如4C16GB的规格设置为4，8C32GB的规格设置为8，16C64GB规格设置为16。
-- 在DMS上，参数设置必须和索引创建的SQL在同一行，否则不能生效。
SET statement_timeout = 0;SET idle_in_transaction_session_timeout = 0;SET fastann.build_parallel_processes = 8;CREATE INDEX ON chunks USING ann(feature) 
WITH (dim=1536, distancemeasure=cosine, hnsw_m=64, pq_enable=1);

使用psql进行异步构建索引时可以采用如下方式：

-- 假定和上面DMS操作的实例一样，也是8C32GB的实例上进行操作。

-- 第一步：将以下内容放到文件create_index.sql中。
SET statement_timeout = 0;
SET idle_in_transaction_session_timeout = 0;
SET fastann.build_parallel_processes = 8;
CREATE INDEX ON chunks USING ann(feature) WITH (dim=1536, distancemeasure=cosine, hnsw_m=64, pq_enable=1);

-- 第二步：执行如下命令开始并行构建索引。
psql -h gp-xxxx-master.gpdb.rds.aliyuncs.com -p 5432 -d dbname -U username -f create_index.sql

相关参考

-- 欧氏距离
SELECT l2_distance(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 内积距离
SELECT inner_product_distance(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 余弦相似度
SELECT cosine_similarity(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 点积距离
SELECT dp_distance(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 汉明距离
SELECT hm_distance(array[1,0,1,0]::int[], array[0,1,0,1]::int[]);
-- 向量数组加法
SELECT vector_add(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 向量数组减法
SELECT vector_sub(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 向量数组乘法
SELECT vector_mul(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 向量数组除法
SELECT vector_div(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);
-- 向量数组累加值
SELECT vector_sum(array[1,1,1,1]::real[]);
-- 向量数组最小值
SELECT vector_min(array[1,1,1,1]::real[]);
-- 向量数组最大值
SELECT vector_max(array[1,1,1,1]::real[]);
-- 向量数组平均值
SELECT vector_avg(array[1,1,1,1]::real[]);
-- 向量数组模长
SELECT vector_norm(array[1,1,1,1]::real[]);
-- 两个向量的夹角
SELECT vector_angle(array[1,1,1,1]::real[], array[2,2,2,2]::real[]);

-- 欧氏距离
SELECT array[1,1,1,1]::real[] <-> array[2,2,2,2]::real[] AS score;
-- 内积距离
SELECT array[1,1,1,1]::real[] <#> array[2,2,2,2]::real[] AS score;
-- 余弦距离
SELECT array[1,1,1,1]::real[] <=> array[2,2,2,2]::real[] AS score;

-- 加法
SELECT array[1,1,1,1]::real[] + array[2,2,2,2]::real[] AS value;
-- 减法
SELECT array[1,1,1,1]::real[] - array[2,2,2,2]::real[] AS value;
-- 乘法
SELECT array[1,1,1,1]::real[] * array[2,2,2,2]::real[] AS value;
-- 除法
SELECT array[1,1,1,1]::real[] / array[2,2,2,2]::real[] AS value;