FedTrip: A Resource-Efficient Federated Learning Method with Triplet Regularization

TL;DR

FedTrip is a novel federated learning regularization technique that reduces resource consumption and accelerates convergence by controlling divergence between models and avoiding historical model correlation.

Contribution

It introduces FedTrip, a resource-efficient regularization method that improves convergence speed and model accuracy in heterogeneous federated learning environments.

Findings

FedTrip outperforms state-of-the-art baselines in accuracy.

It significantly reduces communication and computation overhead.

FedTrip effectively handles data heterogeneity among clients.

Abstract

In the federated learning scenario, geographically distributed clients collaboratively train a global model. Data heterogeneity among clients significantly results in inconsistent model updates, which evidently slow down model convergence. To alleviate this issue, many methods employ regularization terms to narrow the discrepancy between client-side local models and the server-side global model. However, these methods impose limitations on the ability to explore superior local models and ignore the valuable information in historical models. Besides, although the up-to-date representation method simultaneously concerns the global and historical local models, it suffers from unbearable computation cost. To accelerate convergence with low resource consumption, we innovatively propose a model regularization method named FedTrip, which is designed to restrict global-local divergence and decrease current-historical correlation for alleviating the negative effects derived from data heterogeneity. FedTrip helps the current local model to be close to the global model while keeping away from historical local models, which contributes to guaranteeing the consistency of local updates among clients and efficiently exploring superior local models with negligible additional computation cost on attaching operations. Empirically, we demonstrate the superiority of FedTrip via extensive evaluations. To achieve the target accuracy, FedTrip outperforms the state-of-the-art baselines in terms of significantly reducing the total overhead of client-server communication and local computation.