Regularization Strategy for Battery Materials Using Lattice Scaling
Main Article Content
Abstract
Artificial iIntelligence (AI) has emerged as a powerful tool for accelerating materials discovery; however, its effectiveness in battery research remains constrained by the limited size and heterogeneity of available materials datasets. In this work, we introduce a physically informed regularization strategy based on lattice scaling of crystalline structures to reduce overfitting and enhance machine-learning performance in lithium-ion battery materials prediction. The proposed framework generates structurally perturbed variants through controlled isotropic and anisotropic modifications of unit-cell volumes within experimentally realistic limits (±5-30%), mimicking chemo-mechanical variations occurring during electrochemical cycling. The method is evaluated using Crystal Graph Convolutional Neural Networks (CGCNN) trained on electrode materials from the Materials Project Battery Explorer database to predict seven electrochemical and four intrinsic material properties. We demonstrate that lattice scaling - particularly anisotropic scaling within the physically meaningful range of ±5% volume variation - systematically improves predictive accuracy compared with both baseline oversampling and established crystal regularization techniques, achieving improvements of up to 10.5% in the MAD/MAE ratio for key electrochemical descriptors. The results highlight the importance of physically meaningful transformations in materials model regularization and reveal a property-dependent response to regularization strategies. The proposed approach provides a simple, computationally efficient, and architecture-independent pathway to improve data efficiency and generalization in AI-driven battery materials discovery.
Downloads
Article Details
Copyright (c) 2026 Abenza E, et al.

This work is licensed under a Creative Commons Attribution 4.0 International License.
Goodenough JB, Kim Y. Challenges for Rechargeable Li Batteries. Chem Mater. 2010;22:587-603. Available from: https://doi.org/10.1021/cm901452z
Buriak JM, Toro C. Rising to the Challenge: John B. Goodenough and Youngsik Kim, and “Challenges for Rechargeable Li Batteries”. Chem. Mater. 2015;27:5149-5150. Available from: https://pubs.acs.org/doi/10.1021/acs.chemmater.5b02863
Sendek AD, Yang Q, Cubuk ED, Duerloo KN, Cui Y, Reed EJ. Holistic computational structure screening of more than 12 000 candidates for solid lithium-ion conductor materials. Energy Environ Sci. 2017;10:306-320. Available from: https://doi.org/10.1039/C6EE02697D
Batra R, Song L, Ramprasad R. Emerging materials intelligence ecosystems propelled by machine learning. Nat Rev Mater. 2020. Available from: https://doi.org/10.1038/s41578-020-00255-y
Lombardo T, Duquesnoy M, El-Bouysidy H, Årén F, Gallo-Bueno A, Jørgensen PB, et al. Artificial Intelligence Applied to Battery Research: Hype or Reality? Chem Rev. 2022;122)(12):10899-10969. Available from: https://doi.org/10.1021/acs.chemrev.1c00108
Liu Y, Zhao T, Ju W, Shi S. Materials discovery and design using machine learning, J Mat. 2017;3:159-177. Available from: https://doi.org/10.1016/j.jmat.2017.08.002
Barrett DH, Haruna A. Artificial intelligence and machine learning for targeted energy storage solutions. Curr Opin Electrochem. 2020;21:160-166. Available from: https://doi.org/10.1016/j.coelec.2020.02.002
Materials Genome Initiative for Global Competitiveness. 2011. Available from: chrome-extension://efaidnbmnnnibpcajpcglclefindmkaj/https://www.mgi.gov/sites/mgi/files/materials_genome_initiative-final.pdf
Schmidt J, Marques MRG, Botti S, Marques MAL. Recent advances and applications of machine learning in solid-state materials science. NPJ Comput Mater. 2019;5:83. Available from: https://doi.org/10.1038/s41524-019-0221-0
Kusne AG, Yu H, Wu C, Zhang H, Hattrick-Simpers J, DeCost B, et al. On-the-fly closed-loop materials discovery via Bayesian active learning. Nat Commun. 2020;11:5966. Available from: https://doi.org/10.1038/s41467-020-19597-w
Woo S, Shenvi RA. Natural Product Synthesis through the Lens of Informatics. Acc Chem Res. 2021;54(5):1157-1167. Available from: https://doi.org/10.1021/acs.accounts.0c00791
Morgan D, Jacobs R. Opportunities and Challenges for Machine Learning in Materials Science. Annu Rev Mater Res. 2020;50(1):71-103. Available from: https://doi.org/10.1146/annurev-matsci-070218-010015
Zhang Y, Ling C. A strategy to apply machine learning to small datasets in materials science. NPJ Comput Mater. 2018;4:25. Available from: https://doi.org/10.1038/s41524-018-0081-z
Puchala B, Tarcea G, Marquis EA, Hedstrom M, Jagadish HV, Allison JE. The Materials Commons: A Collaboration Platform and Information Repository for the Global Materials Community. JOM. 2016;68:2035-2044. Available from: https://doi.org/10.1007/s11837-016-1998-7
Bruno I, Gražulis S, Helliwell JR, Kabekkodu SN, McMahon B, Westbrook J. Crystallography and Databases. Data Sci J. 2017;16:38. Available from: https://doi.org/10.5334/dsj-2017-038
Allen FH, Shields GP. Crystallographic Databases and Knowledge Bases in Materials Design. Nato Science Series. 1999. Available from: https://doi.org/10.1007/978-94-011-4653-1_21
Groom CR, Bruno IJ, Lightfoot MP, Ward SC. The Cambridge Structural Database. Acta Crystallogr B Struct Sci Cryst Eng Mater. 2016;72:171-179. Available from: https://doi.org/10.1107/S2052520616003954?urlappend=%3Futm_source%3Dresearchgate.net%26utm_medium%3Darticle
Hachmann J, Olivares-Amaya R, Atahan-Evrenk S, Amador-Bedolla C, Sánchez-Carrera RS, Gold-Parker A, et al. The Harvard Clean Energy Project: Large-Scale Computational Screening and Design of Organic Photovoltaics on the World Community Grid. J Phys Chem Lett. 2011;2:2241-2251. Available from: https://doi.org/10.1021/jz200866s
Magar R, Wang Y, Lorsung C, Liang C, Ramasubramanian H, Li P, et al. AugLiChem: Data Augmentation Library of Chemical Structures for Machine Learning. 2021). Available from: https://doi.org/10.48550/arXiv.2111.15112
Maharana K, Mondal S, Nemade B. A Review: Data Pre-Processing and Data Augmentation Techniques. Global Transitions Proceedings. 2022;3(1):91-99. Available from: https://doi.org/10.1016/j.gltp.2022.04.020
Shorten C, Khoshgoftaar TM. A survey on Image Data Augmentation for Deep Learning. J Big Data. 2019;6:60. Available from: https://doi.org/10.1186/s40537-019-0197-0
Magar R, Wang Y, Lorsung C, Liang C, Ramasubramanian H, Li P, et al. AugLiChem: Data Augmentation Library of Chemical Structures for Machine Learning ArXiv. 2021. Available from: https://ui.adsabs.harvard.edu/link_gateway/2022MLS&T...3d5015M/doi:10.1088/2632-2153/ac9c84
Ishidzu K, Oka Y, Nakamura T. Lattice volume change during charge/discharge reaction and cycle performance of Li[NixCoyMnz]O2. Solid State Ionics. 2016;288:176-179. Available from: https://doi.org/10.1016/j.ssi.2016.01.009
Kriston A, Ruiz V, Pfrang A, Kersys A, Marinaro M, Stegmaier P, et al. On the correlation between volume change of anode materials in Li-ion cells and their degradation and failure. Collection of open conferences in research transport. 2018;2018:128. Available from: https://www.scipedia.com/public/Kriston_et_al_2018a
Koerver R, Zhang W, de Biasi L, Schweidler S, Kondrakov AO, Kolling S, et al. Chemo-mechanical expansion of lithium electrode materials – on the route to mechanically optimized all-solid-state batteries. Energy Environ Sci. 2018;11:21422158. Available from: https://doi.org/10.1039/C8EE00907D
Xie T, Grossman JC. Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties. Phys Rev Lett. 2018;120:145301. Available from: https://doi.org/10.1103/PhysRevLett.120.145301
Magar R, Wang Y, Farimani AB. Crystal Twins: Self-supervised Learning for Crystalline Material Property Prediction. 2022NPJ Comput Mat. Available from: https://doi.org/10.1038/s41524-022-00921-5
Battery Explorer-Materials Project. Available from: https://materialsproject.org/batteries
Hautier G, Fischer CC, Jain A, Mueller T, Ceder G. Finding Nature’s Missing Ternary Oxide Compounds Using Machine Learning and Density Functional Theory. Chem Mater. 2010;22:3762-3767. Available from: https://doi.org/10.1021/cm100795d
Urban A, Seo DH, Ceder G. Computational understanding of Li-ion batteries. NPJ Comput Mater. 2016;2:16002. Available from: https://doi.org/10.1038/npjcompumats.2016.2https://doi.org/10.1038/npjcompumats.2016.2
Wang Y, Richards WD, Ong SP, Miara LJ, Kim JC, Mo Y, et al. Design principles for solid-state lithium superionic conductors. Nature Mater. 2015;14(10): 1026-1031. Available from: https://doi.org/10.1038/nmat4369
Urban A, Lee J, Ceder G. The Configurational Space of Rocksalt-Type Oxides for High-Capacity Lithium Battery Electrodes. Adv Energy Mater. 2014;4:1400478. Available from: https://doi.org/10.1002/aenm.201400478
Ceder G. Opportunities and challenges for first-principles materials design and applications to Li battery materials. MRS Bull. 2010;35:693-701. Available from: https://doi.org/10.1557/mrs2010.681
Hautier G. Prediction of new battery materials based on ab initio computations. 2016. Available from: https://doi.org/10.1063/1.4961901
Urban A, Seo DH, Ceder G. Computational understanding of Li-ion batteries. NPJ Comput Mater. 2016;2:16002. Available from: https://doi.org/10.1038/npjcompumats.2016.2
Ong SP, Richards WD, Jain A, Hautier G, Kocher M, Cholia S, et al. Python Materials Genomics (pymatgen): A robust, open-source Python library for materials analysis. Comput Mater Sci. 2013;68:314-319. Available from: https://doi.org/10.1016/j.commatsci.2012.10.028
Choudhary K, DeCost B. Atomistic Line Graph Neural Network for improved materials property predictions. NPJ Comput Mater. 2021;7:185. Available from: https://doi.org/10.1038/s41524-021-00650-1
Balestriero R, Bottou L, LeCun Y. The Effects of Regularization and Data Augmentation are Class Dependent. 2022. Available from: https://proceedings.neurips.cc/paper_files/paper/2022/hash/f73c04538a5e1cad40ba5586b4b517d3-Abstract-Conference.html
Ong SP, Jain A, Hautier G, Kang B, Ceder G. Thermal stabilities of delithiated olivine MPO4 (M=Fe, Mn) cathodes investigated using first principles calculations. Electrochemistry Communications. 2010;12:427-430. Available from: https://doi.org/10.1016/j.elecom.2010.01.010
Wang L, Maxisch T, Ceder G. A First-Principles Approach to Studying the Thermal Stability of Oxide Cathode Materials. Chem Mater. 2007;19: 543-552. Available from: https://pubs.acs.org/doi/10.1021/cm0620943
Zhou F, Cococcioni M, Marianetti CA, Morgan D, Ceder G. First-principles prediction of redox potentials in transition-metal compounds with LDA + U. Phys Rev B. 2004;7:235121. Available from: https://doi.org/10.1103/PhysRevB.70.235121
Adams S, Rao RP. High power lithium ion battery materials by computational design: High power Li ion battery materials by computational design. Phys Status Solidi A. 2011;208:1746-1753. Available from: https://doi.org/10.1002/pssa.201001116