Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Pretrained battery transformer (PBT): A foundation model for universal battery life prediction

About

Early prediction of battery cycle life is essential for improving battery design, manufacturing, and deployment. However, despite encouraging results with machine learning, progress remains constrained by scarce data and data heterogeneity across battery chemistries, specifications, formation protocols, and operating conditions. Although transfer learning has been widely explored to alleviate these challenges, its effectiveness is constrained by the lack of a foundation model that can capture broadly transferable knowledge from diverse battery life data. This gap persists because integration of heterogeneous battery datasets under data scarcity is inherently challenging. Here we introduce the pretrained battery transformer (PBT), a foundation model for battery life prediction that incorporates battery-knowledge-encoded mixture-of-experts layers to learn transferable representations from heterogeneous data. PBT is pretrained on 13 lithium-ion battery datasets and subsequently adapted to downstream battery life prediction tasks through transfer learning. Across 15 datasets covering 977 batteries and 533 sets of aging conditions from lithium-ion, sodium-ion and zinc-ion batteries, PBT achieves state-of-the-art performance, surpassing the strongest competing method by 21.8% on average, with gains of up to 86.9%. Our study establishes the first foundation model for battery life prediction and provides a scalable route towards universal battery lifetime prediction systems, with broader implications for other scientific and technological domains characterized by scarce and heterogeneous data.

Ruifeng Tan, Weixiang Hong, Jia Li, Jiaqiang Huang, Tong-Yi Zhang• 2025

Related benchmarks

TaskDatasetResultRank
Battery cycle life predictionHNEI (test)
Imp97.65
1
Battery cycle life predictionMATR (test)
Imp Score51.85
1
Battery cycle life predictionMICH (test)
Imp Score76.01
1
Battery cycle life predictionXJTU (test)
Imp Score96.04
1
Battery cycle life predictionStanford (test)
Imp9.26
1
Battery cycle life predictionRWTH (test)
Imp53.14
1
Battery cycle life predictionMICH_EXP (test)
Imp94.71
1
Battery cycle life predictionTongji (test)
Imp31.39
1
Battery cycle life predictionHUST (test)
Imp18.95
1
Battery cycle life predictionSNL (test)
Imp92.17
1
Showing 10 of 15 rows

Other info

Follow for update