CARE:不確実さに応じて専門家を割り当てるMixture‑of‑Experts用の自信適応ルーティング
この論文は、トークンごとに必要な計算量を変えることで効率を上げる方法を示します。従来のMixture‑of‑Experts(専門家集合、MoE)を使ったLow‑Rank Adaptation(LoRA)は、各トークンに対して固定数の専門家kを割り当てます。だがトークンごとに「簡単」か「難しい」かが違うため、固定のkでは簡単なトークンに無駄な計算を使い、難しいトークンには十分な処理が行き届かないことがあります。著者らはルーターの出力分布を「トークン毎の自信の信号」と見なし、それを使って必要な専門家だけを動かす手法を提案しています。
提案手法はCARE(Confidence‑Adaptive Routing of Experts)と呼ばれます。ルーターが出す専門家ごとの重みを確率分布と見なして、重みの高い順に専門家を順に「受け入れ」ます。受け入れは累積確率がある閾値に達するまで続き、その結果として自信が高ければ少数の専門家で済み、自信が低ければより多くの専門家が使われます。受け入れた専門家同士で出力に食い違い(不一致)がある場合は少しだけ余分に受け入れる拡張も加えます。さらに「予算サーモスタット」と呼ぶ仕組みで、この閾値を調整し平均的に使う専門家数が任意の目標に合うようにします。CAREは追加の学習パラメータを必要とせず、単一の順伝播(single forward pass)で動作するため、既存のMoE‑LoRAにそのまま差し替え可能です。
検証では、著者らはLLaMA‑3.1‑8BとQwen2.5‑7Bというモデル上で、常識推論の8つのベンチマークや数学、コード、知識系のタスクを使って評価しました。抽象によれば、同じ計算コストに合わせた固定top‑kのMoE‑LoRAと比べてCAREは性能を改善しました。また、固定k=4のベースラインと同等の性能を維持しつつ、実際に動かす専門家数はより少なくできたと報告しています。加えて、ルーターの「自信」と「不一致」の信号は、単純最大確率(MSP)、エントロピー、あるいは複数回推論する近似法よりも外れ値検出(分布外データの検出)を改善することが示されています。
なぜ重要かというと、CAREは計算資源を必要な箇所に回す仕組みだからです。簡単な入力に無駄な専門家を使わず、難しい入力にはより多くの専門家を充てることで、同じ計算量で性能を上げられます。さらに追加の学習やモデル変更が不要で、既存のMoE‑LoRA実装に容易に組み込める点も実用上の利点です。外れ値検出の改善は安全性や信頼性の面でも有用になり得ます。
重要な注意点もあります。ここで紹介している内容は論文の要約に基づくもので、示されている評価は特定のモデル(LLaMA‑3.1‑8B、Qwen2.5‑7B)と限られた種類のタスク群での結果です。別のモデル規模や別のタスク群で同様の利得が得られるかは、追加の検証が必要です。著者らは設計の正当性を「核(nucleus)忠実度」「予算最適性」「不一致の認識論的解釈」で支え、コードも公開していると述べています。