Collaborative Network and Metric Learning Based Label Noise Robust Federated Learning Method
Currently,there is limited research on the problem of label noise in federated learning.The main approaches involve in-troducing a benchmark dataset on the server side to evaluate the client's model,controlling the aggregation weights and feature class centers of the clients.However,most methods still have room for improvement in distinguishing noisy clients or noisy sam-ples.This paper proposes a label-noise robust federated learning method based on co-networks and metric learning.The method consists of the following three parts:1)Client mutual evaluation mechanism.Clients score each other's models,construct a rating matrix,and further transform it into an adjacency matrix to differentiate clean/noisy clients.2)Collaborative network module.By constructing two collaborative equivalent federated network models,the Jensen-Shannon divergence is used to distinguish clean samples from noisy samples for the training of collaborative networks.3)Federated-collaborative network triplet loss.A loss function is designed to constrain the output features of the collaborative networks for the same noisy samples.Experimental veri-fication is conducted on the publicly available datasets CIFAR-10 and CIFAR-100,and the results demonstrate the superiority of the proposed method in accuracy.
NETL
NSTL
万方数据
