アーカイブ / VLM / RAG
画像と文章を一つのモデルで検索するUEmbed:語の重みと全体表現を同時に作る
UEmbed
UEmbedの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1 : Overview of UEmbed . (a) Framework: we append N N learnable special tokens to a decoder-only MLLM and assign each token a disjoint vocabulary subset. (b) Sparse cases: examples of activated lexical tokens produced by UEmbed -9B. (c) Performance: UEmbed achieves strong sparse and dense retrieval results on MMEB-v2.
概要
- UEmbedは一つのデコーダー型モデルから、語ごとの重みを持つ疎ベクトルと入力全体を表す密ベクトルを作る。
- 著者らのMMEB-v2評価で、UEmbed-9Bの総合スコアは密表現71.8、疎表現71.0だった。
- 著者らは不自然な語の出現と英語・中国語への偏りを報告しており、日本語資料への適用には個別の検証が要る。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、UEmbedの仕組み
- 01文章・画像を入力
- 02末尾で内容を集約
- 03語彙別の重みを計算
- 04疎・密ベクトル
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
UEmbedは、文章や画像を検索用の数値表現に変える手法だ。検索では、文書中の語で探す方法は結果を説明しやすい一方、質問と資料で表記が違うと関連情報を見落とし得る。写真付き資料では、画像の内容も検索に使いたい。
学習型疎検索は、元の入力にない関連語にも重みを付けて検索する。ただし既存手法の多くは文章向けの双方向モデルを使い、画像には追加の部品を要した。一つのモデルで両方を扱えるかが論点になる。
手法
UEmbedは入力の末尾に、学習で働き方を決める16個の特殊トークンを付ける。デコーダー型モデルは前にある情報を参照できるため、各トークンは入力内容を踏まえて担当する語の重みを予測する。語彙を重ならない16組に分け、予測結果をつないで疎ベクトルにする。疎ベクトルは、多数の語のうち限られた語に重みを持つ検索用の数値列だ。
特殊トークンの直前に置く終了トークンからは、入力全体をまとめた密ベクトルを得る。たとえば写真付き報告書と質問をそれぞれ変換し、数値表現同士の近さで候補を探す。学習では正解の質問・資料の組を誤候補より近づけ、疎ベクトルで重みを持つ語が広がりすぎないよう調整する。
新規性
UEmbedの提案は、画像も扱うデコーダー型モデルから、語ごとの重みを持つ疎ベクトルと、入力全体を表す密ベクトルを一度の処理で得る点にある。語彙を分担する特殊トークンにより、因果的な読み取りのまま疎ベクトルを作る。
著者らは、MMEB-v2で評価した初の疎検索モデルだと述べる。画像を語へ変換するための別モデルを足さずに、文章と視覚情報を同じ基盤で扱う構成である。
従来手法との違い
BM25は文書に現れる語を手掛かりにするため、同じ内容でも表記が異なる資料を拾いにくい。SPLADEに代表される学習型疎検索は関連語にも重みを付けるが、従来は入力を双方向に読む構成が中心だった。UEmbedは末尾の特殊トークンから語の重みを計算する。
著者らは、同じQwen3.5の基盤モデル、学習データ、疎さを促す条件を使う双方向のSPLADE型とも比較した。これはMMEB-v1の画像部分集合と、50万件を使う分析条件での比較である。
実験結果
著者らのMMEB-v2評価は画像、動画、視覚情報の多い文書を含む78データセットが対象で、UEmbed-9Bの総合スコアは密表現71.8、疎表現71.0だった。同じ表のRzenEmbed-V2-7Bは71.1、独自の大規模データでも学習したQwen3-VL-Embedding-8Bは77.8で、学習条件は同一ではない。
文章検索のBEIRの9データセットでは、上位10件の順位を考慮する指標nDCG@10の平均が、UEmbed-9Bの密表現で56.3、疎表現で55.2だった。疎表現の値はEcho-Mistral-SPLADEの55.2と同じ。別のMMEB-v1画像部分集合では、著者らの疎表現が63.4、比較したSPLADE型が61.3だった。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案:設備点検の写真付き報告書を、現場担当者の短い質問から探す。質問文を入力し、保管済みの写真と本文を検索対象にして、候補報告書と検索に使われた語の重みを表示する。担当者が原本を開いて適合性を判断する流れを想定する。
追加学習には、過去の質問と正解報告書の組、誤候補のデータが必要になる。評価用にも質問と正解の対応ラベルを別に用意する。既存資料を対象とするため新規センサーは想定しない。日本語の設備名への適応は未検証である。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案:まず既存の点検報告書から写真と本文を集め、現場で使う日本語の質問と正解報告書の対応ラベルを作る。追加学習を試す場合は質問・正解・誤候補の組を別に用意し、設備名や略称に合わせて適応する。既存資料を使うため新規センサーは不要だ。
同じ質問群でBM25、UEmbedの疎表現、密表現を比べ、正解が上位10件に入る割合と応答時間を測る。画像だけが手掛かりになる質問と略称を含む質問は分けて集計し、不自然な語の重みや索引の更新時間も確認する。
限界と課題
著者らは、疎ベクトルに「_alt」のような不自然な語が現れる場合と、重みが付く語が英語・中国語に偏る点を報告した。後者には、学習データの言語構成と基盤モデルの言語間の対応力が関係すると説明している。
編集上の検証事項は、日本語の略称や型番で正解資料を拾えるか、不自然な語が結果の説明を妨げないかである。社内資料での索引容量、資料更新にかかる時間、応答時間も用途ごとに測る必要がある。
出典
元のタイトル・要旨を確認する(英語)
UEmbed: Unified Sparse and Dense Multimodal Embeddings
Sparse retrieval underpins modern search systems, from web search to retrieval-augmented generation. Existing work has introduced Learned Sparse Retrieval (LSR) to push beyond exact lexical matching toward richer semantics. Yet LSR has so far remained tied to encoder-style bidirectional architectures, and its extension to multimodal settings still relies heavily on auxiliary cross-modal modules. To address these limitations, we introduce UEmbed (Unified Embedding), a decoder-only multimodal embedding model that produces both sparse lexical and dense representations in one causal forward pass. UEmbed appends N learnable special tokens to the input and partitions the vocabulary into N disjoint subsets. Each token's causal hidden state predicts sparse weights over its assigned subset, and the N subsets are concatenated into the full sparse vector. Trained on public data, we release UEmbed at 2B, 4B, and 9B scales. UEmbed-9B reaches 71.8 (dense) and 71.0 (sparse) on MMEB-v2, outperforming multimodal embedding models trained on publicly available data (e.g., RzenEmbed). On BEIR, UEmbed also remains competitive with strong dense and sparse baselines. Furthermore, we demonstrate the practical utility of UEmbed across three dimensions: effectiveness, efficiency, and agentic applications. Overall, UEmbed offers a new paradigm: it unifies dense and sparse embeddings in one model, while further extending sparse retrieval to unify text and multimodal inputs.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01




