Table 1 Benchmark datasets. List of all 28 material properties used to benchmark the ML models in this work, together with the dataset size and the original training, validation, and test set proportions.

From: Compositionally restricted attention-based network for materials property predictions

Dataset name

Source

Material property

# Samples

(Train/val/test) %

castelli

Castelli et al.36,65

Formation enthalpy (perovskites)

18928

5-fold (72/8/20)

dielectric

MP36,66,67,68

Refractive index

4764

5-fold (72/8/20)

elasticity_log10(G_VRH)

MP36,66,67,69

log10(shear modulus (VRH))

10987

5-fold (72/8/20)

elasticity_log10(K_VRH)

MP36,66,67,69

log10(bulk modulus (VRH))

10987

5-fold (72/8/20)

expt_gap

Experiment16,36

Experimental band gap

4764

5-fold (72/8/20)

jdft2d

Experiment36,70

Exfoliation energy

636

5-fold (72/8/20)

mp_e_form

MP36,66,67

Formation energy per atom

132752

5-fold (72/8/20)

mp_gap

MP36,66,67

Band gap

106113

5-fold (72/8/20)

phonons

MP36,66,67,71

Phonon frequency

1265

5-fold (72/8/20)

steels_yield

MP36,72

Steels yield strength

312

5-fold (72/8/20)

aflow__ael_bulk_modulus_vrh

AFLOW54

Bulk modulus (VRH)

4905

(70/15/15)

aflow__ael_debye_temperature

AFLOW54

Debye temperature

4905

(70/15/15)

aflow__ael_shear_modulus_vrh

AFLOW54

Shear modulus (VRH)

4905

(70/15/15)

aflow__agl_thermal_conductivity_300K

AFLOW54

Thermal conductivity

4896

(70/15/15)

aflow__agl_thermal_expansion_300K

AFLOW54

Thermal expansion

4895

(70/15/15)

aflow__Egap

AFLOW54

Band gap

27841

(70/15/15)

aflow__energy_atom

AFLOW54

Energy per atom

27844

(70/15/15)

CritExam__Ed

Bartel et al.55

Decomposition enthalpy

85014

(70/15/15)

CritExam__Ef

Bartel et al.55

Formation enthalpy

85014

(70/15/15)

mp_bulk_modulus

MP (Oct. 2018)57,66,67,69

Bulk modulus

7632

(70/15/15)

mp_elastic_anisotropy

MP (Oct. 2018)57

Ratio of elastic anisotropy

7659

(70/15/15)

mp_e_hull

MP (Oct. 2018)57

Energy above the convex hull

83983

(70/15/15)

mp_mu_b

MP (Oct. 2018)57

Magnetization of the unit cell

83973

(70/15/15)

mp_shear_modulus

MP (Oct. 2018)57,66,67,69

Shear modulus

7437

(70/15/15)

OQMD_Bandgap

OQMD6

Band gap

341696

(70/15/15)

OQMD_Energy_per_atom

OQMD6

Energy per atom

341788

(70/15/15)

OQMD_Formation_Enthalpy

OQMD6

Formation enthalpy

341788

(70/15/15)

OQMD_Volume_per_atom

OQMD6

Volume per atom

341788

(70/15/15)

  1. The materials properties listed in the top and bottom halves are Matbench and Extended datasets, respectively.