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

since
平成38年9月29日

← 検索結果に戻る

質問と商品情報の両方から検索語を作るCoGR:候補検索での効果を検証

CoGR

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

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

原論文の図の説明を読む

Figure 1 : The overall CoGR training pipeline and performance. CoGR consists a SFT stage and a co-evolving RL stage looping between Item and Query LLMs. CoGR achieves strong performance gains on both internal APP marketplace and public item search benchmarks.

出典:CoGR / arXiv ↗

概要

  • CoGRは検索文と項目の双方から語句を生成し、転置索引で照合して候補を選ぶ。
  • 著者らが報告したCoGR-4Bの候補集合全体のF1は、APP Marketplaceで0.3963、WANDSで0.6819だった。
  • 社内文書への応用には検索文と関連文書のラベルが必要で、検索品質と索引更新の負担を検証する必要がある。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、CoGRの仕組み

左から右へ読む
  1. 01検索文と項目文
  2. 02両側で語句生成
  3. 03転置索引で照合
  4. 04項目候補

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

研究の背景

CoGRは、大量の項目から質問に合う候補を選ぶ検索の最初の段階を扱う。ここで関連項目を落とすと、後段で順位を付け直しても取り戻せない。一方、候補に無関係な項目が多いと後段の処理負担が増す。

従来の語句照合では、意味が近くても表記が異なる質問と項目を結びにくい。候補の取りこぼしと混入を両方抑えるため、質問と項目の表し方をそろえることが課題となる。

手法

CoGRは検索文用と項目用の別々の言語モデルで、少数の検索語句を作る。転置索引は語句から該当項目を引く仕組みで、生成語句に加えて元の検索文・項目も照合に使う。語句が重なる項目を候補にし、BM25という語句照合の点数で並べる。例えば商品の題名・説明文と購入者の検索文から、商品候補を返す。

まず関連項目の語句から検索文側の目標語句を作り、両モデルを教師あり学習で初期化する。次に片側の索引を固定し、検索結果を報酬にする強化学習で交互に更新する。検索文側には候補の適合率と再現率を合わせたF1を与え、項目側にはその項目の語句だけを差し替えた際のF1の変化を与える。

新規性

CoGRの主張する進歩は、言語モデルが質問側だけを補う方法から進み、質問と項目の両方に検索用の語句を作らせ、その語句を直接照合する点にある。

学習でも両側を交互に更新する。項目側には、自分の語句だけを差し替えた場合に、質問から項目を探す際のF1がどう変わるかを報酬として与える。

従来手法との違い

著者らは、語句を照合するBM25とSPLADE-v2、ベクトルで探すDPRとANCE、項目識別子を生成するDSIなど、計10手法を比較した。質問側の書き換えを強化学習するDeepRetrievalも含まれる。

CoGRは両側で生成した語句を転置索引で照合し、候補をBM25で並べる。優位性が報告された範囲は、アプリ検索と商品検索の二つの評価データである。

実験結果

著者らは社内APP Marketplaceと公開の商品検索データWANDSで評価した。質問を学習用と評価用に分け、項目集合は共通とした。評価質問数は順に1,500件と50件で、表のF1は質問ごとに求めた候補集合全体の値の平均である。

CoGR-4BのF1は順に0.3963、0.6819。各データで最も強い比較手法ANCE-Qwen4Bの0.3575、0.5012に対し、著者らは相対改善率10.9%、36.1%を報告した。この分割が示すのは未知の質問への性能であり、未知の項目への性能ではない。

応用の可能性

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

編集上の応用案:社内の手順書検索で、社員の検索文から関連文書の候補を返す。入力は検索文と各手順書の題名・本文、出力は候補一覧とし、担当者が文書の内容を確認する。

適用には、過去の検索文と関連文書を結ぶラベル付きデータを用意し、文書側と検索文側の語句生成器を学習させる。センサー入力は想定しない。社内文書でも検索品質が上がるかは未検証である。

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

導入時の検証

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

導入時の検証案:手順書の題名・本文と実際の検索文を集め、各検索文に関連する文書を担当者がラベル付けする。質問単位で学習用と評価用に分け、項目側の初期語句を作ってから、両側の教師あり学習と交互の更新を試す。

現行の語句検索を基準に、評価用質問で候補集合全体のF1、関連文書の取りこぼし、無関係な候補数を比べる。文書改訂後も測り、索引更新の所要時間を記録する。入力はテキストで、センサーや画像ラベルは想定しない。

限界と課題

確認できた本文の範囲では、CoGR固有の失敗条件や運用上の限界は明示されていない。実験では項目の題名と説明文、質問と項目の関連ラベルを利用し、評価時も学習時と同じ項目集合を保持した。

編集上の検証事項は、関連ラベルが少ない領域での性能、文書の追加・改訂時に語句と索引を更新する負担、曖昧な検索文で無関係な候補が増えないかである。未知の項目に対する性能も別途調べる必要がある。

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

出典

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

It Takes Two to Match: Co-Evolving Generative Retriever with Reinforcement Learning

Retrieval is the first stage of modern search and advertising systems, selecting a candidate set from a large item universe for downstream ranking and auction. Recent work increasingly leverages LLMs to improve retrieval through query expansion, data synthesis, and retrieval-feedback training. However, the generative component is typically used for query-side augmentation, while final matching is still delegated to a downstream retriever. We introduce CoGR, a retrieval framework that instead trains LLMs to directly construct retrieval representations on both query and item sides. Each generator produces a compact set of keywords, which are matched directly through an inverted index, preserving compatibility with existing keyword-based retrieval infrastructure. CoGR uses a two-stage training pipeline. Supervised fine-tuning first establishes an aligned keyword space, after which co-evolving reinforcement learning alternately optimizes the query- and item-side generators with GRPO against the opposite side's frozen index. Both sides optimize the same query-to-item retrieval $F_1$ objective: the query side receives retrieval $F_1$ directly, while the item side receives a counterfactual marginal reward measuring the change in query-side $F_1$ caused by its generated keywords. Across 10 representative sparse, dense, and generative baselines, CoGR achieves the best performance on both an internal APP Marketplace dataset and the public WANDS benchmark, improving $F_1$ over the strongest baseline by $10.9\%$ and $36.1\%$, respectively. Further analysis shows stable co-evolution and increasingly aligned query--item keyword spaces over training.

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