AI・機械学習
中級

埋め込みモデルベンチマーク(ウメコミモデルベンチマーク)

埋め込みモデルベンチマーク(MTEB / BEIR)とは、テキスト埋め込みモデルの性能を多面的かつ標準化された方法で評価するためのフレームワークである。MTEB(Massive Text Embedding Benchmark)は8カテゴリ58データセットをカバーし、BEIRは18の検索データセットで構成される。モデル選定における客観的な判断基準を提供する。

0 回閲覧
0 いいね

埋め込みモデルベンチマークとは

埋め込みモデルの性能評価は、タスクやドメインによって最適なモデルが異なるため、単一の指標では不十分である。MTEB(Massive Text Embedding Benchmark)とBEIR(Benchmarking IR)は、この課題を解決するために設計された標準的な評価フレームワークであり、学術研究から実務でのモデル選定まで広く活用されている。

MTEBはHuggingFaceが管理するリーダーボードとして公開されており、2024年時点で250以上のモデルが登録されている。新しいモデルが発表されると、まずMTEBでの評価結果が注目される。

MTEB(Massive Text Embedding Benchmark)

評価カテゴリ

MTEBは8つのタスクカテゴリで構成され、埋め込みモデルの汎用性を包括的に評価する。

カテゴリデータセット数主要指標評価内容
Classification12Accuracyテキスト分類の精度
Clustering11V-measureクラスタリング品質
Pair Classification3AP (Average Precision)ペアの関係判定
Reranking4MAP候補の再順位付け
Retrieval15nDCG@10情報検索の精度
STS10Spearman相関意味的テキスト類似度
Summarization1Spearman相関要約の質
BitextMining2F1対訳文の発見

リーダーボード上位モデル (2024年)

順位モデル平均スコアRetrievalClassificationパラメータ
1GTE-Qwen2-7B-Instruct67.257.177.97.6B
2Voyage-367.358.276.8非公開
3E5-Mistral-7B66.656.977.67.1B
4BGE-M366.155.874.2568M
5text-embedding-3-large64.655.475.4非公開

MTEB v2

2024年後半にはMTEB v2が提案され、従来の課題(データセットの飽和、評価バイアス)に対処するため、より難易度の高いデータセットの追加、多言語評価の強化、効率性メトリクス(推論速度・メモリ使用量)の統合が計画されている。

BEIR(Benchmarking IR)

概要

BEIRは情報検索(Information Retrieval)に特化したベンチマークで、18のデータセットで検索モデルのゼロショット汎化性能を評価する。医療(BioASQ)、科学(SCIDOCS)、金融(FiQA)、Q&A(Natural Questions)など多様なドメインをカバーしている。

データセットドメインクエリ数文書数特徴
MS MARCO一般Web6,9808.8M最大規模の検索ベンチマーク
Natural QuestionsWikipedia Q&A3,4522.7MGoogle検索クエリベース
HotpotQAマルチホップ推論7,4055.2M複数文書の推論が必要
FiQA金融64857K金融Q&A
SCIDOCS科学1,00025K科学論文の引用関係
BioASQ医療50014.9M生物医学Q&A
DBPedia百科事典4004.6Mエンティティ検索

評価指標

BEIRの主要指標はnDCG@10(Normalized Discounted Cumulative Gain at rank 10)であり、検索結果上位10件の適合度を加重評価する。MAP(Mean Average Precision)やRecall@100も補助指標として使用される。

多言語ベンチマーク

MIRACL

多言語情報検索ベンチマークMIRACL(Multilingual Information Retrieval Across a Complement of Languages)は、18言語のWikipedia記事を対象とした検索タスクである。日本語を含むCJK言語のサポートが充実しており、多言語埋め込みモデルの実力を測る重要な指標となっている。

言語BGE-M3E5-MistralmE5-largeCohere v3
日本語73.566.268.461.8
英語74.172.870.268.5
中国語69.861.465.758.2
韓国語68.258.962.155.7

JMTEB(日本語MTEB)

2024年にはJMTEB(Japanese MTEB)が提案され、日本語に特化した埋め込みモデル評価基準が整備されつつある。形態素解析や漢字・かな混在テキストへの対応力など、日本語固有の課題に焦点を当てている。

ベンチマークの限界と注意点

データ汚染リスク

学習データにベンチマークのテスト分割が混入するデータ汚染(Data Contamination)が問題視されている。特に大規模Webコーパスで学習されたモデルでは、MTEBのデータセットが学習データに含まれている可能性がある。

タスク偏り

MTEB全体の平均スコアだけでモデルを評価すると、分類タスクのスコアが高いモデルが有利になる傾向がある。RAG用途ではRetrieval部門のスコアを重点的に確認すべきであり、多言語検索ではMIRACLの該当言語スコアが最も参考になる。

ドメインギャップ

ベンチマークは主にWikipedia・Web・学術論文のデータに基づいており、特定ドメイン(社内文書、製品マニュアル、コードベース)での性能とは乖離がある。本番導入前に自社データでの評価(カスタムベンチマーク)を実施することが不可欠である。

FAQ

Q1: MTEBスコアだけでモデルを選んで問題ないか?

MTEBスコアは比較の出発点としては有用だが、そのまま採用判断に直結させるべきではない。以下の3点を追加で確認すべきである。(1) 自社ドメインのテストデータでの性能、(2) 推論速度・コスト・デプロイ要件との適合性、(3) 対象言語での性能(英語以外の場合はMIRACLやJMTEBを参照)。

Q2: 自社データでのベンチマーク構築方法は?

50-200件のクエリと対応する正解文書のペアデータを用意し、nDCG@10とRecall@100を計算するのが基本的な方法である。RAGASライブラリやsentence-transformersのInformationRetrievalEvaluatorクラスを使えば、数行のコードで評価パイプラインを構築できる。

Q3: BEIRとMTEBの関係は?

BEIRはMTEBのRetrieval部門に統合されている。MTEBのRetrievalスコアは実質的にBEIRの評価結果に基づいており、BEIRのデータセットがそのまま使用されている。検索性能のみを比較したい場合はMTEB Retrievalスコア(≒BEIRスコア)を参照すれば十分である。

Q4: MTEBリーダーボードはどこで確認できるか?

HuggingFaceのSpaces上でmteb/leaderboardとして公開されている。フィルタ機能でモデルサイズ・言語・タスクカテゴリ別にソートでき、各モデルの詳細なタスク別スコアも確認可能である。更新は不定期だが、主要モデルのリリース後1-2週間以内に反映される傾向がある。