벡터 쿼리 재현율 측정

이 페이지에서는 PostgreSQL용 AlloyDB에서 벡터 쿼리 검색률을 측정하는 방법을 설명합니다. 벡터 검색의 맥락에서 재현율은 색인이 반환하는 실제 최근접 이웃인 벡터의 비율을 나타냅니다. 예를 들어 최근접 이웃 20개에 대한 최근접 이웃 쿼리에서 최근접 이웃 '정답'을 19개 반환하면 재현율은 19/20x100 = 95%가 됩니다.

벡터 쿼리에서 검색에서 검색된 관련 결과의 비율을 측정하기 때문에 재현율이 중요합니다. 재현율은 K-최근접 이웃 (KNN) 검색의 결과와 비교하여 근사 최근접 이웃 (ANN) 검색의 결과 정확성을 평가하는 데 도움이 됩니다.

ANN은 지정된 쿼리 지점과 유사한 데이터 포인트를 찾는 알고리즘으로, 실제 이웃이 아닌 근사 이웃을 찾음으로써 속도를 개선합니다. ANN을 사용하면 속도와 검색의 균형을 맞출 수 있습니다.

KNN은 유사성 측정항목을 기반으로 데이터 세트 내에서 특정 쿼리 벡터와 가장 유사한 'k'개의 벡터를 찾는 알고리즘입니다. k는 쿼리에서 반환할 이웃 수를 나타냅니다.

다음을 비롯한 다양한 벡터 색인에 대해 벡터 검색 쿼리의 검색 결과 검색률을 측정할 수 있습니다.

  • 확장 가능한 최근접 이웃 (ScaNN): 효율적인 벡터 유사성 검색을 위한 알고리즘입니다.
  • Hierarchical Navigable Small World (HNSW): 벡터 데이터베이스에서 효율적인 근사 최근접 이웃 검색에 사용되는 그래프 기반 알고리즘입니다.
  • IVFFLAT (Inverted File with Flat Compression) 및 IVF (Inverted File Flat): 특히 PostgreSQL pgvector 확장 프로그램과 같은 데이터베이스에서 ANN 검색에 사용되는 벡터 색인 유형입니다.

이 페이지에서는 PostgreSQL, AlloyDB, 벡터 검색에 익숙하다고 가정합니다.

시작하기 전에

  1. pgvector 확장 프로그램을 설치하거나 업데이트합니다.

    1. pgvector 확장 프로그램이 설치되어 있지 않으면 vector 확장 프로그램 버전 0.8.0.google-3 이상을 설치하여 생성된 임베딩을 vector 값으로 저장합니다. vector 확장 프로그램에는 pgvector 함수와 연산자가 포함됩니다. Google은 AlloyDB용 최적화로 이 버전의 pgvector를 확장합니다.

      CREATE EXTENSION IF NOT EXISTS vector WITH VERSION '0.8.0.google-3';
      

      자세한 내용은 벡터 저장, 색인 생성, 쿼리를 참고하세요.

    2. pgvector 확장 프로그램이 이미 설치된 경우 vector 확장 프로그램을 버전 0.8.0.google-3 이상으로 업그레이드하여 리콜 평가자 기능을 사용합니다.

      ALTER EXTENSION vector UPDATE TO '0.8.0.google-3';
      
  2. ScaNN 색인을 만들려면 alloydb_scann 확장 프로그램을 설치합니다.

    CREATE EXTENSION IF NOT EXISTS alloydb_scann;
    

벡터 색인에서 벡터 검색어의 재현율 평가

evaluate_query_recall 함수를 사용하여 지정된 구성의 벡터 색인에서 벡터 쿼리의 재현율을 찾을 수 있습니다. 이 함수를 사용하면 원하는 벡터 쿼리 검색 결과를 얻기 위해 매개변수를 조정할 수 있습니다. 재현율은 검색 품질에 사용되는 측정항목으로, 반환된 결과 중 객관적으로 쿼리 벡터에 가장 가까운 결과의 비율로 정의됩니다. evaluate_query_recall 함수는 기본적으로 사용 설정되어 있습니다.

벡터 쿼리의 재현율 찾기

  1. AlloyDB 스튜디오에서 SQL 편집기를 열거나 psql 클라이언트를 엽니다.
  2. ScaNN, HNSW 또는 IVFFLAT 벡터 색인을 만듭니다.

  3. enable_indexscan 플래그가 사용 설정되어 있는지 확인합니다. 플래그가 사용 중지된 경우 색인 스캔이 선택되지 않으며 모든 색인의 검색율은 1입니다.

  4. 쿼리를 매개변수로 사용하고 다음 리콜을 반환하는 evaluate_query_recall 함수를 실행합니다.

    SELECT * FROM evaluate_query_recall( QUERY_TEXT, QUERY_TIME_CONFIGURATIONS, INDEX_METHODS )
    

    이 명령어를 실행하기 전에 다음을 바꿉니다.

    • QUERY_TEXT: $$.
    • QUERY_TIME_CONFIGURATIONS: Optional: the configuration that you can set for the ANN query. This must be in JSON format. The default value is NULL.
    • INDEX_METHODS: Optional: a text array that contains different vector index methods for which you want to calculate the recall. If you set an index method for which a corresponding index doesn't exist, the recall is 1. The input must be a subset of {scann, hnsw, ivf, ivfflat}. If no value is provided, the ScaNN method is used.

      To view differences between query recall and execution time, change the query time parameters for your index.

      The following table lists query time parameters for ScaNN, HNSW, and IVF/IVFFLAT index methods. The parameters are formatted as {"scann.num_leaves_to_search":1, "scann.pre_reordering_num_neighbors":10, "hnsw.ef_search": 1}.

      Index type Parameters
      ScaNN
      • scann.num_leaves_to_search
      • scann.pre_reordering_num_neighbors
      • scann.pct_leaves_to_search
      • scann.num_search_threads
      HNSW
      • hnsw.ef_search
      • hnsw.iterative_scan
      • hnsw.max_scan_tuples
      • hnsw.scan_mem_multiplier
      IVF
      • ivf.probes
      • ivf.iterative_scan
      • ivf.max_probes
      IVFFLAT
      • ivfflat.probes
      • ivfflat.iterative_scan
      • ivfflat.max_probes

      For more information about ScaNN index methods, see AlloyDB ScaNN Index reference. For more information about HNSW and IVF/IVFFLAT index methods, see pgvector.

  5. Optional: You can also add configurations from pg_settings to the QUERY_TIME_CONFIGURATIONS. For example, to run a query with columnar engine scan enabled, add the following config from pg_settings as {"google_columnar_engine.enable_columnar_scan" : on}.

    The configurations are set locally in the function. Adding these configurations doesn't impact the configurations that you set in your session. If you don't set any configurations, AlloyDB uses all of the configurations that you set in the session. You can also set only those configurations that are best suited for your use case.

  6. Optional: To view the default configuration settings, run the SHOW command or view the pg_settings.

  7. Optional: If you have a ScaNN index for which you want to tune the recall, see the tuning parameters in ScaNN index reference.

    The following is an example output, where ann_execution_time is the time that it takes a vector query to execute using index scans. ground_truth_execution_time is the time that it takes the query to run using a sequential scan.

    ann_execution_time and ground_truth_execution_time are different from but directly dependent on Execution time in the query plan. Execution time is the total time to execute the query from the client.

    t=# SELECT * FROM evaluate_query_recall( $$ SELECT id FROM t1 ORDER BY val <=> '[1000,1000,49000]' LIMIT 10 $$, '{"scann.num_leaves_to_search":1, "scann.pre_reordering_num_neighbors":10, "hnsw.ef_search": 1}', ARRAY['scann', 'hnsw']);
    NOTICE:  Recall is 1. This might mean that the vector index is not present on the table or index scan not chosen during query execution.
    id|               query                                               |                                         configurations                                         |  recall |ann_execution_time | ground_truth_execution_time |  index_type
    ----+-------------------------------------------------------------------+------------------------------------------------------------------------------------------------+--------+--------------------+-----------------------------+------------
    1 |  SELECT id FROM t1 ORDER BY val <=> '[1000,1000,49000]' LIMIT 10  | {"scann.num_leaves_to_search":1, "scann.pre_reordering_num_neighbors":10, "hnsw.ef_search": 1} |    0.5 |               4.23 |                     118.211 | scann
    2 |  SELECT id FROM t1 ORDER BY val <=> '[1000,1000,49000]' LIMIT 10  | {"scann.num_leaves_to_search":1, "scann.pre_reordering_num_neighbors":10, "hnsw.ef_search": 1} |      1 |            107.198 |                     118.211 | hnsw
    (2 rows)
    로 묶인 SQL 쿼리
    

    결과가 Recall is 1 (쿼리 리콜이 1)인 경우 테이블에 벡터 색인이 없거나 쿼리 실행 중에 벡터 색인이 선택되지 않았음을 나타낼 수 있습니다. 이 상황은 테이블에 벡터 색인이 없거나 플래너가 벡터 색인 스캔을 선택하지 않는 경우에 발생합니다.

    쿼리가 select id, name from table order by embedding <->'[1,2,3]' LIMIT 10;.이고 열 이름의 예상 값이 NULL인 경우 쿼리를 다음 중 하나로 변경합니다.

    select id, COALESCE(name, 'NULL') as name from table order by embedding <-> '[1,2,3]' LIMIT 10;
    

    또는

    select id from table order by embedding <-> '[1,2,3]' LIMIT 10;
    

다음 단계