Byzantine-tolerant distributed learning of finite mixture models

๐Ÿ“… 2024-07-19
๐Ÿ›๏ธ arXiv.org
๐Ÿ“ˆ Citations: 0
โœจ Influential: 0
๐Ÿ“„ PDF
๐Ÿค– AI Summary
To address the dual challenges of label switching and Byzantine failures in finite mixture model learning under distributed settings, this paper proposes Distance-Filtered Mixture Reduction (DFMR), a hyperparameter-free, Byzantine-robust divide-and-conquer estimator. DFMR introduces the first Byzantine-resilient aggregation mechanism for mixture models in non-Euclidean spaces, leveraging a pairwise distance-based filtering strategy to simultaneously align component parameters and prune outlying local estimates. Theoretically, DFMR achieves statistically optimal convergence rates under mild regularity conditions. Empirically, it significantly outperforms existing methodsโ€”including Mixture Reduction (MR)โ€”on both synthetic benchmarks and the MNIST handwritten digit recognition task, demonstrating superior robustness against Byzantine adversaries while maintaining high statistical efficiency.

Technology Category

Application Category

๐Ÿ“ Abstract
This paper proposes two split-and-conquer (SC) learning estimators for finite mixture models that are tolerant to Byzantine failures. In SC learning, individual machines obtain local estimates, which are then transmitted to a central server for aggregation. During this communication, the server may receive malicious or incorrect information from some local machines, a scenario known as Byzantine failures. While SC learning approaches have been devised to mitigate Byzantine failures in statistical models with Euclidean parameters, developing Byzantine-tolerant methods for finite mixture models with non-Euclidean parameters requires a distinct strategy. Our proposed distance-based methods are hyperparameter tuning free, unlike existing methods, and are resilient to Byzantine failures while achieving high statistical efficiency. We validate the effectiveness of our methods both theoretically and empirically via experiments on simulated and real data from machine learning applications for digit recognition. The code for the experiment can be found at https://github.com/SarahQiong/RobustSCGMM.
Problem

Research questions and friction points this paper is trying to address.

Addresses Byzantine failure in distributed learning of finite mixture models.
Solves label switching problem in distributed mixture model learning.
Ensures robust and accurate aggregation of local model estimates.
Innovation

Methods, ideas, or system contributions that make the work stand out.

Byzantine-tolerant distributed learning adaptation
Distance Filtered Mixture Reduction (DFMR)
Robust filtering via pairwise L2 distances
Renmin University of China | The University of British Columbia
Q
Qiong Zhang
Institute of Statistics and Big Data, Renmin University of China, Beijing, China
J
Jiahua Chen
Department of Statistics, The University of British Columbia, Vancouver, Canada