Explaining Federated Learning Through Concepts in Image Classification.

Jiaxin Shen,Xiaoyi Tao,Liangzhi Li,Zhiyang Li,Bowen Wang
DOI: https://doi.org/10.1007/978-981-97-0808-6_19
2024-01-01
Abstract:Federated learning is a machine learning framework that solves the problem of data silos under secure data protection measures and is gradually becoming a machine learning paradigm for future AI development. In recent years, federated learning has evolved in research areas such as security, model aggregation, and incentive mechanisms. However, the direction of interpretability of the model in the federated learning framework has not been explored. To bridge this gap, this paper proposes an interpretable model of Federated Concept Learning (FCL). FCL is trained on the client side using Bottleneck Concept Learner (BotCL) to generate human-understandable concepts. Each client uploads the co-occurrence scores of the concepts and classes obtained from the training to the server, and in order to mitigate the influence of possible malicious clients on the model, the server aggregates the obtained co-occurrence scores after optimization. The aggregated scores are then sent to the clients to update the model, which performs the classification task only by the presence or absence of the concepts. Experimental results show that our model has equivalent performance to other federated learning methods and successfully mitigates the impact of malicious client degree on the model performance, as well as provides an interpretation of the model classification results.
What problem does this paper attempt to address?