DFML Project Overview and Results
DFML Project Overview and Results
Top-K sparsification and quantization both aim to reduce communication overhead by diminishing the size of data transmitted during model updates, with sparsification focusing on sending only the most significant updates and quantization compressing data size more broadly. In the project, top-K sparsification resulted in a higher accuracy (84.25%) and F1-score (0.802) compared to quantization (76.25% accuracy, 0.7001 F1-score), suggesting it was more effective at maintaining model performance while reducing overhead .
FedProx differs from FedAvg by introducing a proximal term in the local objective function for each client, which helps stabilize convergence and allows for better handling of heterogeneity across client datasets. In the project, FedProx achieved higher accuracy and F1-score compared to FedAvg under full client participation, with 85.41% accuracy and 0.813 F1-score versus FedAvg's 82.66% accuracy and 0.784 F1-score .
Weighted client participation can be particularly beneficial in scenarios where clients have varying dataset sizes and distributions. In the project, client contributions to the global model were weighted based on dataset size, acknowledging that not all data may be equally informative or abundant. This approach yielded better model performance compared to treating all clients equally, achieving an accuracy of 87.6% and F1-score of 0.835 .
The project employed a non-iid data partitioning method by assigning unbalanced labels and data to clients, resulting in client-specific datasets that mimic real-world federated learning environments. This approach highlights the diversity and imbalance in client data, reflecting scenarios where data distribution across clients may vary significantly, thus testing the robustness and scalability of federated learning algorithms .
The Adam optimizer was applied on the server-side to handle sparse gradients effectively and efficiently train the global model despite noisy problems. It combines AdaGrad and RMSProp properties, optimizing the model's convergence. In the project, this optimizer achieved the highest performance metrics, with an accuracy of 94.12% and an F1-score of 0.93, significantly outperforming other optimization methods .
The primary findings indicated that FedProx with full client participation achieved superior results compared to FedAvg, with 85.41% accuracy and 0.813 F1-score versus FedAvg's 82.66% accuracy and 0.784 F1-score. The Adam optimizer provided the best overall performance (94.12% accuracy, 0.93 F1-score). Weighted client participation improved outcomes by considering dataset size, resulting in 87.6% accuracy and 0.835 F1-score. These findings suggest that combining advanced optimization and participation methods enhances federated learning outcomes .
Non-iid data partitioning impacts results interpretability by introducing a more realistic scenario where client data variability can confound outcomes of algorithm performance. It complicates attributing model performance purely to the algorithm as variations may arise from data distribution effects. Such partitioning in the project led to diverse, unbalanced client datasets, mirroring real-world data variance and challenging the model’s ability to generalize, thereby providing a stringent test of algorithm robustness .
Client participation variations affect federated learning performance significantly. The project demonstrated that full client participation using FedAvg resulted in higher accuracy (82.66%) and F1-score (0.784) compared to partial participation (75.59% accuracy, 0.682 F1-score). With FedProx, full participation similarly showed higher metrics (85.41% accuracy, 0.813 F1-score) versus partial participation. Thus, involving more clients generally boosts model reliability and performance by incorporating more data variability into the global model .
The project applied top-K sparsification and quantization techniques to address communication overhead in federated learning. Top-K sparsification reduces communication overhead by transmitting only the most significant model updates, specifically applying sparsification when sending weights from clients to the server, given that clients are typically resource-constrained. Quantization further compresses the model updates by converting the floating-point model weights to lower precision formats like integers, significantly reducing data size transmitted during updates .
Quantization contributed to communication efficiency by compressing model updates through conversion to lower precision formats, thereby reducing the transmission data size. Despite enhancing efficiency, it had lower performance metrics than some other methods with an accuracy of 76.25% and an F1-score of 0.7001. This indicates a trade-off between communication efficiency and model accuracy in federated learning .