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

since
平成38年9月29日

← 検索結果に戻る

商品知識を検索と回答で共有するBSharedRAG:別々の調整層で両タスクを学習

BSharedRAG

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

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

原論文の図の説明を読む

Figure 5: Partial categories of WorthBuying dataset

出典:BSharedRAG / arXiv ↗

概要

  • 商品分野で追加学習した言語モデル本体を共有し、検索用と回答用のLoRAを別々に学習する。
  • 著者らの検索評価では、Hit@3がCPR-Ecomで68.40、WorthBuyingで58.60となり、比較対象の60.35、55.70を上回った。
  • 実務では検索命中率に加え、回答の事実性、商品取り違え、文書更新後の挙動を検証する必要がある。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、BSharedRAGの仕組み

左から右へ読む
  1. 01商品質問
  2. 02関連文書を検索
  3. 03上位3件で回答生成
  4. 04商品回答

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

研究の背景

BSharedRAGは、商品に関する質問に対し、関連文書を探してから回答を作る。RAG(検索拡張生成)とは、外部文書を検索して、その内容を文章生成の根拠に使う方式だ。種類の多い商品や頻繁に変わる情報について、モデルの記憶だけで答えると、根拠のない説明が混じりやすい。

検索器と回答器を別々に作る一般的な構成では、両者が同じ商品知識を身に付けても、その成果を共有しにくい。検索した文書が質問に似ていても、回答に必要な情報を欠く場合があるため、検索順位と回答品質の両方が実務上の課題になる。

手法

学習時には、基盤となる言語モデルを商品分野の文章と一般分野の文章で追加学習し、本体を固定する。その上で、LoRA(モデル本体を大きく変えず、少数の追加パラメータを学ぶ方法)を検索用と回答用に別々に学習する。検索用には関連文書と無関係な文書の違いを学ばせ、上位に出やすい誤った文書も「難しい負例」として使う。

回答用には質問・検索文書・正解回答の組を与え、文書を参照して答えるよう調整する。利用時は商品文書をあらかじめ数値表現に変換しておき、質問に近い上位3件を取り出して回答器へ渡す。たとえば携帯性を問う質問なら、該当商品の寸法や重量が書かれた記事を候補にし、その記事を入力として回答文を生成する。

新規性

著者らの主張する新規性は、商品分野の文章で追加学習した言語モデルの本体を検索と回答で共有し、それぞれに独立したLoRAを載せる構成にある。共通の分野知識を使いながら、検索と文章生成の学習目標を一つの調整層で釣り合わせる必要をなくした。長文の商品分析記事と質問・根拠文書・回答を組にしたWorthBuyingも構築している。

検索器が見つけた文書を回答器へ渡すところまでを一つの仕組みとして評価した点も重要だ。ただし、二つの調整層を使うことだけで改善を説明できるわけではなく、追加学習や難しい負例の効果も合わせて読む必要がある。

従来手法との違い

従来型のSeparate RAGでは検索器と回答器が別のモデルで、商品分野の追加学習から得た知識を本体として共有しない。FullyShared-RAGは両タスクを一つの調整層で学習する。BSharedRAGは本体のみを共有し、検索用と回答用のLoRAを分ける。

検索では、難しい負例で調整したBGE-large-zhなどと比較した。回答ではBaichuan2系列やSeparate RAG、FullyShared-RAGなどを比較対象にした。表のBLEU-3改善率23%はBaichuan2-7b-chatに検索結果を使った指示学習を加えた条件との比較であり、GPT-4を上回ったという意味ではない。

実験結果

著者らは中国語の商品文書を用い、検索をCPR-EcomとWorthBuying、回答生成をWorthBuyingで評価した。WorthBuyingには約73.5万文書と5万組の質問・文書・回答があり、検索テストは1000問、回答テストは人手で確認した500組を使う。Hit@3は上位3件に関連文書が入った割合で、BSharedRAGはCPR-Ecomで68.40、WorthBuyingで58.60を記録した。

最も強い比較対象のBGE-large-zh+難しい負例は、それぞれ60.35、55.70だった。回答のBLEU-3は12.63で、比較条件のBaichuan2-7b-chat+検索結果を使った指示学習の10.24を上回ったと著者らは報告する。BLEU-3は正解文との3語の並びの一致を測る指標で、回答の真偽を直接示すものではない。

応用の可能性

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

編集上の応用案:通販サイトの商品相談窓口で、利用者の質問と商品分析記事を入力し、根拠記事を示した回答案を担当者に渡す。たとえば「この機種は持ち運びに向くか」という質問について、関連する記事を検索し、重量や寸法に触れた箇所を基に文章を作る。

公開前には担当者が根拠と回答の対応、対象商品の一致、記事の更新日を確認する。論文は商品分野の質問応答を扱うが、この窓口運用や担当者確認の効果を検証したものではない。

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

導入時の検証

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

導入時の検証案:一つの商品カテゴリから、出典を確認できる記事と実際の質問を少量選ぶ。まず既存の独立した検索器・回答器を基準にし、同じ文書群と質問でBSharedRAG型の構成を比べる。検索用の正解文書と、回答に必須の事実を人手で記録しておく。

上位3件に根拠文書が含まれる割合、回答が必須事実を満たす割合、商品を取り違える割合を測る。記事を更新した後も同じ質問を流し、古い情報が残る件数を確認する。比較条件の文書、質問、判定基準をそろえて初めて、共有する本体が運用上の改善につながるか判断できる。

限界と課題

著者らは、短く雑音の多い商品レビューや質のばらつく質問回答を既存データの課題と説明する。また、全面共有では検索と生成の学習目標が干渉し得るとしており、表でも全面共有型の成績は低い。一方、提示された実験は中国語の商品文書を中心とする。

導入前の確認事項は、日本語の商品説明でも同じ傾向が出るか、更新された仕様を正しく拾えるか、検索結果に答えがないときに回答を控えられるかである。Hit@3やBLEU-3の改善だけでは、個々の回答の事実性や運用時の更新負荷は判断できない。

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

出典

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

BSharedRAG: Backbone Shared Retrieval-Augmented Generation for the E-commerce Domain

Retrieval Augmented Generation (RAG) system is important in domains such as e-commerce, which has many long-tail entities and frequently updated information. Most existing works adopt separate modules for retrieval and generation, which may be suboptimal since the retrieval task and the generation task cannot benefit from each other to improve performance. We propose a novel Backbone Shared RAG framework (BSharedRAG). It first uses a domain-specific corpus to continually pre-train a base model as a domain-specific backbone model and then trains two plug-and-play Low-Rank Adaptation (LoRA) modules based on the shared backbone to minimize retrieval and generation losses respectively. Experimental results indicate that our proposed BSharedRAG outperforms baseline models by 5% and 13% in Hit@3 upon two datasets in retrieval evaluation and by 23% in terms of BLEU-3 in generation evaluation. Our codes, models, and dataset are available at https://bsharedrag.github.io.

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