Federated Learning with Discriminative Naive Bayes Classifier
This work addresses privacy concerns in federated learning for classification tasks, but it is incremental as it adapts an existing method to a specific variant.
The paper tackled the problem of training Naive Bayes classifiers in federated learning settings while preserving privacy, by proposing a discriminative variant that shares meaningless parameters instead of conditional probability tables, and demonstrated its effectiveness through experiments on 12 datasets with accurate classification results.
Federated Learning has emerged as a promising approach to train machine learning models on decentralized data sources while preserving data privacy. This paper proposes a new federated approach for Naive Bayes (NB) classification, assuming discrete variables. Our approach federates a discriminative variant of NB, sharing meaningless parameters instead of conditional probability tables. Therefore, this process is more reliable against possible attacks. We conduct extensive experiments on 12 datasets to validate the efficacy of our approach, comparing federated and non-federated settings. Additionally, we benchmark our method against the generative variant of NB, which serves as a baseline for comparison. Our experimental results demonstrate the effectiveness of our method in achieving accurate classification.