アーカイブ / LLM
LoRA専門家を小さな行列に集約するCoMoL:トークン単位の切り替えと評価結果
CoMoL
CoMoLの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1: Comparison of LoRA-based methods. (a) LoRA injects trainable low-rank matrices 𝑨 \boldsymbol{A} and 𝑩 \boldsymbol{B} into transformer layers to approximate weight updates. (b) MoE-LoRA employs multiple LoRA experts with a routing network to select and activate experts based on input tokens. (c) CoMoL fuses expert parameters in the low-rank core space, preserving token-level routing while maintaining expert-specific expressiveness with minimal parameters.
概要
- CoMoLは共有の変換行列と専門家固有の小さなコア行列を使い、トークンごとに専門家を重み付けして統合する。
- 著者らのQwen3-8B数学評価では、六データセットの平均正解率がCoMoL 84.48±1.40%、通常のLoRA 82.78±0.47%だった。
- 社内コード生成への応用には、仕様と期待動作のデータを用意し、通常のLoRAと同条件でテスト通過率や時間を検証する必要がある。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、CoMoLの仕組み
- 01入力トークン
- 02低ランク投影
- 03専門家の重み算出
- 04コア行列を統合
- 05補正後の内部表現
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
CoMoLは、大規模言語モデルを仕事に合わせて調整する際、複数の専門家を少ない学習対象で使い分ける方法だ。LoRAは元の重みを固定し、小さな行列だけを学習して計算結果を補正する。専門家ごとにLoRAの行列一式を持つと、専門家の数に応じて学習対象が増える。
入力文全体で同じ専門家を使う方式では、文中の処理単位ごとに選択を変えられない。CoMoLは、トークンという語や語の一部に相当する単位ごとに、専門家の組み合わせを変える。
手法
CoMoLは、各トークンの内部表現を少数の次元に写し、その結果から専門家ごとの重みを求める。この写し方を低ランク投影と呼ぶ。各専門家は、その小さな空間に置くコア行列を持ち、投影に使う行列は専門家間で共有する。
次に、重みに応じてコア行列を足し合わせ、一つの行列として元のモデルの計算に補正を加える。例えば関数の仕様を入力した場合、仕様中の各トークンで異なる重みを計算し、得られた内部表現を後続の語の生成に使う。専門家の出力を個別に高次元へ戻してから足す方式とは、統合する場所が異なる。
新規性
CoMoLは、専門家ごとの差分を小さなコア行列だけに持たせ、入力側と出力側の変換行列を共有する。著者らは、専門家の数に伴う学習対象パラメータの増加を抑える設計として示した。
さらに、各トークンの選択と行列の統合を小さな空間で行う。選択器も入力の低ランク投影結果を再利用するため、トークン単位の切り替えに必要な追加パラメータを減らせる。
従来手法との違い
通常のLoRAは一組の小さな行列で元のモデルを補正する。従来のMoE-LoRAは専門家ごとにその行列を持つが、CoMoLは共有行列と専門家固有のコア行列に分ける。著者らの計算量表では、CoMoLの専門家部分はパラメータ数、演算量とも通常のLoRAと同等の水準とされる。
比較対象のSMEARは入力例全体で一度専門家を統合するが、CoMoLはトークンごとに統合する。ただし、計算量表には選択器の負担が含まれないため、実際の処理時間まで同等とは読めない。
実験結果
著者らはMath14kで学習し、六つの数学データセットで正解率を測った。Qwen3-8Bでの平均はCoMoLが84.48±1.40%、通常のLoRAが82.78±0.47%で、学習対象はそれぞれ25.16M、24.77Mパラメータ。Qwen3-14Bでは86.34±1.01%、85.18±0.86%だった。数学の値は三つの乱数設定による平均と標準偏差である。
コード生成はCodeAlpaca-20kで学習し、HumanEvalのpass@k、つまりk回の生成で少なくとも一つ正解する確率を測った。pass@1はQwen3-8Bで48.11%対39.69%、Llama3.1-8Bで35.00%対26.22%で、いずれもCoMoLとLoRAの順。検証用データでチェックポイントを選び、学習は順に5エポック、1エポック行った。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案:社内の開発支援で、短い関数の仕様からコード案を作る用途を試す。入力は仕様、入出力例、関連する既存コードとし、出力は関数案と確認用のテスト項目に絞る。開発者が動作を確かめて採用を判断する。
適応学習には、利用条件を確認したコード指示データと、正解コードまたは期待動作を示すラベルが要る。入力はテキストなので追加センサーは想定しない。社内の命名規則や依存関係に合わせた調整も必要で、HumanEvalの結果だけでは社内での効果を判断できない。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案:関数仕様、期待する動作を示すラベル、実行できるテストを対応付け、利用条件を確認した学習用と評価用のデータを用意する。入出力例や失敗時の扱いも記録する。社内コードに合わせて適応学習する計画とし、追加センサーは用いない。
同じ基礎モデルとデータ分割で通常のLoRAを比較対象にする。テスト通過率、学習対象パラメータ数、生成時間を測り、類似した関数が学習側と評価側にまたがらないか確認する。複数回の結果のばらつきと、開発者が修正に要した時間も記録して判断する。
限界と課題
著者らが明示する課題は、従来のMoE-LoRAで専門家が増えるとパラメータも増えることと、SMEARでは入力例単位の選択にとどまることだ。確認できた本文の範囲では、CoMoL自体の適用限界を体系的に列挙した記述はない。
編集上の検証事項は、選択器とコア行列の統合を含む処理時間、メモリ使用量、専門家数を変えた場合の挙動である。示された評価は数学問題とコード生成に限られる。社内の長いコード変更や固有の依存関係でも同様に使えるかは、別途確かめる必要がある。
出典
元のタイトル・要旨を確認する(英語)
CoMoL: Efficient Mixture of LoRA Experts via Dynamic Core Space Merging
Large language models (LLMs) achieve remarkable performance on diverse downstream and domain-specific tasks via parameter-efficient fine-tuning (PEFT). However, existing PEFT methods, particularly MoE-LoRA architectures, suffer from limited parameter efficiency and coarse-grained adaptation due to the proliferation of LoRA experts and instance-level routing. To address these issues, we propose Core Space Mixture of LoRA (\textbf{CoMoL}), a novel MoE-LoRA framework that incorporates expert diversity, parameter efficiency, and fine-grained adaptation. Specifically, CoMoL introduces two key components: core space experts and core space routing. Core space experts store each expert in a compact core matrix, preserving diversity while controlling parameter growth. Core space routing dynamically selects and activates the appropriate core experts for each token, enabling fine-grained, input-adaptive routing. Activated core experts are then merged via a soft-merging strategy into a single core expert, which is combined with a shared LoRA to form a specialized LoRA module. Besides, the routing network is projected into the same low-rank space as the LoRA matrices, further reducing parameter overhead without compromising expressiveness. Extensive experiments demonstrate that CoMoL retains the adaptability of MoE-LoRA architectures while achieving parameter efficiency comparable to standard LoRA, consistently outperforming existing methods across multiple tasks.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01




