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

端末性能とデータの違いに合わせて連合学習を分割

FedSAP: Federated Learning with Structured Adaptive Partitioning for Multi-Domain Heterogeneous Edge Devices

Wentao Yue, Tianyou Lai, Hongji Li, Qingyu Mao, Qilei Li

この論文をやさしく読む

ひとことで言うと

連合学習でモデルのチャネルを、全体共有、似たデータを持つ端末間だけの共有、削除の三つへ振り分ける方法です。

何に役立つ?

計算資源にもデータにも違いがある端末群で、モデルを小さくしつつ共同学習するための設計に役立ちます。

この研究の面白いところ

削る量だけを決めず、何を残し、誰と共有するかまで一緒に決めます。勾配の似かたからデータの擬似的なドメインも推定します。

どこまで分かった?

具体的な結果はDigitsとOffice-Caltechで、乱数シード三つにわたる評価です。最大80%の枝刈り対応という結果から、あらゆる端末やデータで同じ精度を保てるとは言えません。

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

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

異種のエッジ端末で連合学習(FL)を行うには、端末ごとに異なる資源予算と、ドメインがずれたローカルデータの両方に対応する必要がある。既存の資源適応型手法は、各クライアントがモデルのどれだけを学習するかは決めるが、残す容量をどこに置き、どのように共有すべきかは決めない。一方、連合ドメイン汎化の手法は通常、完全なアーキテクチャを共通に使うことを前提とする。そのため、一様な圧縮は有用性の高いチャネルを捨ててしまう可能性があり、単一の集約経路は、転用可能な特徴とドメインに敏感な更新を混ぜてしまう。 本研究では、ドメインを考慮した異種端末向けFL枠組みFedSAPを提案する。FedSAPは、構造化枝刈りを、予算制約下での三状態のチャネル割り当てとして定式化する。各保持率を層ごとの非一様な予算へ変換し、安定したチャネルをGlobalプール、有用でドメインに敏感なチャネルを擬似ドメイン別のPrivateプール、有用性が低いチャネルをDropped状態へ割り当てる。この分割により、広く役立つ特徴はクライアント間の共有から利益を得る一方、ドメインに敏感な更新は適合しないクライアントから切り離される。Domain-Guided Assignmentは浅い層の勾配の類似性から擬似ドメインを推定し、Type-Matched Aggregationは各チャネルの共有範囲を、その目的に合う範囲へ制限する。 三つの乱数シードにわたる評価で、FedSAPの平均グローバル正解率はDigitsで76.00%、Office-Caltechで72.67%に達し、最も強いベースラインをそれぞれ1.70ポイントと4.92ポイント上回った。同時に、異種クライアントにおいて最大80%のクライアント枝刈り率に対応した。

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

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

更新履歴

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

原文の要旨

Federated learning (FL) on heterogeneous edge devices must jointly accommodate unequal resource budgets and domain-shifted local data. Existing resource-adaptive methods decide how much of a model each client trains but not where retained capacity should reside or how it should be shared, whereas federated domain-generalization methods usually assume a shared full architecture. Uniform compression can therefore discard high-utility channels, and a single aggregation path can mix transferable features with domain-sensitive updates. We propose FedSAP, a domain-aware heterogeneous FL framework that casts structured pruning as budget-constrained tri-state channel allocation. FedSAP converts each keep ratio into non-uniform layer budgets, assigns stable channels to a Global pool, useful domain-sensitive channels to pseudo-domain-specific Private pools, and low-utility channels to a Dropped state. This partition lets broadly useful features benefit from cross-client pooling while isolating domain-sensitive updates from incompatible clients. Domain-Guided Assignment infers pseudo-domains from shallow-gradient similarity, while Type-Matched Aggregation restricts each channel to its intended sharing scope. Across three random seeds, FedSAP reaches 76.00% and 72.67% mean global accuracy on Digits and Office-Caltech, exceeding the strongest baseline by 1.70 and 4.92 percentage points while supporting client pruning ratios of up to 80% across heterogeneous clients.

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