AI・機械学習
上級

スプリット学習(LLM向け分割学習)(スプリットガクシュウエルエルエムムケブンカツガクシュウ)

LLMを複数の層グループに分割し、各パーティが担当層のみを保持して学習する分散プライバシー保護手法。フェデレーテッド学習と異なり中間アクティベーションのみを交換し、モデル全体を各クライアントに配布しない。

0 回閲覧
0 いいね
2026/6/7 更新
関連タグ
split-learning
llm
privacy
distributed-training
federated-learning

スプリット学習(Split Learning)は、LLMをネットワークの途中(カットレイヤー)で分割し、複数パーティが各担当部分を保持したまま協調学習する手法。フェデレーテッド学習との最大の違いは「モデル重みをクライアントに配布しない」点であり、大規模LLMを計算能力の低いクライアント端末でも部分的に活用できる。

スプリット学習の動作原理

[クライアント(病院端末)]
  入力データ x
    ↓ 前半層(Layer 1〜L_cut)で順伝播
  スマッシュドデータ h(中間アクティベーション)
    ↓ ネットワーク経由で送信

[サーバ(クラウド)]
  スマッシュドデータ h を受信
    ↓ 後半層(Layer L_cut+1〜N)で順伝播
  損失計算 → 逆伝播 → ∂L/∂h をクライアントへ送信

[クライアント]
  受け取った勾配で前半層を更新

元データ x はクライアントから出ない。サーバが見るのはスマッシュドデータ h のみ。

カットレイヤーの選択

カットレイヤーの位置は精度・プライバシー・通信コストのトレードオフを決定する。

カット位置クライアント計算量通信データ量プライバシー
第1層後最小大(高次元h)低(hが入力に近い)
中間層中中中
最終層前最大小(低次元h)高(hが抽象的)

Llama 3 8B(32層)では8〜12層でカットが実用的とされる(2024年論文)。

フェデレーテッド学習との比較

比較軸Split LearningFederated Learning
モデル配布なし(サーバが後半を保持)全クライアントにモデル配布
クライアント計算前半層のみ全モデル学習
通信内容スマッシュドデータ+勾配モデル重み差分
GPU要件(クライアント)低(前半層のみ)高(全モデル)
プライバシー中(hからの復元リスクあり)中(モデル反転攻撃リスク)
通信効率低〜中(逐次通信)高(非同期OK)

プライバシーリスクと対策

スマッシュドデータからの情報漏洩

中間アクティベーション h からの入力復元攻撃(Feature Reconstruction Attack)が報告されている。

対策:

  • Noisy Smashed Data: hにガウスノイズを加算(精度劣化と引き換え)
  • Dropout on Smash: カットレイヤー直前にDropout(p=0.5)を追加
  • SMPC(Secure Multi-Party Computation): hを暗号化して送信(計算コスト大)
  • Adversarial Perturbation: hに敵対的摂動を加えながら精度維持

2025年の主要研究

論文/システム内容効果
SplitFed (AAAI 2023)Split + Fedを組み合わせ両者の弱点を補完
PrivateSplit (2024)DP付きスプリット学習ε=6で医療精度-5%のみ
LLM-Split (NeurIPS 2024)Llama 3への大規模適用32層→8/24分割で精度95%維持
SplitAdapter (2025)LoRA + スプリット学習クライアント計算量50%削減

よくある質問(FAQ)

Q1: スプリット学習はフェデレーテッド学習より良いのか? A: 用途次第。クライアントのGPUが弱い場合(スマホ・IoT端末)はスプリット学習が有利。クライアントが高性能GPUを持つ企業間連合(病院間連携等)はフェデレーテッド学習が通信効率で有利。

Q2: スプリット学習は並列処理できるか? A: 基本のスプリット学習は逐次処理でクライアントAが学習中はクライアントBが待機する。SplitFedやU-ShapeなどのバリアントでN:1並列化が可能だが実装が複雑になる。

Q3: カットレイヤーをどう選ぶべきか? A: クライアントのGPUメモリとネットワーク帯域で決まる。前半層が多いほどクライアントGPU使用量増・通信量減。一般的にはLlama 3 8Bで8〜12層が実用的とされる。

まとめ

  • スプリット学習 = LLMを2部分に分割し、入力データをクライアントに保持
  • クライアントGPU要件が低くIoT・スマートフォン向けに有利
  • スマッシュドデータからの情報漏洩リスクがあり、ノイズ/SMPCで対策
  • 2025年はLoRA + スプリット学習のハイブリッドが研究最前線
この記事について
カテゴリーAI・機械学習
難易度上級
作成日2026/6/7