性能の異なる端末のCPUで言語モデルを協調微調整
Xronos: Heterogeneity-Aware Tensor Parallelism for Collaborative LLM Fine-Tuning on Edge CPUs
この論文をやさしく読む
ひとことで言うと
速さの違うCPU搭載端末で言語モデルを共同調整するとき、各端末への計算配分を工夫して待ち時間を減らす方法です。
何に役立つ?
データを各端末に置いたまま、ゲートウェイなどGPU中心ではない機器で共同学習を行う場合の効率化に役立ちます。
この研究の面白いところ
CPUで計算と通信が競合する問題に着目し、テンソル並列と軽い性能測定を使って機器ごとの担当量を変えます。評価では既存TPに対し18%、PPに対し56%の時間短縮を報告しています。
どこまで分かった?
複数の端末・モデル・課題で精度を維持し、アイドル時間の割合も約5.9倍改善した結果です。要旨には各構成の詳細や消費電力の評価はなく、すべての端末構成で同じ短縮率になるとはいえません。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
エッジ端末での協調的な微調整は、各端末のデータをその端末内に保ちながら、大規模言語モデルを分野固有のデータへ適応させる。最先端の協調微調整手法は主にGPUを搭載したエッジ端末向けに設計され、パイプライン並列(PP)に依存する。しかし、IoTゲートウェイ、スマートホームハブ、車載コンピューターなど、多くのエッジ基盤はCPUを主体としている。 本論文では、CPU主体のエッジ端末でPPが有効に働かないことを報告する。同じCPUがモデル計算と通信の両方を処理するため、CPU資源の競合が深刻になるからである。解析では、この結果、計算が停止している割合がGPU端末より平均5.75倍高くなる。テンソル並列(TP)は計算と通信を分けて競合を緩和できるが、既存のTP手法は端末の性能が均一だと仮定している。性能の異なるCPU端末では、この仮定のため、高速なワーカーが同期点で低速端末を待ち、最大34%の時間を待機状態で過ごすことが分かった。 この制約に対処するため、異種CPUエッジ端末向けの協調微調整基盤Xronosを提案する。XronosはTPを実行の基盤とし、軽量な性能計測と、端末の異質性を考慮したテンソル分割を組み合わせて、遅い端末によるボトルネックを減らす。多様な端末、モデル、ベンチマークタスクにおいて、精度を維持しながら、最先端手法に比べて微調整時間をTP比で18%、PP比で56%短縮し、端末の待機時間の割合を約5.9分の1にする。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-17(UTC)
- 最新改訂
- 2026-09-17 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-17 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Collaborative fine-tuning on edge devices adapts large language models to domain-specific data while keeping each device's data local. State-of-the-art (SOTA) collaborative fine-tuning techniques are largely designed for GPU-based edge devices and rely on pipeline parallelism (PP). However, many edge platforms, including IoT gateways, smart-home hubs, and in-vehicle computers, are primarily CPU-based. This paper reports that PP is ineffective on CPU-based edge devices because the same CPU handles both model computation and communication, which causes severe CPU contention. Our analysis shows that this leads to 5.75$\times$ higher computation stall ratios than on GPU devices on average. Tensor parallelism (TP) can alleviate this contention by separating computation and communication, but existing TP techniques assume homogeneous devices. On heterogeneous CPU edge devices, we find that this assumption causes faster workers to remain idle for up to 34% while waiting for slower devices at synchronization points. To address the limitations, we propose Xronos, a collaborative fine-tuning framework for heterogeneous CPU edge devices. Xronos uses TP as its execution backbone and combines lightweight profiling with heterogeneity-aware tensor partitioning to reduce the straggler bottleneck. Across diverse devices, models, and benchmark tasks, Xronos reduces fine-tuning time by 18% (TP) to 56% (PP) and the ratio of device idle time by $\sim$5.9$\times$ over SOTA techniques, while maintaining the accuracy.
arXiv ID: 2609.19909 / 要約の誤りについて