[Paper Review] A generative machine learning model for designing metal hydrides applied to hydrogen storage

Xiyuan Liu, Christian Hacker+2 more|arXiv (Cornell University)|Jan 28, 2026

Hydrogen Storage and Materials0 citations

TL;DR

The paper presents a causal-discovery guided, lightweight generative ML framework (CDVAE + FCI) to design novel metal hydrides for hydrogen storage, generating 1,000 candidates and identifying six unreported formulas, four of which pass DFT validation.

ABSTRACT

Developing new metal hydrides is a critical step toward efficient hydrogen storage in carbon-neutral energy systems. However, existing materials databases, such as the Materials Project, contain a limited number of well-characterized hydrides, which constrains the discovery of optimal candidates. This work presents a framework that integrates causal discovery with a lightweight generative machine learning model to generate novel metal hydride candidates that may not exist in current databases. Using a dataset of 450 samples (270 training, 90 validation, and 90 testing), the model generates 1,000 candidates. After ranking and filtering, six previously unreported chemical formulas and crystal structures are identified, four of which are validated by density functional theory simulations and show strong potential for future experimental investigation. Overall, the proposed framework provides a scalable and time-efficient approach for expanding hydrogen storage datasets and accelerating materials discovery.

Motivation & Objective

• Identify key features causally related to hydrogen storage performance to reduce data requirements.
• Develop a lightweight generative model that can propose novel metal hydrides from small datasets.
• Integrate causal discovery with a generative model to produce crystal-structure–aware candidates.
• Validate generated candidates using DFT and a fast relaxation model to guide experimental exploration.

Proposed method

• Define a Hydrogen Storage Score combining hydrogen weight fraction and formation energy with a modified energy factor.
• Apply Fast Causal Inference (FCI) to identify the causal neighborhood of the storage score and select features.
• Train a Crystal Diffusion Variational Autoencoder (CDVAE) on 450 MP-based observations to generate new formulas and CIFs.
• Relax generated structures with M3GNet to obtain feasible crystal structures and recalculate properties.
• Use DFT (VASP) to compute formation energies for validation and filter top candidates.
• Filter and rank generated candidates to identify top materials for experimental consideration.

Experimental results

Research questions

• RQ1 Can causal discovery identify a minimal, causally relevant feature subset predictive of hydrogen storage performance in metal hydrides?
• RQ2 Can a lightweight CDVAE-based generator produce chemically valid, novel metal hydrides not in existing databases?
• RQ3 Do DFT-validatable candidates emerge from the generative process, and how many are practically feasible for hydrogen storage applications?

Key findings

• The FCI analysis identifies hydrogen weight fraction, formation energy, and crystal structure as key predictors for the hydrogen storage score.
• From 450 training samples, the CDVAE generates 1,000 candidates, with six unreported alloy hydrides identified and four passing DFT screening.
• The integrated HDL pipeline achieves DFT-compatible predictions with M3GNet-Relaxed MAE ≈ 0.0775 eV for formation energy and overall MSE ≈ 0.018 (on the generated set).
• Four validated candidates exhibit favorable hydrogen storage properties in computational tests, indicating strong potential for experimental follow-up.