Federated Semi-Supervised Learning (FedSSL) has gained rising attention from both academic and industrial researchers, due to its unique characteristics of co-training machine learning models with isolated yet unlabeled data. Most existing FedSSL methods focus on the classical scenario, i.e, the labeled and unlabeled data are stored at the client side. However, in real world applications, client users may not provide labels without any incentive. Thus, the scenario of labels at the server side is more practical. Since unlabeled data and labeled data are decoupled, most existing FedSSL approaches may fail to deal with such a scenario. To overcome this problem, in this paper, we propose FedCon, which introduces a new learning paradigm, i.e., contractive learning, to FedSSL. Experimental results on three datasets show that FedCon achieves the best performance with the contractive framework compared with state-of-the-art baselines under both IID and Non-IID settings. Besides, ablation studies demonstrate the characteristics of the proposed FedCon framework.
ENS(巴黎高等师范学院);CNRS(法国国家科学研究中心);PSL Research University(巴黎文理研究大学)
To date, the most popular federated learning algorithms use coordinate-wise averaging of the model parameters. We depart from this approach by differentiating client contributions according to the performance of local learning and its evolution. The technique is inspired from control theory and its classification performance is evaluated extensively in IID framework and compared with FedAvg.
Contrastive Federated Learning with Tabular Data Silos
The University of Queensland(昆士兰大学);Monash University(莫纳什大学);Indonesia Taxation Office(印度尼西亚税务局);Queensland University of Technology(昆士兰科技大学);SUNY Albany(纽约州立大学奥尔巴尼分校);Dartmouth College(达特茅斯学院);Hong Kong Polytechnic University(香港理工大学);Renmin University of China(中国人民大学)
Federated learning is widely used to perform decentralized training of a global model on multiple devices while preserving the data privacy of each device. However, it suffers from heterogeneous local data on each training device which increases the difficulty to reach the same level of accuracy as the centralized training. Supervised Contrastive Learning which outperform cross-entropy tries to minimizes the difference between feature space of points belongs to the same class and pushes away points from different classes. We propose Supervised Contrastive Federated Learning in which devices ca…
Tsinghua University(清华大学);Microsoft Research Asia(微软亚洲研究院)
Contrastive learning is widely used for recommendation model learning, where selecting representative and informative negative samples is critical. Existing methods usually focus on centralized data, where abundant and high-quality negative samples are easy to obtain. However, centralized user data storage and exploitation may lead to privacy risks and concerns, while decentralized user data on a single client can be too sparse and biased for accurate contrastive learning. In this paper, we propose a federated contrastive learning method named FedCL for privacy-preserving recommendation, which…
Behavioral Anomaly Detection in Distributed Systems via Federated Contrastive Learning
Northeastern University(东北大学);Illinois Institute of Technology(伊利诺伊理工学院);Rochester Institute of Technology(罗切斯特理工学院);University of California, San Diego(加利福尼亚大学圣地亚哥分校);University of Southern California(南加利福尼亚大学);University of Michigan(密歇根大学)
Federated learning (FL) involves multiple distributed devices jointly training a shared model without any of the participants having to reveal their local data to a centralized server. Most of previous FL approaches assume that data on devices are fixed and stationary during the training process. However, this assumption is unrealistic because these devices usually have varying sampling rates and different system configurations. In addition, the underlying distribution of the device data can change dynamically over time, which is known as concept drift. Concept drift makes the learning process…
Federated learning allows multiple clients to collaborate to train high-performance deep learning models while keeping the training data locally. However, when the local data of all clients are not independent and identically distributed (i.e., non-IID), it is challenging to implement this form of efficient collaborative learning. Although significant efforts have been dedicated to addressing this challenge, the effect on the image classification task is still not satisfactory. In this paper, we propose FedProc: prototypical contrastive federated learning, which is a simple and effective feder…
National University of Singapore(新加坡国立大学);UC Berkeley(加州大学伯克利分校)
Federated learning enables multiple parties to collaboratively train a machine learning model without communicating their local data. A key challenge in federated learning is to handle the heterogeneity of local data distribution across parties. Although many studies have been proposed to address this challenge, we find that they fail to achieve high performance in image datasets with deep learning models. In this paper, we propose MOON: model-contrastive federated learning. MOON is a simple and effective federated learning framework. The key idea of MOON is to utilize the similarity between m…
Qilu University of Technology (Shandong Academy of Sciences)(齐鲁工业大学(山东省科学院));Shandong Computer Science Center(山东省计算中心);Guizhou University(贵州大学);Zhejiang University(浙江大学)
Federated learning is a distributed learning that allows each client to keep the original data locally and only upload the parameters of the local model to the server. Despite federated learning can address data island, it remains challenging to train with data heterogeneous in a real application. In this paper, we propose FedSiam-DA, a novel dual-aggregated contrastive federated learning approach, to personalize both local and global models, under various settings of data heterogeneity. Firstly, based on the idea of contrastive learning in the siamese network, FedSiam-DA regards the local and…
Personalized Federated Learning for Spatio-Temporal Forecasting: A Dual Semantic Alignment-Based Contrastive Approach
Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);The Hong Kong University of Science and Technology (Guangzhou)(香港科技大学(广州))
Breaking the Prototype Bias Loop: Confidence-Aware Federated Contrastive Learning for Highly Imbalanced Clients
Key Laboratory of Water Big Data Technology of Ministry of Water Resources, College of Computer Science and Software Engineering, Hohai University, Nanjing, China(水利部水大数据技术重点实验室,计算机科学与软件工程学院,河海大学,南京,中国);Department of Computer Science, City University of Hong Kong, Hong Kong, China(计算机科学系,香港城市大学,香港,中国);State Key Laboratory for Novel Software Technology, Nanjing University, Nanjing, China(新型软件技术国家重点实验室,南京大学,南京,中国);School of Artificial Intelligence and Information Technology, Nanjing University of Chinese Medicine, Nanjing, China(人工智能与信息技术学院,南京中医药大学,南京,中国)
We present federated momentum contrastive clustering (FedMCC), a learning framework that can not only extract discriminative representations over distributed local data but also perform data clustering. In FedMCC, a transformed data pair passes through both the online and target networks, resulting in four representations over which the losses are determined. The resulting high-quality representations generated by FedMCC can outperform several existing self-supervised learning methods for linear evaluation and semi-supervised learning tasks. FedMCC can easily be adapted to ordinary centralized…
Department of Computer Science, University of Oxford(牛津大学计算机科学系);University of Oxford(牛津大学);Oxford, UK(英国牛津(这里仅保留城市和国家,实际机构为牛津大学))
A label-efficient paradigm in computer vision is based on self-supervised contrastive pre-training on unlabeled data followed by fine-tuning with a small number of labels. Making practical use of a federated computing environment in the clinical domain and learning on medical images poses specific challenges. In this work, we propose FedMoCo, a robust federated contrastive learning (FCL) framework, which makes efficient use of decentralized unlabeled medical data. FedMoCo has two novel modules: metadata transfer, an inter-node statistical data augmentation module, and self-adaptive aggregation…
USTC(中国科学技术大学);University of Leeds(利兹大学);Beihang University(北京航空航天大学);NERC-RPP, CAEIT(中国电子科技集团公司信息科学研究院国家电磁辐射控制工程技术研究中心);Tsinghua University(清华大学)
Social bot detection is of paramount importance to the resilience and security of online social platforms. The state-of-the-art detection models are siloed and have largely overlooked a variety of data characteristics from multiple cross-lingual platforms. Meanwhile, the heterogeneity of data distribution and model architecture makes it intricate to devise an efficient cross-platform and cross-model detection framework. In this paper, we propose FedACK, a new federated adversarial contrastive knowledge distillation framework for social bot detection. We devise a GAN-based federated knowledge d…
FedCoSR: Personalized Federated Learning with Contrastive Shareable Representations for Label Heterogeneity in Non-IID Data
Department of Data Science and AI, Faculty of IT and Monash Energy Institute, Monash University(数据科学与人工智能系,IT学院和莫纳什能源研究所,莫纳什大学);School of Computer Science and Technology, University of Electronic Science and Technology of China (UESTC), Chengdu, and Shenzhen Institute for Advanced Study of UESTC, Shenzhen, China(计算机科学与技术学院,电子科技大学(UESTC)成都校区,以及UESTC深圳高级研究所,深圳,中国)
Towards Performance-Enhanced Model-Contrastive Federated Learning using Historical Information in Heterogeneous Scenarios
Key Laboratory of Computing Power Network and Information Security, Ministry of Education, Shandong Computer Science Center, Qilu University of Technology (Shandong Academy of Sciences)(计算能力网络与信息安全重点实验室,教育部,山东计算机科学中心,齐鲁工业大学(山东科学院));School of Information and Software Engineering, University of Electronic Science and Technology of China(信息与软件工程学院,电子科技大学);School of Computer Science and Engineering, Shandong University of Science and Technology(计算机科学与工程学院,山东科技大学);School of Big Data and Software Engineering, Chongqing University(大数据与软件工程学院,重庆大学)
Federated Contrastive Learning for Personalized Semantic Communication
Beijing University of Posts and Telecommunications(北京邮电大学);Tsinghua University(清华大学);University of Essex(埃塞克斯大学);Peng Cheng Laboratory(鹏城实验室);Queen Mary University of London(伦敦玛丽女王大学)
FedMCP: Parameter-Efficient Federated Learning with Model-Contrastive Personalization
School of Data Science and Engineering, East China Normal University(华东师范大学数据科学与工程学院);Manning College of Information & Computer Sciences, University of Massachusetts Amherst(马萨诸塞大学阿默斯特分校曼宁信息与计算机科学学院)
University of Pittsburgh(匹兹堡大学);University of Notre Dame(圣母大学)
Supervised deep learning needs a large amount of labeled data to achieve high performance. However, in medical imaging analysis, each site may only have a limited amount of data and labels, which makes learning ineffective. Federated learning (FL) can help in this regard by learning a shared model while keeping training data local for privacy. Traditional FL requires fully-labeled data for training, which is inconvenient or sometimes infeasible to obtain due to high labeling cost and the requirement of expertise. Contrastive learning (CL), as a self-supervised learning approach, can effectivel…
Australian Artificial Intelligence Institute(澳大利亚人工智能研究所);University of Technology Sydney(悉尼科技大学);Google Research(谷歌研究院);University of Washington(华盛顿大学);University of Maryland(马里兰大学)
Federated Learning (FL) is a machine learning paradigm that allows decentralized clients to learn collaboratively without sharing their private data. However, excessive computation and communication demands pose challenges to current FL frameworks, especially when training large-scale models. To prevent these issues from hindering the deployment of FL systems, we propose a lightweight framework where clients jointly learn to fuse the representations generated by multiple fixed pre-trained models rather than training a large-scale model from scratch. This leads us to a more practical FL problem…
FedMAC: Tackling Partial-Modality Missing in Federated Learning with Cross-Modal Aggregation and Contrastive Regularization
Hanoi University of Science and Technology(河内科学技术大学);Nagoya University(名古屋大学);Washington State University(华盛顿州立大学);Swiss Federal Institute of Technology Lausanne(洛桑联邦理工学院)
University of Pittsburgh(匹兹堡大学);University of Notre Dame(圣母大学);George Mason University(乔治梅森大学);University of Pittsburgh Medical Center(匹兹堡大学医学中心)
In dermatological disease diagnosis, the private data collected by mobile dermatology assistants exist on distributed mobile devices of patients. Federated learning (FL) can use decentralized data to train models while keeping data local. Existing FL methods assume all the data have labels. However, medical data often comes without full labels due to high labeling costs. Self-supervised learning (SSL) methods, contrastive learning (CL) and masked autoencoders (MAE), can leverage the unlabeled data to pre-train models, followed by fine-tuning with limited labels. However, combining SSL and FL h…
Multimodal Federated Learning via Contrastive Representation Ensemble
Institute for AI Industry Research, Tsinghua University(清华大学人工智能产业研究院);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);University of Waterloo(滑铁卢大学);Carnegie Mellon University(卡内基梅隆大学)
FedSC: Provable Federated Self-supervised Learning with Spectral Contrastive Objective over Non-i.i.d. Data
University of California, San Francisco(加州大学旧金山分校);Lehigh University(利哈伊大学);New Jersey Institute of Technology(新泽西理工学院);University of Louisiana at Lafayette(路易斯安那大学拉法叶分校)
Federated Attack Campaign Detection via Contrastive Encoding of Threat Indicators in Gradient Updates
Technical University of Applied Sciences Würzburg-Schweinfurt(维尔茨堡-施韦因富特应用技术大学);Bielefeld University(比勒费尔德大学);Center for Cybersecurity TTZ-WUE(TTZ-WUE网络安全中心)
Defending Against Backdoor Attacks via Alignment Checking in Model-Contrastive Federated Learning
Shandong Computer Science Center(山东计算机科学中心);Qilu University of Technology (Shandong Academy of Sciences)(齐鲁工业大学(山东省科学院));College of computer science and technology, China University of Petroleum(中国石油大学计算机科学与技术学院);School of Computer Science and Engineering, University of Electronic Science and Technology of China(电子科技大学计算机科学与工程学院);Big Data Institute, Qilu University of Technology(齐鲁工业大学大数据研究院)
Tsinghua University(清华大学);Microsoft Research Asia(微软亚洲研究院);Sony AI(索尼人工智能);USTC(中国科学技术大学);Shenzhen University(深圳大学);Zhongguancun Laboratory(中关村实验室)
Vertical federated learning (VFL) is a privacy-preserving machine learning paradigm that can learn models from features distributed on different platforms in a privacy-preserving way. Since in real-world applications the data may contain bias on fairness-sensitive features (e.g., gender), VFL models may inherit bias from training data and become unfair for some user groups. However, existing fair machine learning methods usually rely on the centralized storage of fairness-sensitive features to achieve model fairness, which are usually inapplicable in federated scenarios. In this paper, we prop…
University of Pittsburgh(匹兹堡大学);University of Notre Dame(圣母大学);George Mason University(乔治梅森大学);University of New Mexico(新墨西哥大学);University of Pittsburgh Medical Center(匹兹堡大学医学中心)
Deep learning models have been deployed in an increasing number of edge and mobile devices to provide healthcare. These models rely on training with a tremendous amount of labeled data to achieve high accuracy. However, for medical applications such as dermatological disease diagnosis, the private data collected by mobile dermatology assistants exist on distributed mobile devices of patients, and each device only has a limited amount of data. Directly learning from limited data greatly deteriorates the performance of learned models. Federated learning (FL) can train models by using data distri…
Tackling Feature-Classifier Mismatch in Federated Learning via Prompt-Driven Feature Transformation
School of Computer Science and Engineering, Beihang University(北京航空航天大学计算机科学与工程学院);Zhongguancun Laboratory(中关村实验室);Sichuan University(四川大学);University at Buffalo(纽约州立大学布法罗分校)
Semi-Supervised Federated Learning via Dual Contrastive Learning and Soft Labeling for Intelligent Fault Diagnosis
School of Electronic and Optical Engineering, Nanjing University of Science and Technology(南京理工大学电子与光学工程学院);School of Information Science and Engineering, Southeast University(东南大学信息科学与工程学院);Institute of Artificial Intelligence Research, Jiangsu Second Normal University(江苏第二师范学院人工智能研究所);Digital Research Institute, ENN Group(ENN集团数字研究院);School of Computing, Engineering and Mathematical Sciences, La Trobe University(拉筹伯大学计算、工程与数学科学学院);College of Science and Engineering, James Cook University(詹姆斯库克大学科学与工程学院)
FedTGP: Trainable Global Prototypes with Adaptive-Margin-Enhanced Contrastive Learning for Data and Model Heterogeneity in Federated Learning
Shanghai Jiao Tong University(上海交通大学);Institute for AI Industry Research, Tsinghua University(清华大学智能产业研究院);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);Queen’s University Belfast(英国女王大学)
Towards Optimal Customized Architecture for Heterogeneous Federated Learning with Contrastive Cloud-Edge Model Decoupling
Southwestern University of Finance and Economics(西南财经大学);Beijing University of Posts and Telecommunications(北京邮电大学);City University of Hong Kong(香港城市大学)
Federated learning (FL) has been recognized as one of the most promising solutions for distributed machine learning (ML). In most of the current literature, FL has been studied for supervised ML tasks, in which edge devices collect labeled data. Nevertheless, in many applications, it is impractical to assume existence of labeled data across devices. To this end, we develop a novel methodology, Cooperative Federated unsupervised Contrastive Learning (CF-CL), for FL across edge devices with unlabeled datasets. CF-CL employs local device cooperation where data are exchanged among devices through …
KAIST(韩国科学技术院);Institute for Basic Science(基础科学研究所);Microsoft Research Asia(微软亚洲研究院);Tsinghua University(清华大学)
This paper presents FedX, an unsupervised federated learning framework. Our model learns unbiased representation from decentralized and heterogeneous local data. It employs a two-sided knowledge distillation with contrastive learning as a core component, allowing the federated system to function without requiring clients to share any data features. Furthermore, its adaptable architecture can be used as an add-on module for existing unsupervised algorithms in federated settings. Experiments show that our model improves performance significantly (1.58--5.52pp) on five unsupervised algorithms.