A Unified Machine Learning Framework for Li-Ion Battery State Estimation and Prediction

The accurate estimation and prediction of internal states in lithium-ion (Li-Ion) batteries, such as State of Charge (SoC) and Remaining Useful Life (RUL), are vital for optimizing battery performance, safety, and longevity in electric vehicles and other applications. This paper presents a unified, modular, and extensible machine learning (ML) framework designed to address the heterogeneity and complexity of battery state prediction tasks. The proposed framework supports flexible configurations across multiple dimensions, including feature engineering, model selection, and training/testing strategies. It integrates standardized data processing pipelines with a diverse set of ML models, such as a long short-term memory neural network (LSTM), a convolutional neural network (CNN), a feedforward neural network (FFNN), automated machine learning (AutoML), and classical regressors, while accommodating heterogeneous datasets. The framework’s applicability is demonstrated through five distinct use cases involving SoC estimation and RUL prediction using real-world and benchmark datasets. Experimental results highlight the framework’s adaptability, methodological transparency, and robust predictive performance across various battery chemistries, usage profiles, and degradation conditions. This work contributes to a standardized approach that facilitates the reproducibility, comparability, and practical deployment of ML-based battery analytics.