Goldilocks CSLR quantile random forest for k-index prediction model
Contributors
Project manager:
Project members:
- 1. UKRI STFC
Description
A quantile random forest that predicts which mesh a periodic material needs from Goldilocks Core's ordered ladder of k-point meshes. It returns a rung on that ladder, not a k-point spacing: rung 0 is the Gamma-only (1, 1, 1) mesh, indices are 0-based, and the ladder was enumerated to 50 k-points per reciprocal-lattice axis.
Trained on PSDI record d5ds2-64f16: 17757 MC3D structures with converged Quantum ESPRESSO k-mesh studies, split 70/10/10/10 and grouped by reduced composition so that polymorphs of one composition cannot straddle the split. Inputs are 174 composition, structure, direct-lattice and reciprocal-lattice descriptors, matching the extractor Goldilocks Core already uses. No SOAP and no learned metallicity representation.
The model publishes a whole rung under a stated floor rather than its best estimate. The two directions of being wrong do not cost the same: a rung too low is an under-converged calculation that does not look wrong, while a rung too high only costs machine time. Read at its median the forest is right to the rung 44% of the time and too low 29.7% of the time. What it publishes instead is the cheapest quantile that keeps under-prediction at or below 6%, chosen on the validation split, lifted by two rungs above rung 11 so the floor holds where the model is weakest rather than only on average. On the held-out test split it comes in below the true rung 4.4% of the time, at a mean of 2.42 rungs more mesh than was needed.
Because the published value is deliberately conservative, symmetric error metrics describe the price of that safety rather than the accuracy of the estimator: test MAE is 2.63 and R-squared 0.04, while the same forest read at its median scores 1.12 and 0.73. Structures that truly need rung 11 or above are still under-converged 14.6% of the time, because 0.6% of the training labels sit above rung 20 and the model cannot place them. That limit is stated in the model card rather than smoothed over.
The record contains the fitted forest, its separately recorded calibration, and model.json, which was written by the run that fitted the model and records the feature contract, all 174 column names, the target contract, the decision rule and the rule that chose it, the hyperparameter search, and the digests.
Files
README.md
Files
(75.6 MB)
| Name | Size | Download all |
|---|---|---|
|
md5:765b638373afe829e17eb8915943640d
|
609 Bytes | Preview Download |
|
md5:d266ecfbed18607b9c6441fc8e332d24
|
75.6 MB | Download |
|
md5:6efae6f704301efe99a650325bf46670
|
1.2 kB | Preview Download |
|
md5:9563f8353761b998ecd5fc4bc8bcfca9
|
9.8 kB | Preview Download |
|
md5:fbbe29cd724971cb16967e0e17fae182
|
10.6 kB | Preview Download |