arXiv AI By Matteo Tolloso, Davide Bacciu, Luigi Bonati

Multi-Fidelity Training Reshapes Uncertainty Decomposition in Atomistic Models

Read the original on arXiv AI →

The study examines how multi‑fidelity training—using abundant low‑fidelity data and scarce high‑fidelity labels—affects uncertainty decomposition in atomistic models. By training deep ensembles of mean‑variance MACE models on ANI‑1x, the authors compare high‑fidelity training from scratch with low‑fidelity pretraining followed by fine‑tuning, evaluating aleatoric, epistemic, and total uncertainty under unseen molecular systems and higher‑energy configurations. Results show that pretraining lowers prediction error and improves uncertainty‑based ranking, but does not produce well‑calibrated predictive variances, especially under energy extrapolation, and that fine‑tuning yields a more effective uncertainty‑based acquisition function.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv AI
Oct 9

Zatom-2: Multitask Pretraining on Atomistic Data for Generative Modeling across Domains

Zatom-2 is a multitask generative model for atomistic data that has been pretrained on about five million structures from the OMol25 and OMat24 datasets. It uses a multiscale Transformer with conditional flow matching to support tasks such as generation, structure prediction, and energy/force prediction for both molecules and materials. The model outperforms its predecessor, Zatom-1, on molecular distribution fidelity and benchmark generation tasks, and improves protein backbone designability from 67.8% to 74.8% after finetuning on 2,000 protein domains.

By Miruna Cretu, Alex Abrudan, Antonia Panescu, Tynan Perez, Rishabh Anand, N. Benjamin Erichson, Michael W. Mahoney, Samuel Blau, Joseph Jacobson, Rafael G\'omez-Bombarelli, Rex Ying, Tuomas Knowles, Pietro Li\`o, Alex Morehead
arXiv Machine Learning
Jul 1

ElemeNet: Multiscale Molecular Machine Learning with Uncertainty Quantification Across the Periodic Table

arXiv:2606. 30961v1 Announce Type: cross Abstract: Advances in deep learning architectures and representations have enabled ML-driven chemical property prediction, but state-of-the-art (SOTA) models have remained largely confined to independent codebases and lack support for diverse chemical species.

By Jacob W. Toney, Samir Darouich, Yiran Wang, Aaron G. Garrison, Johannes K\"astner, Heather J. Kulik
arXiv Machine Learning
Jul 7

AquaGen: Scaling generative models to molecular dynamics precision on thousands of atoms

arXiv:2607. 03513v1 Announce Type: cross Abstract: We present AquaGen, the first all-atom, explicit solvent, periodic-boundary-condition-aware generative model that produces molecular configurations from the Boltzmann distribution at a fraction of the cost of molecular dynamics (MD).

By Emmanuel Bengio, Sanjeev Raja, Yui Tik Pang, Kerstin Klaeser, Cristian Gabellini, Nikhil Shenoy, Francesco Di Giovanni, Prudencio Tossou