表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

E5は文章検索をどう改善するか:大規模な文章ペアで学ぶ埋め込み表現

E5

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

E5の原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 1: Overview of our data curation pipeline and model architecture.

出典:E5 / arXiv ↗

概要

  • E5は関連する文章ペアを近づける対比学習で、文章を一つのベクトルに変換する。
  • BEIRの15データ平均では、事前学習のみのE5-PT-baseがnDCG@10で42.9、BM25が41.7だった。
  • 長文や厳密な語句一致には課題があり、日本語の社内文書では現行検索との比較が必要である。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、E5の仕組み

左から右へ読む
  1. 01質問と文章の組
  2. 02対応関係を学習
  3. 03文章を数値化
  4. 04近い文章を検索

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

文章検索では、質問と答えに同じ語が出てこないことがある。「経費の締切」と「精算申請期限」のように表現がずれると、語の一致に頼る検索は候補を見落とし得る。

文章埋め込みは文章を数値の並びで表し、意味が近い文章を探すための方法である。E5は、検索だけでなく分類や文章の類似度判定にも使える、一つの文章から一つのベクトルを作るモデルを目指した。

手法

著者らは質問と対応する文章の組をCCPairsに集め、関連する組を近づけ、関係しない組を遠ざける「対比学習」でE5を訓練した。同じ学習単位に入った別の組の文章を、関連しない例として使う。さらに、先に学習したモデルで組の整合性を調べ、質の低い組を除いた。

利用時は質問に「query:」、検索対象に「passage:」を付け、共通の符号化器でそれぞれ一つのベクトルにする。たとえば「申請期限は」と手順書の段落を数値化し、ベクトルの向きの近さで候補を並べる。必要なら、ラベル付きの質問応答や自然言語推論のデータで追加学習する。

新規性

E5の特徴は、質問と関連文章の組を多様なウェブ由来データから集め、質を確かめながら大規模な対比学習に用いた点にある。CCPairsでは約13億組を予備収集し、学習したモデルとの整合性による選別後、約2億7000万組を残した。

著者らは、ラベル付きデータを使わない事前学習だけでも、複数の検索課題でBM25の平均を上回ったと報告する。さらに少量のラベル付きデータで追加学習できるため、共通の文章表現を用途に合わせて調整できる。

従来手法との違い

BM25は質問と文書に現れる語を使って順位を付ける検索法で、語の一致が重要な場合に強い。E5は文章を一つの数値ベクトルに変換し、ベクトル同士の近さで関連度を測る。異なる表現の文章を結び付ける可能性がある一方、厳密な語の一致が必要な検索では差が出る。

論文のBEIR検索評価では、事前学習のみのE5-PT-baseの平均nDCG@10が42.9、BM25が41.7だった。ただしBM25が勝つ個別データもあるため、平均値だけで置き換えを判断できない。

実験結果

著者らはBEIRの公開入手できる15検索データと、MTEBの英語部分集合56データで評価した。BEIRのnDCG@10は、正解文書が上位10件のどこに現れたかを重み付きで測る指標である。事前学習のみのE5-PT-baseは平均42.9、BM25は41.7だった。

追加学習したE5-baseはMTEBの平均60.3、E5-largeは61.3と報告される。論文の表では、約40倍のパラメータを持つGTR-xxlは59.0である。ただし課題別の差があり、BEIRではTrec-CovidやTouche-2020などBM25が優位な例も載る。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:社内の手順書から、社員の質問に対応する段落を探す検索に使う。入力は「出張費の精算期限はいつか」といった質問と、あらかじめ区切った手順書の文章である。出力は質問に近い順の段落一覧と文書の所在にする。

文書側のベクトルを先に計算して保存し、質問を受けた時に近さを比べる。利用者には原文を開いて確認できる形で示す。論文は一般的な検索性能を評価しており、社内手順書での正答率を示したものではない。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:社内文書から質問と正解段落の組を少数作り、現行の語句検索を基準にE5の上位候補を比べる。質問には略語、言い換え、文書名を正確に含むものを混ぜ、文書の長さも記録する。

上位5件に正解が含まれる割合、最上位の正解率、検索にかかる時間、保存するベクトルの容量を測る。誤検索は語句一致の不足か、意味の取り違えかに分けて確認し、必要なら両方式の結果を併用する。

限界と課題

著者らは、長い文書の内容を固定長の一つのベクトルに収め切れるかは未解決と述べる。またBM25には単純さ、効率、結果を説明しやすい利点があり、長文や厳密な語の一致を要する検索では密なベクトル検索に課題が残るとしている。

導入時の確認事項は、日本語の社内文書や略語での検索精度、更新文書を反映する手間、重要な語を含む文書の取りこぼしである。論文のMTEB評価は英語部分集合が中心で、日本語の実務検索へ数値をそのまま移せない。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

Text Embeddings by Weakly-Supervised Contrastive Pre-training

This paper presents E5, a family of state-of-the-art text embeddings that transfer well to a wide range of tasks. The model is trained in a contrastive manner with weak supervision signals from our curated large-scale text pair dataset (called CCPairs). E5 can be readily used as a general-purpose embedding model for any tasks requiring a single-vector representation of texts such as retrieval, clustering, and classification, achieving strong performance in both zero-shot and fine-tuned settings. We conduct extensive evaluations on 56 datasets from the BEIR and MTEB benchmarks. For zero-shot settings, E5 is the first model that outperforms the strong BM25 baseline on the BEIR retrieval benchmark without using any labeled data. When fine-tuned, E5 obtains the best results on the MTEB benchmark, beating existing embedding models with 40x more parameters.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-29