本文へ移動
Hugging Face Blog·· nullおすすめAIスコア62

Sentence Transformersでマルチベクトル埋め込みを圧縮するトークンプーリング対応学習

Token Pooling-Aware Training for Multi-Vector Embedding Models

AIによる概要

Hugging Face Blogの執筆者が、Sentence Transformersを使ったトークンプーリング対応学習の方法と評価結果を紹介しています。pool_factor=4では、通常学習モデルのnDCG@10が49.03だったのに対し、プーリング対応学習モデルは51.23となり、文書ベクトル数を非圧縮時より74.8%削減しました。

推荐理由

同じベクトル数で通常学習とプーリング対応学習を比較し、検索精度と保存量のトレードオフを数値で確認できます。

出典:Hugging Face Blog · huggingface.co