@inproceedings{
maser2023moleclues,
title={Mole{CLUE}s: Molecular Conformers Maximally In-Distribution for Predictive Models},
author={Michael Maser and Natasa Tagasovska and Jae Hyeon Lee and Andrew Watkins},
booktitle={NeurIPS 2023 AI for Science Workshop},
year={2023},
url={https://openreview.net/forum?id=5iryL0a6x4}
}
Structure-based molecular ML (SBML) models can be highly sensitive to input geometries and give predictions with large variance.
We present an approach to mitigate the challenge of selecting conformations for such models by generating conformers that explicitly minimize predictive uncertainty. To achieve this, we compute estimates of aleatoric and epistemic uncertainties that are differentiable w.r.t. latent posteriors. We then iteratively sample new latents in the direction of lower uncertainty by gradient descent. As we train our predictive models jointly with a conformer decoder, the new latent embeddings can be mapped to their corresponding inputs, which we call MoleCLUEs, or (molecular) counterfactual latent uncertainty explanations (Antorán et al, 2020). We assess our algorithm for the task of predicting drug properties from 3D structure with maximum confidence. We additionally analyze the structure trajectories obtained from conformer optimizations, which provide insight into the sources of uncertainty in SBML.
MoleCLUEs: Molecular Conformers Maximally In-Distribution for Predictive Models
Michael Maser Natasa Tagasovska Jae Hyeon Lee Andrew Martin Watkins