MADField: Multi-fidelity Amortized Density Field for Adsorption in Nanoporous Materials

Note: For background on gas adsorption, GCMC, and classical DFT, see our earlier adsorption tutorial.

Grand canonical Monte Carlo (GCMC) estimates adsorption by sampling molecular configurations under a chosen force field. Our model, MADField, learns the corresponding density field from two sources: broad classical DFT (cDFT) data and a smaller GCMC dataset. Integrating the predicted field gives uptake without running a new particle simulation. In the paper’s ARC-MOF methane-screening benchmark, MADField recovers 95% of high-capacity targets within the top 1.7% of its ranking, with 56× higher average precision than the strongest learned baseline. The reported per-material inference speedup is about 250,000× relative to the GCMC reference; it excludes the upfront cost of generating labels and training the model.

The full method and experiments are in our paper, MADField (arXiv:2606.21284).

Material “adsorbent” discovery

Adsorption is the process by which gas molecules accumulate inside the pores of a porous solid. The solid that holds the gas is the adsorbent; the gas it holds is the adsorbate. Methane inside a porous crystal, or carbon dioxide pulled from flue gas, are both adsorption.

Finding good adsorbents is one of the large axes of materials discovery. Porous materials like zeolites, metal–organic frameworks (MOFs), covalent organic frameworks (COFs), and amorphous carbons are all actively explored. The 2025 Nobel Prize in Chemistry, awarded to Robson, Kitagawa, and Yaghi for the design and synthesis of MOFs (Nobel Prize in Chemistry, 2025), reflects how central this class of materials has become.

These porous materials span an enormous space — hundreds of thousands synthesized, and millions more proposed computationally (Wilmer et al., 2012) — but a large candidate count does not tell us which structures meet an application’s requirements. Screening must identify useful candidates within an affordable evaluation budget.

Bottleneck: evaluating candidates at scale

Evaluating a candidate means computing its gas uptake $N$: the amount of gas it adsorbs at a given temperature and pressure. Uptake is a common screening observable. Working capacity, selectivity, stability, and regeneration cost can also matter, depending on the application.

The equilibrium density field \(\rho(\mathbf{r})\) provides a spatially resolved route to uptake. Its integral is the mean particle count in a unit cell:

\[N(P, T) = \int_{\mathcal{V}} \rho_{\mathrm{eq}}(\mathbf{r})\, d\mathbf{r},\]

Here \(N\) denotes the mean count, not the fluctuating count in one snapshot. Converting that count to volumetric working capacity also requires the unit-cell volume and the specified reference-gas volume convention. GCMC can estimate the mean count directly; an explicit density grid is useful but not required for uptake alone. Two simulation routes are:

• Grand canonical Monte Carlo (GCMC) (Adams, 1975; Dubbeldam et al., 2016) — the particle-based gold standard. Randomly insert, delete, and move gas molecules in the pores over millions of steps, then average the snapshots into $\rho(\mathbf{r})$. Accurate, but several hours per material per condition.
• Classical density functional theory (cDFT) — skip the particles and solve for $\rho(\mathbf{r})$ directly, minimizing a free-energy functional (Evans, 1979) by fixed-point iteration. Faster, but rests on an approximate functional and still iterates for every case.

A practical screening covers hundreds of thousands of materials, and even cDFT — the faster of the two — costs roughly 5 GPU-minutes per material at a fixed temperature and pressure, so screening 100,000 candidates runs to about 350 GPU-days. GCMC would take far longer even with multicore CPU. And those hundreds of thousands are only the experimental structures; the space of hypothetical frameworks proposed in silico runs from millions to trillions, where brute-force simulation is hopeless from the start.

MADField predicts $\rho(\mathbf{r})$ directly

MADField is a neural network that predicts the 3D equilibrium gas density field in porous materials. Given a material, an adsorbate, and a thermodynamic condition, it outputs the full density field $\rho(\mathbf{r})$ in sub-second — and gas uptake $N$ is computed by integration.

Result 1: benchmark on a practical MOF screening pipeline

For methane storage the quantity that matters is working capacity (WC): the CH₄ a material actually delivers between a full and an empty tank — its uptake swing between the storage and depletion pressures,

\[\mathrm{WC} = N(P_{\mathrm{ads}}, T) - N(P_{\mathrm{des}}, T),\]

with $P_{\mathrm{ads}} = 65~\mathrm{bar}$, $P_{\mathrm{des}} = 5.8~\mathrm{bar}$, and $T = 298~\mathrm{K}$, so each candidate needs two uptake evaluations. We call a framework a high-capacity target when its $\mathrm{WC} \ge 200~\mathrm{cm^3/cm^3}$.

Across the full ARC-MOF database (Burner et al., 2023) of 270,583 frameworks, only 167 clear that bar — 0.06% of the database, a true needle in a haystack. Ranking all of them by predicted WC, MADField recovers 95% of the targets within the top 1.7% of candidates, at 56× higher average precision¹ than the best learned baseline,² and five orders of magnitude less cost than GCMC.

Result 2: uptake accuracy across material classes

MADField generalizes to materials it never saw during training. On in-distribution MOFs every model does reasonably well — MADField tracks the reference isotherm (uptake versus pressure at a fixed temperature) most closely, but strong baselines such as Uni-MOF (Wang et al., 2024) stay near it too. The distinction is distribution shift, not simply whether a structure was withheld from training. On material classes absent from training: for disordered solids like amorphous carbons (Thyagarajan & Sholl, 2020) — along with polymers of intrinsic microporosity (PIMs), hyper-cross-linked polymers, and kerogens³ — MADField still follows the reference closely, while the evaluated baselines have larger errors.

Result 3: density accuracy

MADField predicts the density field itself accurately, not just its integral. The mean Tanimoto similarity to the reference density is 0.996 (cDFT) and 0.965 (GCMC) across MOFs — versus 0.943 and 0.891 for the strongest density baseline, DeepAPD (Burner et al., 2026) — and on unseen amorphous carbons the margin widens to 0.885 against 0.492.

Result 4: acceleration of cDFT

Beyond replacing simulation, MADField can also speed it up. Because it predicts an unnormalized density in physical units, that prediction is a ready-made starting guess for the cDFT fixed-point solver: warm-starting from it cuts solver iterations by 2.0× and recovers convergence in 42% of cases that fail from the standard initialization.

Two ideas behind MADField’s success

Two ideas drive these results. The first is the prediction target: MADField outputs the full density field $\rho(\mathbf{r})$ rather than the scalar uptake $N$ — a single object that yields uptake, binding sites, and isotherms together, and the source of most of the accuracy. The second is how it is trained: a broad, cheap cDFT prior corrected by a little high-fidelity GCMC, which is what lets it generalize to materials it never saw. We take each in turn.

$\rho(\mathbf{r})$ as a unified target, not the scalar $N$

Most ML models for adsorption predict scalar uptake \(N\) directly. We predict the full 3D equilibrium gas density field \(\rho(\mathbf{r})\) instead, and recover uptake by integrating it:

\[N = \int_{\mathcal{V}} \rho(\mathbf{r})\,d\mathbf{r}\]

Integrating the field recovers uptake, but it carries more than its integral: where \(\rho(\mathbf{r})\) concentrates marks the binding sites, and sweeping pressure traces the full isotherm. One prediction gives uptake and a spatial density at one condition. An isotherm requires repeating the prediction at multiple pressures.

In the paper’s scalar-head ablation, the same backbone trained to regress \(N\) directly has 7.2× higher uptake MAE than the density-head model. This supports the benefit of density supervision in the tested setup; it does not establish that density targets are always preferable.

Density-field prediction is not new (Sun & Siepmann, 2024; Burner et al., 2026), but prior work has a critical gap: the density baselines discussed here predict a normalized field or use uptake as an auxiliary input. Integrating it gives 1, not uptake. Some models require the uptake \(N\) itself as an input to work around this. Neither can independently predict how much gas a material holds.

MADField predicts a number-density field in physical units (molecules per ų), without normalizing its integral to one. Integration therefore estimates the mean adsorbed count without supplying uptake as an input.

Multi-fidelity: cDFT teaches, GCMC corrects

Predicting a density field rather than a scalar makes the data problem harder: we need density-field labels, not just uptake numbers. GCMC-derived density fields are the accurate target, but expensive to generate at scale. cDFT fields are cheap and abundant, but inherit the approximation error of the cDFT functional.

We use both. MADField is first pre-trained on 280,000 cDFT calculations spanning 4,000 MOFs and nine adsorbates — we call this cDFT-only model MADField-cDFT — then fine-tuned on a GCMC dataset 14.7× smaller to give MADField-GCMC. cDFT teaches a broad prior — how geometry, adsorbate identity, and thermodynamic conditions shape the density field. GCMC corrects the remaining fidelity gap.

How good is the cheap, approximate cDFT data on its own? Good enough that MADField-cDFT — trained without a single GCMC label — already outranks every learned baseline on the screening above, and on per-MOF working-capacity accuracy it places second only to MADField-GCMC, ahead of Uni-MOF and the rest. Starting from that strong cDFT foundation, the small GCMC set then delivers a decisive gain: fine-tuning raises screening average precision 8× (MADField-cDFT → MADField-GCMC, 0.068 → 0.557). cDFT supplies the breadth, GCMC the fidelity — the full result needs both.

This is what drives generalization. We also tried training on GCMC alone, without the cDFT prior, and performance dropped sharply — most of all on out-of-distribution materials. The cDFT prior is what carries MADField to material classes it never saw during fine-tuning.

Conclusion

MADField uses a density field as a shared target for uptake prediction, spatial adsorption structure, and cDFT initialization. The scalar-head and pretraining ablations support the roles of density supervision and multi-fidelity training in the reported benchmarks.

For screening, the important result is retrieval: how many high-capacity targets appear within a limited follow-up budget. Mean uptake error alone does not capture that requirement. The ARC-MOF ranking evaluates it directly, against simulation-derived targets.

The scope remains single-component adsorption, with pretraining on the non-polar adsorbates studied in the paper. Mixtures, broader chemistry, experimental discrepancies, and the cost of retraining for new regimes need separate evaluation. Fast inference is useful for prioritization, not a guarantee that every top-ranked material will perform as predicted.

References

• Y. Kim, S. Kim, S. Ahn, H. Kim. (2026). MADField: Multi-fidelity Amortized Density Field for Adsorption in Nanoporous Materials. arXiv:2606.21284.
• Burner, J., et al. (2023). ARC-MOF: A Diverse Database of Metal-Organic Frameworks with DFT-Derived Partial Charges and Descriptors for Enhanced Machine Learning Predictions. Chemistry of Materials. ↩
• Burner, J., Marchand, O., Cicciarella, R., Gibaldi, M. & Woo, T. K. (2026). Rapid prediction of single-site adsorbate probability distributions in metal–organic frameworks using graph neural networks (DeepAPD). Digital Discovery. ↩
• Sun, Y. & Siepmann, J. I. (2024). Understanding and predicting the spatially resolved adsorption properties of nanoporous materials (SorbIIT). Journal of Chemical Theory and Computation, 20. ↩
• Wang, J., et al. (2024). A comprehensive transformer-based approach for high-accuracy gas adsorption predictions in metal–organic frameworks (Uni-MOF). Nature Communications, 15. ↩
• Kang, Y., Park, H., Smit, B. & Kim, J. (2023). A multi-modal pre-training transformer for universal transfer learning in metal–organic frameworks (MOFTransformer). Nature Machine Intelligence, 5. ↩
• Cui, J., et al. (2023). Direct prediction of gas adsorption via spatial atom interaction learning (DeepSorption). Nature Communications, 14. ↩
• Sarikas, A. P., Gkagkas, K. & Froudakis, G. E. (2024). Gas adsorption meets deep learning: voxelizing the potential energy surface of metal–organic frameworks (RetNet). Scientific Reports, 14. ↩
• Lin, E., Zhong, Y., Chen, G. & Deng, S. (2025). Unified physio-thermodynamic descriptors via learned CO₂ adsorption properties in metal–organic frameworks (IsothermNet). npj Computational Materials, 11. ↩
• Evans, R. (1979). The nature of the liquid-vapour interface and other topics in the statistical mechanics of non-uniform, classical fluids. Advances in Physics, 28(2):143–200. ↩
• Thyagarajan, R. & Sholl, D. S. (2020). A Database of Porous Rigid Amorphous Materials. Chemistry of Materials. ↩
• The Royal Swedish Academy of Sciences. (2025). The Nobel Prize in Chemistry 2025 — Susumu Kitagawa, Richard Robson, Omar M. Yaghi. NobelPrize.org. ↩
• Wilmer, C. E., et al. (2012). Large-scale screening of hypothetical metal–organic frameworks. Nature Chemistry, 4. ↩
• Adams, D. J. (1975). Grand canonical ensemble Monte Carlo for a Lennard-Jones fluid. Molecular Physics, 29. ↩
• Dubbeldam, D., Calero, S., Ellis, D. E. & Snurr, R. Q. (2016). RASPA: molecular simulation software for adsorption and diffusion in flexible nanoporous materials. Molecular Simulation, 42. ↩
• Rosen, J., et al. (2021). Machine learning the quantum-chemical properties of metal–organic frameworks for accelerated materials discovery (QMOF). Matter.

---