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

Weighted Ranker

Weighted Ranker は、それぞれに異なる重要度の重みを割り当てることで、複数の検索パスからの結果をインテリジェントに組み合わせて優先順位付けします。熟練したシェフが複数の材料のバランスを取って完璧な料理を作るように、Weighted Ranker は異なる検索結果のバランスを取り、最も関連性の高い組み合わせ結果を提供します。このアプローチは、複数の vector フィールドやモダリティをまたいで検索する際に、特定のフィールドが他のフィールドよりも最終ランキングに大きく寄与すべき場合に最適です。

Weighted Ranker を使用するタイミング

Weighted Ranker は、複数の vector 検索パスからの結果を組み合わせる必要があるハイブリッド検索シナリオ向けに特別に設計されています。特に次のようなケースで効果的です。

ユースケースWeighted Ranker が適している理由
E コマース検索画像の類似性とテキスト説明を組み合わせた商品検索小売業者が、ファッションアイテムでは視覚的な類似性を優先し、技術製品ではテキスト説明を重視できるようにします
メディアコンテンツ検索視覚的特徴と音声文字起こしの両方を用いた動画検索クエリの意図に応じて、視覚コンテンツと話し言葉の対話の重要度をバランスできます
ドキュメント検索異なるセクション向けの複数の埋め込みを使用する企業向けドキュメント検索フルテキスト埋め込みも考慮しつつ、タイトルや要約の埋め込みにより高い重みを与えられます

ハイブリッド検索アプリケーションで、複数の検索パスを組み合わせつつその相対的重要度を制御する必要がある場合、Weighted Ranker は理想的な選択肢です。

Weighted Ranker の仕組み

WeightedRanker 戦略の主なワークフローは次のとおりです。

  1. 検索スコアの収集: vector 検索の各パスから結果とスコアを収集します(score_1、score_2)。

  2. スコアの正規化: 各検索では異なる類似度メトリクスを使用する場合があり、その結果スコア分布も異なります。たとえば、類似度タイプとして Inner Product (IP) を使用するとスコア範囲は [−∞,+∞] になる可能性があり、Euclidean distance (L2) を使用するとスコア範囲は [0,+∞] になります。異なる検索から得られるスコア範囲は異なり、直接比較できないため、各検索パスからのスコアを正規化する必要があります。通常は、arctan 関数を適用してスコアを [0, 1] の範囲に変換します(score_1_normalized、score_2_normalized)。1 に近いスコアほど高い類似性を示します。

  3. 重みの割り当て: 異なる vector フィールドに割り当てられた重要度に基づいて、正規化されたスコア(score_1_normalized、score_2_normalized)に重み(wi)を割り当てます。各パスの重みは [0,1] の範囲である必要があります。結果として得られる重み付きスコアは score_1_weighted と score_2_weighted です。

  4. スコアのマージ: 重み付きスコア(score_1_weighted、score_2_weighted)を高い順に並べて、最終的なスコアセット(score_final)を生成します。

GdmNwbkN8haZO8bpQkOc2NIWnqF

Weighted Ranker の例

この例では、画像とテキストを含むマルチモーダル Hybrid Search(topK=5)を示し、WeightedRanker 戦略が 2 つの ANN 検索結果をどのように再ランキングするかを説明します。

  • 画像に対する ANN 検索の結果(topK=5):

    IDScore (image)
    1010.92
    2030.88
    1500.85
    1980.83
    1750.8
  • テキストに対する ANN 検索の結果(topK=5):

    IDScore (text)
    1980.91
    1010.87
    1100.85
    1750.82
    2500.78
  • WeightedRanker を使用して、画像検索結果とテキスト検索結果に重みを割り当てます。たとえば、画像 ANN 検索の重みが 0.6、テキスト検索の重みが 0.4 であるとします。

    IDScore (image)Score (text)Weighted Score
    1010.920.870.6×0.92+0.4×0.87=0.90
    2030.88N/A0.6×0.88+0.4×0=0.528
    1500.85N/A0.6×0.85+0.4×0=0.51
    1980.830.910.6×0.83+0.4×0.91=0.86
    1750.800.820.6×0.80+0.4×0.82=0.81
    110Not in Image0.850.6×0+0.4×0.85=0.34
    250Not in Image0.780.6×0+0.4×0.78=0.312
  • 再ランキング後の最終結果(topK=5):

    RankIDFinal Score
    11010.90
    21980.86
    31750.81
    42030.528
    51500.51

Weighted Ranker の使用方法

WeightedRanker 戦略を使用する際は、重みの値を入力する必要があります。入力する重みの値の数は、Hybrid Search における基本 ANN 検索リクエストの数に対応している必要があります。入力する重みの値は [0,1] の範囲内である必要があり、1 に近い値ほど重要度が高いことを示します。

Weighted Ranker を作成する

たとえば、Hybrid Search に 2 つの基本 ANN 検索リクエスト(テキスト検索と画像検索)があるとします。テキスト検索の方が重要と見なされる場合は、より大きい重みを割り当てる必要があります。

python
from pymilvus import Function, FunctionType

rerank = Function(
name="weight",
input_field_names=[], # Must be an empty list
function_type=FunctionType.RERANK,
params={
"reranker": "weighted",
"weights": [0.1, 0.9],
"norm_score": True # Optional
}
)
ParameterRequired?DescriptionValue/Example
nameYesこの Function の一意の識別子"weight"
input_field_namesYes関数を適用する vector フィールドのリスト(Weighted Ranker では空である必要があります)[]
function_typeYes呼び出す Function のタイプ。再ランキング戦略を指定するには RERANK を使用しますFunctionType.RERANK
params.rerankerYes使用する再ランキング手法を指定します。
Weighted Ranker を使用するには weighted に設定する必要があります。
"weighted"
params.weightsYes各検索パスに対応する重みの配列。値 ∈ [0,1]。
詳細については、Weighted Ranker の仕組みを参照してください。
[0.1, 0.9]
params.norm_scoreNo重み付けの前に生のスコアを正規化(arctan を使用)するかどうか。
詳細については、Weighted Ranker の仕組みを参照してください。
True

Weighted Ranker は、複数の vector フィールドを組み合わせるハイブリッド検索操作向けに特別に設計されています。ハイブリッド検索を実行する際は、各検索パスに対する重みを指定する必要があります。

python
from pymilvus import MilvusClient, AnnSearchRequest

# Connect to Milvus server
milvus_client = MilvusClient(
uri="YOUR_CLUSTER_ENDPOINT",
token="YOUR_CLUSTER_TOKEN"
)

# Assume you have a collection setup

# Define text vector search request
text_search = AnnSearchRequest(
data=["modern dining table"],
anns_field="text_vector",
param={},
limit=10
)

# Define image vector search request
image_search = AnnSearchRequest(
data=[image_embedding], # Image embedding vector
anns_field="image_vector",
param={},
limit=10
)

# Apply Weighted Ranker to product hybrid search
# Text search has 0.8 weight, image search has 0.3 weight
hybrid_results = milvus_client.hybrid_search(
collection_name,
[text_search, image_search], # Multiple search requests
ranker=rerank, # Apply the weighted ranker
limit=10,
output_fields=["product_name", "price", "category"]
)

ハイブリッド検索の詳細については、Multi-Vector Hybrid Search を参照してください。

Ctrl I