Tablestore SDK for Java を使用して、近似最近傍探索のための KNN ベクタークエリを実行する - Tablestore

k-最近傍 (KNN) ベクタークエリ機能により、ベクターに基づく近似最近傍探索が可能になります。これにより、大規模なデータセット内でクエリ対象のベクターに最も類似したデータ項目を見つけることができます。

前提条件

Tablestore クライアントが初期化されている。
データテーブルに対して多次元インデックスが作成され、ベクターフィールドが指定されている。

使用上の注意

Tablestore SDK for Java V5.17.0 以降は、KNN ベクタークエリ機能をサポートしています。正しいバージョンの Tablestore SDK for Java がインストールされていることを確認してください。
説明
Tablestore SDK for Java のバージョン履歴に関する詳細については、「Tablestore SDK for Java のバージョン履歴」をご参照ください。
ベクターフィールドの数とベクターフィールドのディメンション数には制限があります。詳細については、「多次元インデックスの制限」をご参照ください。
多次元インデックスサーバーには複数のパーティションがあります。多次元インデックスサーバーの各パーティションは、クエリしたいベクターに最も近い上位 K 個の近傍を返します。パーティションによって返された上位 K 個の最近傍は、クライアントノードで集約されます。トークンを使用してすべてのデータをページ単位でクエリする場合、応答の合計行数は多次元インデックスサーバーのパーティション数に関連します。

パラメーター

パラメーター	必須	説明
fieldName	はい	ベクターフィールドの名前。
topK	はい	クエリしたいベクターと最も類似度が高い上位 K 件のクエリ結果。topK パラメーターの最大値に関する詳細については、「多次元インデックスの制限」をご参照ください。重要 K の値が大きいほど、取得率、クエリの待機時間、およびコストが高くなります。
float32QueryVector	はい	類似度をクエリしたいベクター。
filter	いいえ	フィルター。KNN ベクタークエリ条件ではないクエリ条件の組み合わせを使用できます。

例

次のサンプルコードは、テーブル内で指定されたベクターと最も類似度が高い上位 10 個のベクターをクエリする方法の例を示しています。この例では、上位 10 個のベクターは次のクエリ条件を満たす必要があります: Col_Keyword 列の値が hangzhou であり、Col_Long 列の値が 4 未満であること。

private static void knnVectorQuery(SyncClient client) {
    SearchQuery searchQuery = new SearchQuery();
    KnnVectorQuery query = new KnnVectorQuery();
    query.setFieldName("Col_Vector");
    query.setTopK(10); // テーブル内で指定されたベクターと最も類似度が高い上位 10 個のベクターを返します。
    query.setFloat32QueryVector(new float[]{0.1f, 0.2f, 0.3f, 0.4f});
    // 上位 10 個のベクターのクエリ条件を指定します: Col_Keyword 列の値が hangzhou で、Col_Long 列の値が 4 未満であること。
    query.setFilter(QueryBuilders.bool()
            .must(QueryBuilders.term("Col_Keyword", "hangzhou"))
            .must(QueryBuilders.range("Col_Long").lessThan(4))
    );
    searchQuery.setQuery(query);
    searchQuery.setLimit(10);
    // スコアに基づいてクエリ結果をソートします。
    searchQuery.setSort(new Sort(Collections.singletonList(new ScoreSort())));
    SearchRequest searchRequest = new SearchRequest("<TABLE_NAME>", "<SEARCH_INDEX_NAME>", searchQuery);
    SearchRequest.ColumnsToGet columnsToGet = new SearchRequest.ColumnsToGet();
    columnsToGet.setColumns(Arrays.asList("Col_Keyword", "Col_Long"));
    searchRequest.setColumnsToGet(columnsToGet);
    // Search 操作を呼び出します。
    SearchResponse resp = client.search(searchRequest);
    for (SearchHit hit : resp.getSearchHits()) {
        // スコアを表示します。
        System.out.println(hit.getScore());
        // データを表示します。
        System.out.println(hit.getRow());
    }
}

よくある質問

Tablestore KNN ベクタークエリのパフォーマンスを最適化するにはどうすればよいですか?

Tablestore:KNN ベクタークエリ

前提条件

使用上の注意

パラメーター

例

よくある質問

関連ドキュメント