Vector Field Oriented Diffusion Model for Crystal Material Generation

التفاصيل البيبلوغرافية
العنوان: Vector Field Oriented Diffusion Model for Crystal Material Generation
المؤلفون: Klipfel, Astrid, Fregier, Yaël, Sayede, Adlane, Bouraoui, Zied
سنة النشر: 2023
المجموعة: Computer Science
Condensed Matter
مصطلحات موضوعية: Condensed Matter - Materials Science, Computer Science - Artificial Intelligence, Computer Science - Machine Learning
الوصف: Discovering crystal structures with specific chemical properties has become an increasingly important focus in material science. However, current models are limited in their ability to generate new crystal lattices, as they only consider atomic positions or chemical composition. To address this issue, we propose a probabilistic diffusion model that utilizes a geometrically equivariant GNN to consider atomic positions and crystal lattices jointly. To evaluate the effectiveness of our model, we introduce a new generation metric inspired by Frechet Inception Distance, but based on GNN energy prediction rather than InceptionV3 used in computer vision. In addition to commonly used metrics like validity, which assesses the plausibility of a structure, this new metric offers a more comprehensive evaluation of our model's capabilities. Our experiments on existing benchmarks show the significance of our diffusion model. We also show that our method can effectively learn meaningful representations.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2401.05402
رقم الأكسشن: edsarx.2401.05402
قاعدة البيانات: arXiv