An effective method for generating crystal structures based on the variational autoencoder and the diffusion model

Two dimensional (2D) materials based on boron and carbon have attracted wide attention due to their unique properties. BC compounds have rich active sites and diverse chemical coordination, showing great potential in optoelectronic applications. However, due to the limitation of calculation and experimental conditions, it is still a challenging task to predict new 2D BC monolayer materials. Specifically, we utilized Crystal Diffusion Variational Autoencoder (CDVAE) and pre-trained Materials Graph Neural Network with 3-Body Interactions (M3GNet) model to generate novel and stable BCP materials. Each crystal structure was treated as a high-dimensional vector, where the encoder extracted lattice information and element coordinates, mapping the high-dimensional data into a low-dimensional latent space. The decoder then reconstructed the latent representation back into the original data space. Additionally, our designed attribute predictor network combined the advantages of dilated convolutions and residual connections, effectively increasing the model's receptive field and learning capacity while maintaining relatively low parameter count and computational complexity. By progressively increasing the dilation rate, the model can capture features at different scales. We used the DFT data set of about 1600 BCP monolayer materials to train the diffusion model, and combined with the pre-trained M3GNet model to screen the best candidate structure. Finally, we used DFT calculations to confirm the stability of the candidate structure. The results show that the combination of generative deep learning model and attribute prediction model can help accelerate the discovery and research of new 2D materials, and provide effective methods for exploring the inverse design of new two-dimensional materials. [Display omitted] We introduce a novel method that combines a Variational Autoencoder (VAE) and a diffusion model to generate crystal structures. This integrated approach, incorporating deep generative models, deep learning-based attribute prediction, and density functional theory (DFT), enables efficient exploration of the chemical space and precise prediction of material structures, offering a fresh perspective in material research and facilitating accelerated material discovery.