arXiv論文メモ
新着一覧
cs.LG · 査読状況未確認

全エキスパートの知識を少ない計算とメモリで使う

IntBMoE: Integrating Block-Level Conditioning into Expert Composition for Full-Participation Mixture-of-Experts

Ran Cheng, Longfei Xu, Zheng Liu, Kaikui Liu, Xiangxiang Chu

この論文をやさしく読む

ひとことで言うと

複数の専門モデルの知識を合成して使いながら、実際に動かす部品と保存する部品の数を抑える仕組みです。知識への参加と計算実行を分けています。

何に役立つ?

考えられる用途は、計算時間とメモリに制約のある分類・言語処理・推薦です。要旨はAMap推薦システムへの導入とオンラインA/Bテストも報告しています。

この研究の面白いところ

入力のたびに新しい合成モデルを無制限に作らず、小さなコードブックでブロック数を固定します。全エキスパートの知識と疎な実行を両立させようとしています。

どこまで分かった?

UVCTRの2.4%はオンライン試験での相対改善です。60ミリ秒は運用上の遅延予算であり、全要求の実測遅延が同じ値だという意味ではありません。一般的な全MoE構成での優位を示したものではありません。

v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。

アブストラクトの日本語訳

混合エキスパート(MoE)はモデル容量を拡張するが、既存の設計では三つの量を独立には設定できない。単一トークンについて、参加数とは出力に知識を提供するエキスパートの数、実行数とは実際に計算する数、すなわち計算コスト、実体化数とは構築して保存する必要があるエキスパート規模のパラメータ集合の数、すなわちメモリコストである。疎なルーティングは実行数と実体化数を抑えるが、各トークンへ貢献するエキスパートが少数となり、参加数を減らす。密な出力混合は全参加を回復するが、実行量がエキスパート数とともに増える。パラメータ統合は実行を一つのエキスパートに抑えるが、実体化数はルーティング判断の数とともに増える。 本研究では、密なエキスパート合成と疎なブロック実行を組み合わせて三量を切り離す、ブロック条件付きMoEであるIntBMoEを提案する。ブロックは小さな学習済みコードブックの各エントリから一つずつ作られる。各内部層では、軽量ハイパーネットワークが、その層のプール内の全エキスパート基底を一つの合成エキスパートへ統合する。各合成エキスパートはプール全体から情報を得るため、参加は全体に及ぶ。ルーターが各トークンを少数のブロックにだけ送るため、実行は疎に保たれる。存在するブロック数は入力ではなくコードブックが決めるため、実体化数は有界となる。さらに、Dual-Path Residual Gating(DPRG)が、独立に合成された二つの経路を乗法的ゲートで結合する。 画像分類実験では、代表的な疎および密なMoE基準手法に対して一貫した改善を示す。言語モデリングと逐次推薦での追加実験により、画像以外への一般化も検証する。IntBMoEはAMapの生成型推薦システムに全面導入され、60ミリ秒の遅延予算の下で数億人の利用者にサービスを提供している。オンラインA/BテストではUVCTRが相対的に2.4%向上した。コードはhttps://github.com/AMAP-ML/DreamX-Rec/で公開している。

v1の要旨から自動生成。本文の精読・人による確認は未実施。

初稿
2026-09-18(UTC)
最新改訂
2026-09-18 · v1
査読・掲載
査読状況未確認
arXivで読むPDF

更新履歴

取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。

原文の要旨

Mixture-of-Experts (MoE) scales capacity, but existing designs cannot set three quantities independently. For a single token, participation is how many experts contribute knowledge to its output, execution is how many are actually computed (compute cost), and materialization is how many expert-sized parameter sets must be built and stored (memory cost). Sparse routing keeps execution and materialization low, but shrinks participation: for each token, only a few experts contribute. Dense output-mixing restores full participation, but its execution grows with the number of experts. Parameter-merging keeps execution at one expert, but its materialization grows with the number of routing decisions. We propose IntBMoE, a block-conditioned MoE that decouples all three by pairing dense expert composition with sparse block execution. Its blocks come from a small learned codebook, one per entry. At each internal layer, a lightweight hypernetwork merges all expert bases in that layer's pool into one composed expert. Participation is full, because every composed expert draws on the entire pool. Execution stays sparse, because a router sends each token to only a few blocks. Materialization is bounded, because the codebook, not the input, fixes how many blocks exist. Dual-Path Residual Gating (DPRG) further couples two independently composed paths through multiplicative gating. Experiments on image classification show consistent gains over representative sparse and dense MoE baselines. Additional experiments on language modeling and sequential recommendation validate its generalization beyond vision. IntBMoE is fully deployed in AMap's generative recommendation system, serving hundreds of millions of users under a 60ms latency budget, with a 2.4% relative UVCTR gain in online A/B testing. Our code is available at https://github.com/AMAP-ML/DreamX-Rec/.

arXiv ID: 2609.21346 / 要約の誤りについて