本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

帰属誘導とカバレッジ最大化による構造的MoE圧縮のためのプルーニング

記事の要約

研究者らは、混合エキスパート(MoE)モデル向けの構造的プルーニングフレームワークを提案した。プルーニング比率の割り当てをチャネルスコアのカバレッジ最大化問題として再定式化し、帰属ベースの近似で効率的に解く。DeepSeekとQwenのMoEモデルでの実験では、4ビット量子化と組み合わせて50%または25%の構造的プルーニング下でも精度を維持し、Qwen3-30B-A3Bでメモリフットプリントを5.27倍削減、多様なベンチマークで最先端ベースラインを一貫して上回った。

ソースarXiv Machine Learning著者: Yifu Ding, Jiacheng Wang, Ge Yang, Yongcheng Jing, Jinyang Guo, Xianglong Liu, Dacheng Tao
帰属誘導とカバレッジ最大化による構造的MoE圧縮のためのプルーニング
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

混合エキスパート(MoE)モデルは、一部のエキスパートのみを活性化することで計算を効率的に拡張できるが、その大きなメモリフットプリントと推論オーバーヘッドのため、展開コストは依然として高い。従来の圧縮手法は主にエキスパートレベルで動作し、エキスパート全体を削除するか、粗粒度の重要度スコアでエキスパートをランク付けしていた。しかし、このようなエキスパート単位の決定はしばしば粗すぎて細粒度の冗長性を捉えられず、プルーニング予算の配分ミスと限定的な圧縮につながっていた。

この問題に対処するため、Yifu Dingら7名の研究者は重要な観察を行った。それは、MoEエキスパート内の情報は少数のチャネルに高度に集中しており、重要と見なされるエキスパートでさえも相当な冗長性が残っているというものである。この発見は、圧縮の粒度をエキスパートレベルからチャネルレベルへと細分化し、冗長なパラメータをより正確に特定・除去できることを意味する。

以上の観察に基づき、研究チームはMoEモデルに特化した構造的プルーニングフレームワークを提案した。この手法の核心的な革新は、プルーニング比率の割り当て問題をチャネルスコアのカバレッジ最大化問題として再定式化し、帰属ベースの近似手法を用いて効率的に解く点にある。具体的には、各エキスパートに独立してプルーニング比率を割り当てるのではなく、グローバルなチャネル重要度のカバレッジという観点から、プルーニング後のモデルが重要な情報チャネルを最大限に保持できるようにする。

実験面では、この手法はDeepSeekおよびQwenシリーズのMoEモデルで検証された。結果として、4ビット量子化と組み合わせた場合、50%または25%の構造的プルーニング比率でもモデル精度が維持されることが示された。特に注目すべきは、Qwen3-30B-A3Bモデルにおいて、このアプローチがメモリフットプリントを5.27倍削減し、多様なベンチマークテストで現在の最先端ベースラインを一貫して上回った点である。

本研究は、エキスパートを単純に削除するのではなく、より細粒度の構造的プルーニングを通じてモデル内部の冗長性を掘り下げるという、MoEモデルの効率的な展開への新たな道を開くものである。論文は全9ページ、5枚の図を含み、ICML 2026に投稿済みである。全文はarXivで公開されている。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • MoEの情報は少数のチャネルに高度に集中しており、重要なエキスパートでも冗長性が大きいため、エキスパート単位のプルーニングは粗すぎる。
  • プルーニング比率の割り当てをチャネルスコアのカバレッジ最大化問題としてモデル化し、帰属ベースの近似で解く。
  • DeepSeekとQwenのMoEモデルで、50%または25%の構造的プルーニングと4ビット量子化を組み合わせても精度を維持。
  • Qwen3-30B-A3Bでメモリフットプリントを5.27倍削減し、多様なベンチマークで最先端ベースラインを上回る。

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。

帰属誘導とカバレッジ最大化による構造的MoE圧縮のためのプルーニング | AI News Hub