arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.20573cs.LGq-bio.BMq-bio.QMstat.ML
收藏

通过变分自编码器的不确定性引导微调增强生成式分子设计

Enhancing Generative Molecular Design via Uncertainty-guided Fine-tuning of Variational Autoencoders

Texas A&M University(德克萨斯农工大学);Brookhaven National Laboratory(布鲁克海文国家实验室)

A N M Nafiz Abeer, Sanket Jantre, Nathan M Urban, Byung-Jun Yoon

AI总结:

本文提出了一种基于主动学习的模型不确定性引导微调方法,通过在变分自编码器参数的低维活跃子空间中量化不确定性并结合黑盒优化,有效提升了生成式分子设计在下游属性优化任务中的表现。

摘要:

近年来,深度生成模型已成功应用于各种分子设计任务,特别是在生命科学和材料科学领域。对于预训练的生成式分子设计(GMD)模型而言,一个关键挑战是微调这些模型,使其更适合旨在优化特定分子属性的下游设计任务。然而,为每个新设计任务从头重新设计和训练一个现有的有效生成模型是不切实际的。此外,典型下游任务(如属性预测)的黑盒性质,使得以任务特定方式优化生成模型并非易事。在本研究中,我们提出了一种新方法,通过在主动学习环境中的性能反馈,对基于预训练变分自编码器(VAE)的GMD模型进行模型不确定性引导的微调。其核心思想是量化生成模型中的模型不确定性,通过在高维VAE参数的活跃子空间内工作来实现高效性,该低维活跃子空间解释了模型输出中的大部分变异性。引入模型不确定性通过解码器多样性扩展了可行分子的空间。随后,我们利用活跃子空间的低维特性使黑盒优化变得易于处理,从而探索由此产生的模型不确定性类别。这使我们能够识别并利用一组多样化的高性能模型来生成增强的分子。在六个目标分子属性上使用多个基于VAE的生成模型进行的实证结果表明,我们的不确定性引导微调方法始终优于原始的预训练模型。

英文摘要:

In recent years, deep generative models have been successfully adopted for various molecular design tasks, particularly in the life and material sciences. A critical challenge for pre-trained generative molecular design (GMD) models is to fine-tune them to be better suited for downstream design tasks aimed at optimizing specific molecular properties. However, redesigning and training an existing effective generative model from scratch for each new design task is impractical. Furthermore, the black-box nature of typical downstream tasks$\unicode{x2013}$such as property prediction$\unicode{x2013}$makes it nontrivial to optimize the generative model in a task-specific manner. In this work, we propose a novel approach for a model uncertainty-guided fine-tuning of a pre-trained variational autoencoder (VAE)-based GMD model through performance feedback in an active learning setting. The main idea is to quantify model uncertainty in the generative model, which is made efficient by working within a low-dimensional active subspace of the high-dimensional VAE parameters explaining most of the variability in the model's output. The inclusion of model uncertainty expands the space of viable molecules through decoder diversity. We then explore the resulting model uncertainty class via black-box optimization made tractable by low-dimensionality of the active subspace. This enables us to identify and leverage a diverse set of high-performing models to generate enhanced molecules. Empirical results across six target molecular properties, using multiple VAE-based generative models, demonstrate that our uncertainty-guided fine-tuning approach consistently outperforms the original pre-trained models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑