IDEAS home Printed from https://ideas.repec.org/a/eee/energy/v335y2025ics0360544225039052.html

Enhanced few-shot state-of-health estimation for lithium-ion batteries via Masked Autoencoder

Author

Listed:
  • Shen, Yifan
  • Guo, Dongxu
  • Wang, Yu
  • Chen, Jianguo
  • Liu, Xuyang
  • Han, Xuebing
  • Zheng, Yuejiu
  • Ouyang, Minggao

Abstract

Accurately estimating the state-of-health (SOH) of lithium-ion batteries (LIBs) is crucial for optimizing performance, ensuring operational safety, and enabling predictive maintenance in battery management systems. With the widespread adoption of LIBs, a large amount of field data has been generated, yet current data-driven SOH estimation methods often fail to fully utilize it due to the lack of labeled data. To address this, we propose a method based on semi-supervised learning to exploit large-scale unlabeled data for accurate SOH estimation. A generative unsupervised model, the Masked Autoencoder (MAE), is pre-trained on unlabeled field charging data to automatically extract latent representations related to SOH. The model is then fine-tuned with a small amount of labeled data. Experimental results show that using only 20 % of the labeled data usually required for supervised learning, the method achieves an RMSE of 2.14 %. The latent representation extraction capability of the MAE is validated via incremental capacity (IC) analysis, which explains the 14 % improvement in estimation accuracy (RMSE of 1.84 %) when using data from a specific voltage range (3.8–3.9 V). Furthermore, experiments demonstrate that even with only 21.33 min of charging data—consisting of only charge quantity and voltage signals—the model can still achieve a competitive RMSE of 1.94 %. This work introduces a novel approach for SOH estimation using large-scale, unlabeled field data and provides valuable insights for battery management in the era of artificial intelligence.

Suggested Citation

  • Shen, Yifan & Guo, Dongxu & Wang, Yu & Chen, Jianguo & Liu, Xuyang & Han, Xuebing & Zheng, Yuejiu & Ouyang, Minggao, 2025. "Enhanced few-shot state-of-health estimation for lithium-ion batteries via Masked Autoencoder," Energy, Elsevier, vol. 335(C).
  • Handle: RePEc:eee:energy:v:335:y:2025:i:c:s0360544225039052
    DOI: 10.1016/j.energy.2025.138263
    as

    Download full text from publisher

    File URL: http://www.sciencedirect.com/science/article/pii/S0360544225039052
    Download Restriction: Full text for ScienceDirect subscribers only

    File URL: https://libkey.io/10.1016/j.energy.2025.138263?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to

    for a different version of it.

    References listed on IDEAS

    as
    1. Cheng, Yujie & Song, Dengwei & Wang, Zhenya & Lu, Chen & Zerhouni, Noureddine, 2020. "An ensemble prognostic method for lithium-ion battery capacity estimation based on time-varying weight allocation," Applied Energy, Elsevier, vol. 266(C).
    2. Zhang, Yajun & Liu, Yajie & Wang, Jia & Zhang, Tao, 2022. "State-of-health estimation for lithium-ion batteries by combining model-based incremental capacity analysis with support vector regression," Energy, Elsevier, vol. 239(PB).
    3. Ma, Yan & Shan, Ce & Gao, Jinwu & Chen, Hong, 2022. "A novel method for state of health estimation of lithium-ion batteries based on improved LSTM and health indicators extraction," Energy, Elsevier, vol. 251(C).
    4. Wang, Yilin & Shen, Lei & Zhang, Yuxuan & Li, Yuanxiang & Zhang, Ruixin & Yang, Yongshen, 2023. "Self-supervised Health Representation Decomposition based on contrast learning," Reliability Engineering and System Safety, Elsevier, vol. 239(C).
    5. Huang, Xucong & Peng, Zhaoqin & Tang, Diyin & Chen, Juan & Zio, Enrico & Zheng, Zaiping, 2024. "A physics-informed autoencoder for system health state assessment based on energy-oriented system performance," Reliability Engineering and System Safety, Elsevier, vol. 242(C).
    6. Jiang, Fusheng & Ren, Yi & Tang, Ting & Wu, Zeyu & Xia, Quan & Sun, Bo & Yang, Dezhen, 2024. "An adaptive semi-supervised self-learning method for online state of health estimation of lithium-ion batteries," Energy, Elsevier, vol. 305(C).
    7. Wen, Jianping & Chen, Xing & Li, Xianghe & Li, Yikun, 2022. "SOH prediction of lithium battery based on IC curve feature and BP neural network," Energy, Elsevier, vol. 261(PA).
    8. Gu, Xinyu & See, K.W. & Li, Penghua & Shan, Kangheng & Wang, Yunpeng & Zhao, Liang & Lim, Kai Chin & Zhang, Neng, 2023. "A novel state-of-health estimation for the lithium-ion battery using a convolutional neural network and transformer model," Energy, Elsevier, vol. 262(PB).
    9. Pepe, Simona & Ciucci, Francesco, 2023. "Long-range battery state-of-health and end-of-life prediction with neural networks and feature engineering," Applied Energy, Elsevier, vol. 350(C).
    10. Peng, Simin & Zhu, Junchao & Wu, Tiezhou & Tang, Aihua & Kan, Jiarong & Pecht, Michael, 2024. "SOH early prediction of lithium-ion batteries based on voltage interval selection and features fusion," Energy, Elsevier, vol. 308(C).
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Ke, Leisi & Lin, Mingqiang & Meng, Jinhao & Wu, Ji & Wang, Wei & Wang, Fengxiang, 2026. "A sparse SOH estimation framework based on clustered voltage segments and multi-head CNN under arbitrary charging condition," Energy, Elsevier, vol. 344(C).
    2. Wang, Yaxuan & Guo, Shilong & Cui, Yue & Deng, Liang & Zhao, Lei & Li, Junfu & Wang, Zhenbo, 2025. "A comprehensive review of machine learning-based state of health estimation for lithium-ion batteries: data, features, algorithms, and future challenges," Renewable and Sustainable Energy Reviews, Elsevier, vol. 224(C).

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Chen, Kui & Luo, Yang & Long, Zhou & Li, Yang & Nie, Guangbo & Liu, Kai & Xin, Dongli & Gao, Guoqiang & Wu, Guangning, 2025. "Big data-driven prognostics and health management of lithium-ion batteries:A review," Renewable and Sustainable Energy Reviews, Elsevier, vol. 214(C).
    2. Wang, Rui & Lin, Huizhong & Choi, Jeongsub & Hashemi, Abolfazl & Zhu, Mengmeng, 2025. "Novel differential voltage features based machine learning approach to lithium-ion batteries SOH prediction at various C-rates," Energy, Elsevier, vol. 334(C).
    3. Chen, Liping & Xie, Siqiang & Lopes, António M. & Li, Huafeng & Bao, Xinyuan & Zhang, Chaolong & Li, Penghua, 2024. "A new SOH estimation method for Lithium-ion batteries based on model-data-fusion," Energy, Elsevier, vol. 286(C).
    4. Wang, Tong & Wu, Yan & Zhu, Keming & Cen, Jianmeng & Wang, Shaohong & Huang, Yuqi, 2025. "Deep learning and polarization equilibrium based state of health estimation for lithium-ion battery using partial charging data," Energy, Elsevier, vol. 317(C).
    5. Xin Ma & Xingke Ding & Chongyi Tian & Changbin Tian & Rui Zhu, 2025. "Estimation of Lithium-Ion Battery State of Health-Based Multi-Feature Analysis and Convolutional Neural Network–Long Short-Term Memory," Sustainability, MDPI, vol. 17(9), pages 1-20, April.
    6. Li, Ziyang & Zhang, Xiangwen & Gao, Wei, 2024. "State of health estimation of lithium-ion battery during fast charging process based on BiLSTM-Transformer," Energy, Elsevier, vol. 311(C).
    7. Wang, Yaxuan & Guo, Shilong & Cui, Yue & Deng, Liang & Zhao, Lei & Li, Junfu & Wang, Zhenbo, 2025. "A comprehensive review of machine learning-based state of health estimation for lithium-ion batteries: data, features, algorithms, and future challenges," Renewable and Sustainable Energy Reviews, Elsevier, vol. 224(C).
    8. Xue, Jingsong & Ma, Wentao & Feng, Xiaoyang & Guo, Peng & Guo, Yaosong & Hu, Xianzhi & Chen, Badong, 2023. "Stacking integrated learning model via ELM and GRU with mixture correntropy loss for robust state of health estimation of lithium-ion batteries," Energy, Elsevier, vol. 284(C).
    9. Peng, Simin & Zhu, Junchao & Wu, Tiezhou & Tang, Aihua & Kan, Jiarong & Pecht, Michael, 2024. "SOH early prediction of lithium-ion batteries based on voltage interval selection and features fusion," Energy, Elsevier, vol. 308(C).
    10. Chenyuan Liu & Heng Li & Kexin Li & Yue Wu & Baogang Lv, 2025. "Deep Learning for State of Health Estimation of Lithium-Ion Batteries in Electric Vehicles: A Systematic Review," Energies, MDPI, vol. 18(6), pages 1-20, March.
    11. Dai, Houde & Wang, Jiaxin & Huang, Yiyang & Lai, Yuan & Zhu, Liqi, 2024. "Lightweight state-of-health estimation of lithium-ion batteries based on statistical feature optimization," Renewable Energy, Elsevier, vol. 222(C).
    12. Li, Xiaopeng & Zhao, Minghang & Zhong, Shisheng & Li, Junfu & Fu, Song & Yan, Zhiqi, 2024. "BMSFormer: An efficient deep learning model for online state-of-health estimation of lithium-ion batteries under high-frequency early SOC data with strong correlated single health indicator," Energy, Elsevier, vol. 313(C).
    13. Cao, Mengda & Zhang, Tao & Liu, Yajie & Zhang, Yajun & Wang, Yu & Li, Kaiwen, 2022. "An ensemble learning prognostic method for capacity estimation of lithium-ion batteries based on the V-IOWGA operator," Energy, Elsevier, vol. 257(C).
    14. Peng, Simin & Wang, Yujian & Tang, Aihua & Jiang, Yuxia & Kan, Jiarong & Pecht, Michael, 2025. "State of health estimation joint improved grey wolf optimization algorithm and LSTM using partial discharging health features for lithium-ion batteries," Energy, Elsevier, vol. 315(C).
    15. Zhang, Hao & Gao, Jingyi & Kang, Le & Zhang, Yi & Wang, Licheng & Wang, Kai, 2023. "State of health estimation of lithium-ion batteries based on modified flower pollination algorithm-temporal convolutional network," Energy, Elsevier, vol. 283(C).
    16. Zhang, Zhengjie & Cao, Rui & Zheng, Yifan & Zhang, Lisheng & Guang, Haoran & Liu, Xinhua & Gao, Xinlei & Yang, Shichun, 2024. "Online state of health estimation for lithium-ion batteries based on gene expression programming," Energy, Elsevier, vol. 294(C).
    17. Tang, Telu & Yang, Xiangguo & Li, Muheng & Li, Xin & Huang, Hai & Guan, Cong & Huang, Jiangfan & Wang, Yufan & Zhou, Chaobin, 2025. "Deep learning model-based real-time state-of-health estimation of lithium-ion batteries under dynamic operating conditions," Energy, Elsevier, vol. 317(C).
    18. Gu, Xinyu & See, K.W. & Li, Penghua & Shan, Kangheng & Wang, Yunpeng & Zhao, Liang & Lim, Kai Chin & Zhang, Neng, 2023. "A novel state-of-health estimation for the lithium-ion battery using a convolutional neural network and transformer model," Energy, Elsevier, vol. 262(PB).
    19. Ge, Dongdong & Jin, Guiyang & Wang, Jianqiang & Zhang, Zhendong, 2024. "A novel data-driven IBA-ELM model for SOH/SOC estimation of lithium-ion batteries," Energy, Elsevier, vol. 305(C).
    20. Guo, Yongfang & Yu, Xiangyuan & Wang, Yashuang & Huang, Kai, 2024. "Health prognostics of lithium-ion batteries based on universal voltage range features mining and adaptive multi-Gaussian process regression with Harris Hawks optimization algorithm," Reliability Engineering and System Safety, Elsevier, vol. 244(C).

    More about this item

    Keywords

    ;
    ;
    ;
    ;
    ;

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:eee:energy:v:335:y:2025:i:c:s0360544225039052. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Catherine Liu (email available below). General contact details of provider: http://www.journals.elsevier.com/energy .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.