メインコンテンツまでスキップ

検索のためのデータモデル設計

情報検索システムは検索エンジンとも呼ばれ、Retrieval-augmented generation(RAG)、ビジュアル検索、製品レコメンデーションなど、さまざまな AI アプリケーションに不可欠です。これらのシステムの中核となるのは、情報を整理・インデックス化し、取得するために慎重に設計されたデータモデルです。

Zilliz Cloud では、コレクションスキーマを通じて検索データモデルを指定し、非構造化データ、その密またはスパースなベクトル表現、および構造化メタデータを整理できます。テキスト、画像、その他のデータ型を扱う場合でも、このハンズオンガイドは、実践的な検索データモデルを設計するために、主要なスキーマの概念を理解して適用するのに役立ちます。

Kc3Cweq1AhAmMGbrVgRcTlTKnUf

データモデル​

検索システムのデータモデル設計では、ビジネスニーズを分析し、情報をスキーマで表現されたデータモデルとして抽象化します。適切に定義されたスキーマは、データモデルをビジネス目標に合わせ、データの一貫性とサービス品質を確保するうえで重要です。さらに、適切なデータ型とインデックスを選択することは、ビジネス目標を経済的に達成するうえで重要です。

ビジネスニーズの分析​

ビジネスニーズに効果的に対応することは、ユーザーが実行するクエリの種類を分析し、最も適切な検索方法を決定することから始まります。

  • ユーザークエリ: ユーザーが実行すると想定されるクエリの種類を特定します。これにより、スキーマが実際のユースケースをサポートし、検索パフォーマンスを最適化できるようになります。たとえば、次のようなものがあります。

    • 自然言語クエリに一致するドキュメントの取得

    • 参照画像に類似する画像の検索、またはテキスト記述に一致する画像の検索

    • 名前、カテゴリ、ブランドなどの属性による商品の検索

    • 構造化メタデータ(公開日、タグ、評価など)に基づくアイテムのフィルタリング

    • ハイブリッドクエリでの複数条件の組み合わせ(例:ビジュアル検索において、画像とそのキャプションの両方の意味的な類似性を考慮)

  • 検索方法: ユーザーが実行するクエリの種類に合った適切な検索手法を選択します。手法によって目的が異なり、多くの場合、組み合わせることでより強力な結果を得られます。

    • セマンティック検索: 密ベクトルの類似性を使用して、意味が類似するアイテムを見つけます。テキストや画像などの非構造化データに最適です。

    • 全文検索: キーワードマッチングによってセマンティック検索を補完します。全文検索では字句解析を利用して、長い単語が断片化されたトークンに分割されるのを回避し、取得時に特殊な用語を捉えることができます。

    • メタデータフィルタリング: ベクトル検索に加えて、日付範囲、カテゴリ、タグなどの制約を適用します。

ビジネス要件を検索データモデルに変換する​

次のステップでは、情報の主要な構成要素とその検索方法を特定することにより、ビジネス要件を具体的なデータモデルに変換します。

  • 保存する必要のあるデータを定義します。たとえば、生のコンテンツ(テキスト、画像、音声)、関連するメタデータ(タイトル、タグ、著者情報)、およびコンテキスト属性(タイムスタンプ、ユーザーの行動など)です。

  • 各要素に適切なデータ型と形式を決定します。たとえば、次のとおりです。

    • テキスト記述 → string

    • 画像またはドキュメントの埋め込み → 密またはスパースのベクトル

    • カテゴリ、タグ、フラグ → string、array、bool

    • 価格や評価などの数値属性 → integer または float

    • 著者の詳細などの構造化情報 -> json

これらの要素を明確に定義することで、データの一貫性、正確な検索結果、およびダウンストリームのアプリケーションロジックとの統合の容易さが確保されます。

スキーマ設計​

Zilliz Cloud では、データモデルはコレクションスキーマで表現されます。コレクションスキーマ内の適切なフィールドを設計することが、効果的な取得を可能にする鍵となります。各フィールドは、コレクションに保存される特定の種類のデータを定義し、検索プロセスにおいて独自の役割を果たします。大まかに言うと、Zilliz Cloud は ベクトルフィールド と スカラーフィールド という 2 つの主要なフィールドタイプをサポートしています。

ここで、データモデルを、ベクトルと補助的なスカラーフィールドを含むフィールドのスキーマにマッピングできます。各フィールドがデータモデルの属性と対応していることを確認し、特にベクトル型(密またはスパース)とその次元に注意してください。

ベクトルフィールド​

ベクトルフィールドには、テキスト、画像、音声などの非構造化データ型の埋め込みが保存されます。これらのベクトル埋め込みは、データ型と使用する取得方法に応じて、密、スパース、またはバイナリになります。一般に、密ベクトルはセマンティック検索に使用され、スパースベクトルは全文検索や字句マッチングに適しています。バイナリベクトルは、ストレージと計算リソースが限られている場合に役立ちます。コレクションには複数のベクトルフィールドを含めることで、マルチモーダルまたはハイブリッドな取得戦略を実現できます。このトピックの詳細なガイドについては、マルチベクトルハイブリッド検索 を参照してください。

Zilliz Cloud は、密ベクトル 用の FLOAT_VECTOR、スパースベクトル 用の SPARSE_FLOAT_VECTOR、バイナリベクトル 用の BINARY_VECTOR というベクトルデータ型をサポートしています。

スカラーフィールドと複合フィールド​

スカラーフィールドには、数値、文字列、日付など、一般にメタデータと呼ばれる原始的な構造化値が保存されます。これらのスカラー値はベクトル検索の結果とともに返すことができ、フィルタリングと並べ替えに不可欠です。これらを使用すると、ドキュメントを特定のカテゴリや定義された期間に限定するなど、特定の属性に基づいて検索結果を絞り込めます。

Zilliz Cloud は、BOOL、INT8/16/32/64、FLOAT、DOUBLE、VARCHAR などのスカラー型に加えて、JSON や ARRAY などの複合型をサポートしており、非ベクトルデータの保存とフィルタリングに使用できます。これらの型は、検索操作の精度とカスタマイズ性を高めます。

スキーマ設計で高度な機能を活用する​

スキーマを設計する際には、サポートされているデータ型を使用してデータをフィールドにマッピングするだけでは十分ではありません。フィールド間の関係と、構成に利用できる戦略を十分に理解することが重要です。設計段階で主要な機能を念頭に置くことで、スキーマが当面のデータ処理要件を満たすだけでなく、将来のニーズにも拡張・適応できるようになります。これらの機能を慎重に統合することで、Zilliz Cloud の機能を最大限に活用し、より広範なデータ戦略と目標を支える強固なデータアーキテクチャを構築できます。ここでは、コレクションスキーマを作成する主要な機能の概要を説明します。

プライマリキー​

プライマリキーフィールドは、コレクション内の各エンティティを一意に識別するため、スキーマの基本的な構成要素です。プライマリキーの定義は必須です。整数型または文字列型のスカラーフィールドであり、is_primary=True としてマークする必要があります。オプションで、プライマリキーに auto_id を有効にできます。この場合、コレクションにデータが取り込まれるにつれて単調に増加する整数が自動的に割り当てられます。

詳細については、プライマリフィールドとAutoID を参照してください。

パーティショニング​

検索を高速化するために、オプションでパーティショニングを有効にできます。パーティショニング用に特定のスカラーフィールドを指定し、検索時にこのフィールドに基づいてフィルタリング条件を指定すると、検索範囲を関連するパーティションのみに効果的に限定できます。この方法は、検索対象の範囲を減らすことで、取得操作の効率を大幅に高めます。

詳細については、Partition Key を使用する を参照してください。

アナライザー​

アナライザーは、テキストデータを処理して変換するための不可欠なツールです。その主な機能は、生のテキストをトークンに変換し、インデックス作成と取得のために構造化することです。これは、文字列をトークン化し、ストップワードを削除し、個々の単語をステミングしてトークンに変換することで行います。

詳細については、アナライザーの概要 を参照してください。

関数​

Zilliz Cloud では、スキーマの一部として組み込み関数を定義し、特定のフィールドを自動的に導出できます。たとえば、VARCHAR フィールドからスパースベクトルを生成して全文検索をサポートする組み込みの BM25 関数を追加できます。これらの関数によって導出されるフィールドは前処理を効率化し、コレクションが自己完結的でクエリ可能な状態を維持できるようにします。

詳細については、全文検索 を参照してください。

実例​

このセクションでは、上の図に示すマルチメディアドキュメント検索アプリケーションのスキーマ設計とコード例を概説します。このスキーマは、次のフィールドにマッピングされるデータを含む記事のデータセットを管理するように設計されています。

フィールドデータソース使用する検索方法プライマリキーパーティションキーアナライザーFunction Input/Output
article_id (INT64)auto_id を有効にして自動生成Get を使用したクエリYNNN
title (VARCHAR)記事のタイトルテキストマッチNNYN
timestamp (INT32)公開日パーティションキーによるフィルタリングNYNN
text (VARCHAR)記事の生テキストマルチベクトルハイブリッド検索NNYinput
text_dense_vector (FLOAT_VECTOR)テキスト埋め込みモデルによって生成された密ベクトル基本ベクトル検索NNNN
text_sparse_vector (SPARSE_FLOAT_VECTOR)組み込みの BM25 関数によって自動生成されたスパースベクトル全文検索NNNoutput

スキーマの詳細と、さまざまな種類のフィールドを追加するための詳しいガイダンスについては、スキーマの解説 を参照してください。

ステップ 1: スキーマを初期化する​

まず、空のスキーマを作成する必要があります。このステップでは、データモデルを定義するための基盤となる構造を確立します。

python
from pymilvus import MilvusClient

schema = MilvusClient.create_schema()

ステップ 2: フィールドを追加する​

スキーマを作成したら、次のステップでは、データを構成するフィールドを指定します。各フィールドは、それぞれのデータ型と属性に関連付けられます。

python
from pymilvus import DataType

schema.add_field(field_name="article_id", datatype=DataType.INT64, is_primary=True, auto_id=True, description="article id")
schema.add_field(field_name="title", datatype=DataType.VARCHAR, enable_analyzer=True, enable_match=True, max_length=200, description="article title")
schema.add_field(field_name="timestamp", datatype=DataType.INT32, description="publish date")
schema.add_field(field_name="text", datatype=DataType.VARCHAR, max_length=2000, enable_analyzer=True, description="article text content")
schema.add_field(field_name="text_dense_vector", datatype=DataType.FLOAT_VECTOR, dim=768, description="text dense vector")
schema.add_field(field_name="text_sparse_vector", datatype=DataType.SPARSE_FLOAT_VECTOR, description="text sparse vector")

この例では、フィールドに次の属性が指定されています。

  • プライマリキー: article_id はプライマリキーとして使用され、受信するエンティティにプライマリキーが自動的に割り当てられるようにします。

  • パーティションキー: timestamp はパーティションキーとして割り当てられ、パーティションによるフィルタリングを可能にします。

  • テキストアナライザー: テキストアナライザーは 2 つの文字列フィールド title と text に適用され、それぞれテキストマッチと全文検索をサポートします。

ステップ 3:(任意)関数を追加する​

データクエリ機能を強化するために、関数をスキーマに組み込むことができます。たとえば、特定のフィールドに関連する処理を行う関数を作成できます。

python
from pymilvus import Function, FunctionType

bm25_function = Function(
name="text_bm25",
input_field_names=["text"],
output_field_names=["text_sparse_vector"],
function_type=FunctionType.BM25,
)

schema.add_function(bm25_function)

この例では、スキーマに組み込みの BM25 関数を追加し、text フィールドを入力として使用して、生成されたスパースベクトルを text_sparse_vector フィールドに保存します。

次のステップ​