arXiv:2510.13916cs.CL2025-10被引 1

用文本生成元素嵌入,提升材料属性预测精度

Element2Vec: Build Chemical Element Representation from Text for Property Prediction

  • 从维基百科文本中提取元素信息,生成全局与局部嵌入向量
  • 针对数据稀疏问题设计自注意力测试时训练,降低预测误差
  • 适合材料科学中缺乏实验数据的元素属性研究

化学元素的准确属性数据对材料设计与制造至关重要,但许多属性因设备限制难以直接测量。传统方法依赖其他元素或相关属性进行数值预测,难以建模复杂关系,且非所有特性可表示为标量。近年虽尝试使用语言模型进行属性估计,但仍存在幻觉和可解释性差的问题。本文提出Element2Vec,从自然语言中有效表示化学元素,基于维基百科文本生成单一通用嵌入(Global)和一组属性突出的向量(Local)。由于常见描述与专业科学文本分布差异,以及仅118种已知元素导致特定属性数据极度稀疏,我们进一步设计基于自注意力的测试时训练方法,显著缓解纯回归带来的预测误差。本工作有望推动材料科学中的AI驱动发现。

原文摘要 · Abstract (English)

Accurate property data for chemical elements is crucial for materials design and manufacturing, but many of them are difficult to measure directly due to equipment constraints. While traditional methods use the properties of other elements or related properties for prediction via numerical analyses, they often fail to model complex relationships. After all, not all characteristics can be represented as scalars. Recent efforts have been made to explore advanced AI tools such as language models for property estimation, but they still suffer from hallucinations and a lack of interpretability. In this paper, we investigate Element2Vecto effectively represent chemical elements from natural languages to support research in the natural sciences. Given the text parsed from Wikipedia pages, we use language models to generate both a single general-purpose embedding (Global) and a set of attribute-highlighted vectors (Local). Despite the complicated relationship across elements, the computational challenges also exist because of 1) the discrepancy in text distribution between common descriptions and specialized scientific texts, and 2) the extremely limited data, i.e., with only 118 known elements, data for specific properties is often highly sparse and incomplete. Thus, we also design a test-time training method based on self-attention to mitigate the prediction error caused by Vanilla regression clearly. We hope this work could pave the way for advancing AI-driven discovery in materials science.

元素表示属性预测语言模型材料科学

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。