Search papers, labs, and topics across Lattice.
This paper introduces FedXDS, a novel approach that leverages explainable AI (XAI) methods to address the challenges of data heterogeneity in federated learning. By utilizing propagation-based feature attribution techniques, FedXDS identifies which data elements should be selectively shared among clients to enhance model performance while maintaining privacy. Experimental results show that FedXDS outperforms existing methods in accuracy and convergence speed across diverse client distributions, while also providing formal privacy guarantees against common adversarial attacks.
Selective data sharing guided by XAI can significantly boost federated learning performance, achieving higher accuracy and faster convergence even in heterogeneous environments.
Explainable AI (XAI) methods have demonstrated significant success in recent years at identifying relevant features in input data that drive deep learning model decisions, enhancing interpretability for users. However, the potential of XAI beyond providing model transparency has remained largely unexplored in adjacent machine learning domains. In this paper, we show for the first time how XAI can be utilized in the context of federated learning. Specifically, while federated learning enables collaborative model training without raw data sharing, it suffers from performance degradation when client data distributions exhibit statistical heterogeneity. We introduce FedXDS (Federated Learning via XAI-guided Data Sharing), the first approach to utilize feature attribution techniques to identify precisely which data elements should be selectively shared between clients to mitigate heterogeneity. By employing propagation-based attribution, our method identifies task-relevant features through a single backward pass, enabling selective data sharing that aligns client contributions. To protect sensitive information, we incorporate metric privacy techniques that provide formal privacy guarantees while preserving utility. Experimental results demonstrate that our approach consistently achieves higher accuracy and faster convergence compared to existing methods across varying client numbers and heterogeneity settings. We provide theoretical privacy guarantees and empirically demonstrate robustness against both membership inference and feature inversion attacks. Code is available at https://github.com/MaxH1996/FedXDS.
