Auditable Session Admission for Cross-Silo Federated Learning

技術要約：クロスサイロ型連合学習における監査可能なセッション・アドミッション

問題提起
クロスサイロ型連合学習（FL）は、生データをローカルに保持したまま共有モデルを訓練することを可能にする。しかし、その展開においては、特定のセッション範囲の操作（例：参加、モデルの取得、更新の送信）を実行する権限が誰にあるかを決定する「セッション・アドミッション（セッション承認）」に関するガバナンスのボトルネックにより、プロジェクトが停滞することが頻繁にある。現在の実装は、中央集権的なポリシーサービス、プラットフォーム固有の設定、あるいはオーケストレーション・コード内に埋め込まれたアドホックなチェックといった、継ぎ接ぎの仕組みに依存している。これらの手法には、以下の3つの決定的な欠陥がある。

1. 非ポータビリティ（移植性の欠如）： 組織間でアイデンティティやデプロイメントのスタックが異なるため、統合が個別のカスタマイズを必要とする。
2. 監査性のギャップ： 意思決定が設定の中に分散しているため、実行後に「なぜその操作が許可されたのか」を事後的に再構成することが困難である。
3. 脆弱性（ブリットルネス）： 運用のドリフト（乖離）によって、実行時の決定記録を残すことなく、アドミッションの挙動が密かに変化してしまう可能性がある。

既存のアイデンティティおよびアクセス管理（IAM）ソリューション（例：OAuth、RBAC）は、認証とインフラストラクチャへのアクセスは処理するが、FL操作のためのポータブルでセッション範囲の限定されたアドミッション・セマンティクス、および境界検証可能なエビデンス（証拠）は提供しない。

手法
本論文では、クロスサイロの境界において、事前承認されたセッション・ケーパビリティ（機能）を強制するセッション・アドミッション・ゲートウェイを提案する。このメカニズムは、セッションのセットアップ（ステートフル）と実行時のアドミッション（ステートレス）を分離し、主に2つの暗号学的アーティファクトに依存する。

1. セッション・ケーパビリティ・トークン (SCT):

   • セッションのセットアップ時に、組織の管理者が発行（ミント）する。
   • 特定の session_id と保持者のアイデンティティ (sub) に紐付けられた署名済みアーティファクト（JOSEフレームワークを用いたJWTとして実現）。
   • 許可された操作（例：fetch model、submit update）および制約（例：時間枠、ラウンド数）を列挙する。
   • 保持者の公開鍵のサムプリントを含む cnf.jkt クレームを含み、所有証明（PoP）検証を行う。

2. リクエストに紐付いた所有証明 (PoP):

   • トークンの再利用や盗難によるなりすましを防ぐため、すべてのリクエストにはPoP署名が含まれる。
   • DPoP (RFC 9449) の手法を用い、クライアントはリクエストの標準的な表現（メソッド、エンドポイント、session_id、nonce、タイムスタンプ、およびボディのハッシュ）に対して、自身の秘密鍵を用いて署名を行う。
   • ゲートウェイは、PoP署名がSCT内の保持者アイデンティティ (cnf.jkt) に対応していること、およびリクエストが最新であることを検証する。

アドミッション・ワークフロー

• セットアップ: 管理者がロールと制約を承認し、特定の参加者に向けたSCTを発行する。
• 実行時: クライアントはSCTとPoPを伴うリクエストを送信する。アドミッション・ゲートウェイは、以下のステートレスな検証を行う：
  1. SCT検証: 署名、発行者の信頼アンカー、有効期限、および必須フィールドを検証する。
  2. ケーパビリティ・マッチング: リクエストされたエンドポイントを抽象的な操作にマッピングし、それが session_id に紐付いたSCTによって許可されているかを確認する。
  3. PoP検証: 標準的なリクエスト文字列を再構成し、保持者の公開鍵に対する署名を検証し、鮮度（nonce/タイムスタンプ）を確認する。
• エビデンス生成: すべての決定（ALLOW/DENY）は、操作、session_id、理由コード、タイムスタンプ、およびトークンとリクエストのハッシュを含む構造化されたJSONレコードを出力する。これらのレコードはエビデンス・ログに追記され、実行中のアーティファクト（モデルのチェックポイント、メトリクス）と相関付けられる。

主な貢献

1. セッション境界のアドミッション: リクエストが session_id に紐付いた署名済みケーパビリティを介して承認され、組織内部のIAMとは独立して境界で検証される、FLネイティブなアドミッション層。
2. 証明を伴うリクエスト: PoPを通じて各リクエストを保持者キーに紐付けることで、盗まれたトークンによるなりすましを防ぎ、リクエスト単位のアドミッションを可能にする。
3. 監査可能な決定記録: 実行時のアーティファクトと相関付けられた構造化決定記録を生成することで、一時的なプラットフォーム構成の再構築なしに、事後的な監査を可能にする。

結果と評価
著者らは、Flower FLフレームワークと、代替ワークロードとしてのMNISTデータセットを用いた、コンテナ化されたオープンソースのプロトタイプを使用して提案手法を検証した。システムは、ポジティブパスおよびネガティブケース（ケーパビリティの不一致、PoPの不一致/リプレイ）にわたってテストされた。

• 正確性: ゲートウェイは、トークンの検証、ケーパビリティのマッチング、またはPoPチェックに失敗したリクエストを拒否し、セッション範囲の権限を正常に強制した。
• パフォーマンス: 単一ホスト（Intel Xeon）上でのマイクロベンチマークにより、エンドツーエンドのアドミッション遅延はサブミリ秒（中央値 約0.261 ms for ALLOW）であることが示された。オーバーヘッドは暗号学的検証（SCTおよびPoP署名の検証）によって支配されており、ケーパビリティのマッチングによる影響は無視できる程度であった。
• 再現性: プロトタイプには、再現可能なテスト、エビデンスログ、およびアーティファクト・ストアが含まれており、実行後のアドミッション決定の再検証が可能である。

意義と主張
本論文は、セッション・インターフェースに対するクロスサイロ型FLのガバナンスにおける「実用的なボトルネック」を、リクエストに付随するエビデンスとしてアドミッション・セマンティクスを外部化することで解決すると主張している。その意義は以下の点にある：

• ポータビリティ: このメカニズムは、既存のフェデレーテッド・アイデンティティやクラウドIAMを置き換えるのではなく、それらと構成（コンポーズ）できるため、異種混合の組織スタックにも適応可能である。
• 監査性: 特定の実行に紐付いた決定論的かつ暗号的なアクセス・トレイルを提供し、決定が構成のドリフトによって失われるという「監査のギャップ」を解決する。
• モジュール性: アドミッション層を学習プロセスから隔離しているため、プライバシーや堅牢性のメカニズム（例：セキュア集計）を修正することなく、その上にレイヤーとして重ねることができる。

著者らは、本研究がプライバシー保護メカニズム（例：差分プライバシー）や悪意のあるモデル更新に対する堅牢性には対処しておらず、あくまでクロスサイロ型の権限付与に関するデプロイ可能性と監査性に焦点を当てていることを明示している。