arXiv reaDer
分散システムにおける教師なし表現の一般化と個人化の学習
More Generalized and Personalized Unsupervised Representation Learning In A Distributed System
対照学習などの教師なし学習の判別方法は、集中化されたデータの一般化された視覚的表現を学習する能力を実証しています。それにもかかわらず、ユーザーのスタイルや好みにより、このような方法を、ラベル付けされていないプライベートな異種クライアント データを使用する分散システムに適応させることは困難です。フェデレーテッド ラーニングにより、ローカル クライアント間のプライバシー侵害を引き起こすことなく、複数のクライアントがグローバル モデルを集合的に学習できます。一方、連合学習の別の方向性は、局所的な異質性に対処するための個別化された方法を研究しています。ただし、分散化された設定でラベルなしで一般化とパーソナライズの両方を解決する作業は、まだなじみがありません。この作業では、対照的な学習のためにローカル スタイル情報にローカル コンテンツ情報を注入することにより、より一般化されたグローバル モデルを学習し、下流タスクのローカル スタイル情報を誘導することにより、よりパーソナライズされたローカル モデルを学習するための新しい方法 FedStyle を提案します。スタイル情報は、元のローカル データと強力に拡張されたローカル データ (ソーベル フィルター処理された画像) を対比することによって抽出されます。 IID 設定と非 IID 設定の両方で線形評価を行った広範な実験を通じて、FedStyle が様式化された分散設定での一般化ベースライン方法とパーソナライズベースライン方法の両方よりも優れていることを示しています。包括的なアブレーションを通じて、スタイルインフュージョンのデザインと様式化されたパーソナライゼーションがパフォーマンスを大幅に向上させることを実証します。
Discriminative unsupervised learning methods such as contrastive learning have demonstrated the ability to learn generalized visual representations on centralized data. It is nonetheless challenging to adapt such methods to a distributed system with unlabeled, private, and heterogeneous client data due to user styles and preferences. Federated learning enables multiple clients to collectively learn a global model without provoking any privacy breach between local clients. On the other hand, another direction of federated learning studies personalized methods to address the local heterogeneity. However, work on solving both generalization and personalization without labels in a decentralized setting remains unfamiliar. In this work, we propose a novel method, FedStyle, to learn a more generalized global model by infusing local style information with local content information for contrastive learning, and to learn more personalized local models by inducing local style information for downstream tasks. The style information is extracted by contrasting original local data with strongly augmented local data (Sobel filtered images). Through extensive experiments with linear evaluations in both IID and non-IID settings, we demonstrate that FedStyle outperforms both the generalization baseline methods and personalization baseline methods in a stylized decentralized setting. Through comprehensive ablations, we demonstrate our design of style infusion and stylized personalization improve performance significantly.
updated: Fri Nov 11 2022 20:14:26 GMT+0000 (UTC)
published: Fri Nov 11 2022 20:14:26 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト