用无标签晶体数据生成原子向量表示,提升晶体性质预测精度
CrysAtom: Distributed Representation of Atoms for Crystal Property Prediction
- 基于无监督学习从晶体数据中学习原子的密集向量表示
- 在多个晶体性质预测任务上显著提升基线模型性能
- 适合材料科学中需要高效特征提取的研究者使用
人工智能在基础科学研究中广泛应用,机器学习与深度学习方法在过去十年推动了蛋白质结构、药物靶点结合亲和力及分子性质预测等领域的显著进展。在材料科学中,晶体材料具有拓扑结构,可表示为图,利用图神经网络(GNN)可将其编码到增强的表示空间。现有框架通常依赖手工设计的原子特征与结构表示,通过监督学习预测形成能、带隙、总能等电子性质。本文提出无监督框架 CrysAtom,利用未标注的晶体数据生成原子的密集向量表示,可直接集成至现有 GNN 性质预测模型,显著提升预测性能。实验表明,该表示有效捕捉原子化学特性,增强基线模型表现。
原文摘要 · Abstract (English)
Application of artificial intelligence (AI) has been ubiquitous in the growth of research in the areas of basic sciences. Frequent use of machine learning (ML) and deep learning (DL) based methodologies by researchers has resulted in significant advancements in the last decade. These techniques led to notable performance enhancements in different tasks such as protein structure prediction, drug-target binding affinity prediction, and molecular property prediction. In material science literature, it is well-known that crystalline materials exhibit topological structures. Such topological structures may be represented as graphs and utilization of graph neural network (GNN) based approaches could help encoding them into an augmented representation space. Primarily, such frameworks adopt supervised learning techniques targeted towards downstream property prediction tasks on the basis of electronic properties (formation energy, bandgap, total energy, etc.) and crystalline structures. Generally, such type of frameworks rely highly on the handcrafted atom feature representations along with the structural representations. In this paper, we propose an unsupervised framework namely, CrysAtom, using untagged crystal data to generate dense vector representation of atoms, which can be utilized in existing GNN-based property predictor models to accurately predict important properties of crystals. Empirical results show that our dense representation embeds chemical properties of atoms and enhance the performance of the baseline property predictor models significantly.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。