KDDFeb, 2024

利用多模式先验的 VAE - 回归进行材料微结构设计

TL;DR我们提出了一种基于变分自编码器 (VAE) 的模型,用于构建前向和逆向的结构 - 性质联系,这是计算材料科学中至关重要的问题。我们的模型通过一个双层先验条件于回归变量来系统地结合 VAE 和回归模型。回归损失与变分自编码器的重构损失一起进行优化,学习与属性预测和重构相关的微观结构特征。该模型可以用于前向和逆向预测,即预测给定微观结构的属性以及预测获取给定属性所需的微观结构。由于逆向问题存在不适定性(多对一关系),我们使用多模态高斯混合先验导出了目标函数,使模型能够推断出多个适用于一组目标属性的微观结构。我们显示,在前向预测方面,我们的模型与最先进的仅前向模型一样准确。此外,我们的方法实现了直接逆向推断。我们表明,使用我们的模型推断的微观结构能够相当准确地实现所需的属性,避免了昂贵的优化过程。