連合学習において一部のクライアントに異常や悪意がある場合にも高精度にAIモデルを学習可能な手法を開発～LLM tsuzumi の学習への適用やIOWN機能としての実用化を目指す～

日本電信電話株式会社（本社：東京都千代田区、代表取締役社長：島田 明、以下「NTT」）と株式会会社NTTデータ数理システム（本社：東京都新宿区、代表取締役社長：箱守 聰、以下「NTTデータ数理システム」）は、複数のクライアント(個人や組織)でデータを保持したまま分散して学習する連合学習において、一部のクライアントに異常や悪意がある場合にも高精度にAIモデルを学習可能な学習手法を開発しました。クライアント間で統計的なデータの偏りがある場合において、耐障害性が最も優位なアルゴリズムです。 個人情報などの機微なデータを使って複数者で学習する場合、データを手元に置いたまま安全に学習可能な連合学習が注目されています。しかし、異常な動きをする参加者がいた場合に、連合学習で得られるモデルの精度劣化を引き起こすことが知られています。 本技術（モメンタムスクリーニング技術※1）により、すべての参加者が正常であることを前提とせず、高精度なAIモデルを分散して構築できるようになり、安全かつ安定した学習の実現ができます。本技術は機械学習分野の三大トップカンファレンスの 1 つである International Conference on Learning Representation 2024 (ICLR2024：5月7日～11日)に採択され、オーストリア・ウィーンで発表予定です。1. 背景 近年、組織や分野横断でデータを活用することで、新たな価値を創造することが大きく期待されています。例として、社内文書を活用した業務DXや希少疾患の医療診断を支援するAIモデルの構築などが挙げられます。 しかし、扱われるデータは個人情報や機密情報を含むことがあり、情報漏洩のリスクなどの観点から組織間でのデータの共有は進んでいません。そこで、データを共有せず、それぞれのデータを用いて学習されたAIモデルを共有することで、それぞれのデータの特徴を反映した一つのモデルを作成する技術として連合学習が注目されています。 図1のように、連合学習では、複数のクライアントがそれぞれのデータを使ってローカルモデルを学習し、中央サーバは複数のローカルモデルの平均化などの処理により、単一のグローバルモデルを作成するプロセス(ラウンドと呼ぶ)を繰り返すことでAIモデルを学習します。[画像1]https://digitalpr.jp/simg/2341/87612/700_332_2024050108115266317ab885e84.JPG[画像出典] 連合学習とは？Federated Learningの基礎知識をわかりやすく解説（※2） 連合学習の課題として、学習に参加するクライアントの一部が異常あるいは悪意を持って、正常なクライアントとは大きく異なる学習モデルを共有した場合に、連合学習自体が困難になることが挙げられます。悪意のあるクライアントがいる場合でも、悪意のあるクライアントがいない場合と同様に、連合学習自体はうまくいくことが求められ、それが実現されるアルゴリズムはビザンチン耐障害性があるとされます。ビザンチン耐障害性は、ラベル付のミスや、計測エラーによる学習データの品質劣化に対して効果を発揮します。また、連合学習に不特定多数が参加し、その中でクライアントの乗っ取りなど悪意のあるクライアントが想定される場合にも有効です。[画像2]https://digitalpr.jp/simg/2341/87612/700_346_2024050108115266317ab862ac1.JPG2. 研究の成果 ビザンチン耐障害性のある連合学習手法では、全クライアントのうち悪意のある行動(実際とは異なる学習モデルの共有など)をするクライアントが50% 未満である場合、悪意のあるクライアントがどのような行動をしたとしても、その手法の精度の低下を小さくすることができます。 連合学習における既存のビザンチン耐障害性のある従来手法として、CClip※3が挙げられます。この手法は連合学習の各ラウンドで、正常なクライアントの勾配(モデルの更新方向)の平均推定量を用いて、全てのクライアントの勾配の影響が小さくなるように補正(Clipping)を行います。これにより悪意のあるクライアントが学習全体に与える影響が小さくなり、ビザンチン耐障害性を得ていました。しかし、各クライアントが持つデータの統計的性質が大きく異なる際には、正常なクライアントの学習も悪意のあるクライアント同様に補正の影響を受けてしまい、学習したモデルの精度が下がる欠点がありました。 本技術では、ラウンドごとではなく、過去のラウンドでの学習情報を適切に用いて、正常および悪意あるクライアントの判定を行うことで、正常なクライアントの学習には補正が加わらず、そのことにより全体で学習したモデルの精度が下がることなく, よりよいモデルを生成することができます。3. 技術のポイント モメンタムスクリーニング技術のポイントは以下の通りです。モメンタム：学習ごとの勾配ではなく、過去の勾配の指数平均を用いて学習スクリーニング：各クライアントのモメンタム同士を比較し、近いモメンタムの割合を閾値以上であれば正常、そうでなければ異常と判断 モメンタムは図3のように過去の勾配の指数平均と勾配を加算し、勾配の代わりに用いることです。スクリーニングは図4のように各クライアントのモメンタム同士を比較し、近いモメンタムが閾値以上あれば正常、そうでなければ異常と判断することです。異常と判断されたモメンタムの情報は使わずにモデル全体の更新を行います。[画像3]https://digitalpr.jp/simg/2341/87612/700_281_2024050108115266317ab87acf8.JPG このようにモメンタムにスクリーニングを適用し、精度の高いモデルを学習することができます。また、数理的な証明により、一般的なクライアント間のデータの統計的な偏りを想定した目的関数において、CClipと比べて学習誤差を小さくできることを確認し、さらに、提案手法を上回る手法が存在しないことを確認しました※4。また図5に示すように、画像認識のベンチマークテストを通じて、ビザンチンノードが存在する状況下で、いくつかの従来手法(CClipを含む)よりも、異常挙動に対する最悪ケースの比較において高いテスト認識率のモデルを学習できることを実験的に確認しました。[画像4]https://digitalpr.jp/simg/2341/87612/700_243_2024050108115266317ab86a05c.JPG4. 役割NTT:ビザンチン耐性のあるアルゴリズムの考案NTTデータ数理システム：主に考案手法の数理的およびシミュレーションにおける正確さの検証実施5. 今後の取り組み NTTおよびNTTデータ数理システムは、情報の保護と活用を両立する技術として期待される連合学習技術の開発を通じて、組織横断的に分散蓄積されるデータを用いたAI等の利活用の促進に貢献するとともに、共同実験などを通じて連合学習技術の研究を邁進していきます。また、将来的にはNTTが研究するLLM tsuzumi の学習への適用やNTTが推進するIOWN PETs ※5機能として実用化することを目指します。〈参考・用語解説〉※1. T. Murata, K. Niwa, T. Fukami, and I. Tyou, &ldquo;Simple Minimax Optimal Byzantine Robust Algorithm for Nonconvex Objectives with Uniform Gradient Heterogeneity,&rdquo; ICLR 2024.URL：https://openreview.net/forum?id=1ii8idH4tH※2. 連合学習とは？Federated Learningの基礎知識をわかりやすく解説URL：https://www.msiism.jp/article/federated-learning.html※3. S. P. Karimireddy, L. He, and M. Jaggi, &ldquo;Byzantine-Robust Learning on Heterogeneous Datasets via Bucketing,&rdquo; ICLR 2022.URL：https://openreview.net/forum?id=jXKKDEi5vJt※4. 仮定した目的関数のクラスにおいて、提案手法がミニマックス最適性を満たすことを証明。注釈※1参照※5. IOWN PETsURL：https://www.rd.ntt/sil/project/iown-pets/iown-pets.html関連リンク連合学習とは？Federated Learningの基礎知識をわかりやすく解説https://www.msiism.jp/article/federated-learning.html