Search papers, labs, and topics across Lattice.
The paper introduces FedTVD, a novel federated learning algorithm that addresses data heterogeneity by weighting client contributions based on both data quality and quantity. By integrating Total Variation Distance (TVD) to assess the divergence of local label distributions from a uniform global distribution, FedTVD effectively mitigates the influence of skewed datasets during model aggregation. Experimental results demonstrate that FedTVD outperforms traditional methods like FedAvg, achieving up to a 10.6% improvement on CIFAR-10 under highly skewed conditions while maintaining robust performance across varying data distributions.
FedTVD redefines client weighting in federated learning, achieving significant performance gains by balancing data quality and quantity.
Federated Learning (FL) enables collaborative model training across distributed client devices while preserving data privacy. However, FL faces significant challenges due to data heterogeneity, particularly in terms of label distribution skewness and variations in dataset sizes, which can lead to biased model updates and hinder convergence. To address this, we propose FedTVD, a novel FL algorithm that weights client contributions during aggregation by considering both data quality and quantity. Unlike traditional FL approaches such as FedAvg, which rely solely on dataset size for client weighting, FedTVD integrates Total Variation Distance (TVD) to measure the divergence between each client's local label distribution and a uniform global distribution. Clients with highly skewed distributions receive lower weights, preventing unbalanced datasets with imbalances from disproportionately influencing the global model. At the same time, dataset size is incorporated to ensure scalability and fairness. This dual-weighting mechanism effectively mitigates the impact of data imbalance, leading to more stable and generalized global models. Experimental results show that FedTVD consistently outperforms state-of-the-art methods across all datasets (FMNIST, CIFAR-10, and CIFAR-100) and all levels of data heterogeneity. Notably, it achieves up to 10.6% improvement over FedAvg on CIFAR-10 under highly skewed data, while maintaining top performance even under moderate and IID settings.
