LLM2D
基于变压器增强的变分自编码器在晶体结构预测中的应用
Transformer-Enhanced Variational Autoencoder for Crystal Structure Prediction
作者: Ziyi Chen, Yang Yuan, Siming Zheng, Jialong Guo, Sihan Liang, Yangang Wang, Zongguo Wang
发布日期: 2/14/2025
arXiv ID: oai:arXiv.org:2502.09423v1

摘要

arXiv:2502.09423v1 Announce Type: cross 摘要:晶体结构是理解材料的物理和化学性质的基础。生成模型已经成为了晶体结构预测(CSP)的新范式,然而准确捕捉晶体结构的关键特征,如周期性和对称性,仍然是一个重大挑战。在本文中,我们提出了一种增强的变换器变分自编码器用于晶体结构预测(TransVAE-CSP),该模型学习稳定材料的特征分布空间,使其能够重建和生成晶体结构。TransVAE-CSP 通过结合自适应距离扩展和不可约表示来有效地捕捉晶体结构的周期性和对称性,其编码器是一个基于等变点积注意力机制的变换器网络。在碳_24、perov_5 和 mp_20 数据集上的实验结果表明,TransVAE-CSP 在各种建模指标下的结构重建和生成任务中均优于现有方法,为其在晶体结构设计和优化中的应用提供了强大的工具。