Explainable AI-Based Battery Management System with Federated Learning for Multi-Chemistry State-of-Health Estimation in Heterogeneous EV Fleets
DOI:
https://doi.org/10.29284/t4e4j308Keywords:
Federated Learning, Battery State-of-Health, Explainable AI, Electric Vehicles, Privacy-Preserving Machine LearningAbstract
The increased adoption rate of electric vehicles strengthens the demand for efficient battery management solutions with precise state-of-health estimation and support for data privacy within various heterogeneous fleet setups. This work presents a framework for explainable artificial intelligence with the application of federated learning for battery state-of-health prediction with support for privacy preservation within distributed electric vehicle fleets. This work presents a combination model with Long Short-Term Memory networks and the FedAVG learning approach for collaborative model training on ten distributed participants without any central aggregation. With the application of the MIT-Stanford battery dataset with a total number of 124 Li-ion battery cells, the system provides a root mean squared error with a measure of 1.92%, a mean absolute error of 1.49%, and a coefficient of determination with a measure of 0.826 on equal and proper benchmarks while providing complete privacy and local support within all datasets. Additionally, this system comes with a window-based feature engineering approach with dual normalization and complete evaluation for system efficiency on non-identically distributed datasets. The core contributions for this work come with the union and integration between battery health monitoring and distributed federative learning for complete management system support with sub-2% prediction accuracy at reduced prices and from aggregated datasets.
References
[1] International Energy Agency, "Global EV Outlook 2023," IEA Publications, 2023. Available: https://www.iea.org/reports/global-ev-outlook-2023
[2] Y. Li, K. Liu, A. M. Foley, A. Zülke, M. Berecibar, E. Nanini-Maury, J. Van Mierlo, and H. E. Hoster, "Data-driven health estimation and lifetime prediction of lithium-ion batteries: A review," Renewable and Sustainable Energy Reviews, vol. 113, p. 109254, 2019. https://doi.org/10.1016/j.rser.2019.109254
[3] I. Bhattacharya, J. Rajan, R. Maddipatla, S. Panchanathan, and P. Bahl, "State of charge and state of health estimation in electric vehicles: Challenges, approaches and future directions," Batteries, vol. 11, no. 1, p. 32, 2025. https://doi.org/10.3390/batteries11010032
[4] M. A. Hannan, M. S. H. Lipu, A. Hussain, and A. Mohamed, "A review of lithium-ion battery state of charge estimation and management system in electric vehicle applications: Challenges and recommendations," Renewable and Sustainable Energy Reviews, vol. 78, pp. 834-854, 2017. https://doi.org/10.1016/j.rser.2017.05.001
[5] Y. Li, W. Li, S. Liu, and T. Liu, "Review of battery state estimation methods for electric vehicles - Part II: SOH estimation," Journal of Energy Storage, vol. 85, p. 111364, 2024. https://doi.org/10.1016/j.est.2024.111364
[6] M. Çali and D. Kayabasi, "Evaluation of advances in battery health prediction for electric vehicles from traditional linear filters to latest machine learning approaches," Batteries, vol. 10, no. 10, p. 356, 2024. https://doi.org/10.3390/batteries10100356
[7] G. Xu, J. Xu, and Y. Zhu, "LSTM-based estimation of lithium-ion battery SOH using data characteristics and spatio-temporal attention," PLOS ONE, vol. 19, no. 12, p. e0312856, 2024. https://doi.org/10.1371/journal.pone.0312856
[8] European Parliament and Council, "Regulation (EU) 2016/679 (General Data Protection Regulation)," Official Journal of the European Union, 2016.
[9] S. Yan, H. Fang, J. Li, T. Ward, N. O'Connor, and M. Liu, "Privacy-aware energy consumption modeling of connected battery electric vehicles using federated learning," IEEE Transactions on Transportation Electrification, vol. 10, no. 2, pp. 3212-3224, 2024. https://doi.org/10.1109/TTE.2023.3343106
[10] C. Lyu, W. Zhao, J. Song, Y. Song, H. Zhang, and L. Wang, "Review on battery state estimation and management solutions for next-generation connected vehicles," Energies, vol. 17, no. 1, p. 202, 2024. https://doi.org/10.3390/en17010202
[11] B. McMahan, E. Moore, D. Ramage, S. Hampson, and B. A. y Arcas, "Communication-efficient learning of deep networks from decentralized data," in Proceedings of the 20th International Conference on Artificial Intelligence and Statistics (AISTATS), 2017, pp. 1273-1282.
[12] T. Li, A. K. Sahu, A. Talwalkar, and V. Smith, "Federated learning: Challenges, methods, and future directions," IEEE Signal Processing Magazine, vol. 37, no. 3, pp. 50-60, 2020. https://doi.org/10.1109/MSP.2020.2975749
[13] Q. Yang, Y. Liu, T. Chen, and Y. Tong, "Federated machine learning: Concept and applications," ACM Transactions on Intelligent Systems and Technology, vol. 10, no. 2, pp. 1-19, 2019. https://doi.org/10.1145/3298981
[14] C. V. Prakash, Q. Liang, C. G. Brinton, Z. A. Khanam, and Z. Wang, "Federated learning for connected and automated vehicles: A survey of existing approaches and challenges," IEEE Transactions on Intelligent Vehicles, vol. 9, no. 1, pp. 119-137, 2024. https://doi.org/10.1109/TIV.2023.3331961
[15] H. Khan, M. K. Hasan, S. Tarmizi, S. A. A. Shah, I. U. Khan, A. H. Ahmed, A. Tan, and N. R. Mohamad, "Privacy preserving machine learning for electric vehicles: A survey," arXiv preprint arXiv:2205.08462, 2024. https://doi.org/10.48550/arXiv.2205.08462
[16] A. A. El Bahnasawi, M. Abdellatif, M. Salem, P. Fasouli, M. Gebser, and K. Kyamakya, "Balancing performance and transparency: Explainable AI for battery health monitoring," in Designing the Conceptual Landscape for a XAIR Validation Infrastructure (DCLXVI 2024), Lecture Notes in Networks and Systems, vol. 1375, Springer, 2025, pp. 123-138. https://doi.org/10.1007/978-3-031-89274-5_8
[17] P. Linardatos, V. Papastefanopoulos, and S. Kotsiantis, "Explainable AI: A review of machine learning interpretability methods," Entropy, vol. 23, no. 1, p. 18, 2021. https://doi.org/10.3390/e23010018
[18] S. M. Lundberg and S. I. Lee, "A unified approach to interpreting model predictions," in Advances in Neural Information Processing Systems (NeurIPS), 2017, pp. 4765-4774.
[19] R. Dwivedi, D. Dave, H. Naik, S. Singhal, R. Omer, P. Patel, B. Qian, Z. Wen, T. Shah, G. Morgan, and R. Ranjan, "Explainable AI (XAI): Core ideas, techniques, and solutions," ACM Computing Surveys, vol. 55, no. 9, pp. 1-33, 2023. https://doi.org/10.1145/3561048
[20] W. Li, M. Rentemeister, J. Badeda, D. Jöst, D. Schulte, and D. U. Sauer, "Decentralized deep learning approach for lithium-ion batteries state of health forecasting using federated learning," IEEE Transactions on Transportation Electrification, 2024. https://doi.org/10.1109/TTE.2024.3382407
[21] P. Abbaraju, S. K. Kundu, and S. Kundu, "A novel federated & ensembled learning-based battery state-of-health estimation for connected electric vehicles," IEEE Open Journal of Intelligent Transportation Systems, vol. 5, pp. 445-453, 2024. https://doi.org/10.1109/OJITS.2024.3430843
[22] X. Chen, X. Wang, and Y. Deng, "Federated learning-based prediction of electric vehicle battery pack capacity using time-domain and frequency-domain feature extraction," Energy, vol. 313, p. 134449, 2025. https://doi.org/10.1016/j.energy.2025.134449
[23] L. Ling and Y. Wei, "A study on data privacy protection in power battery SOH prediction based on federated learning," Ionics, vol. 31, no. 4, pp. 2345-2358, 2025. https://doi.org/10.1007/s11581-025-06606-5
[24] Z. Xiao, B. Jiang, J. Zhu, X. Wei, and H. Dai, "State of health estimation for lithium-ion batteries using an explainable XGBoost model with parameter optimization," Batteries, vol. 10, no. 11, p. 394, 2024. https://doi.org/10.3390/batteries10110394
Downloads
Published
Issue
Section
License
This work is licensed under a Creative Commons Attribution 4.0 International License.
This work is licensed under a Creative Commons Attribution 4.0 International License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
