arXiv:2409.06080cond-mat.mtrl-scics.LG2024-09被引 16

用LLM做材料分子性质回归,仅靠化学式就能达到不错效果

Regression with Large Language Models for Materials and Molecular Property Prediction

  • 仅用分子式字符串输入,微调生成损失来预测性质
  • 在QM9数据集上性能接近随机森林和全连接网络
  • 比GPT-3.5、GPT-4o更优,适合快速原型验证

我们展示大型语言模型(LLMs)在材料与分子性质回归任务中的能力,这与传统使用方式有显著差异。在QM9数据集的多个分子性质及28种材料性质上,仅使用成分描述字符串作为输入,对LLaMA 3进行生成损失微调。结果表明,当使用SMILES表示分子时,LLaMA 3能提供有实用价值的回归结果,在QM9上表现可媲美随机森林或全连接神经网络等标准模型。然而,其误差仍为使用原子类型和坐标等细粒度表示的顶尖模型的5-10倍。在28种材料性质任务中,使用仅化学描述的输入,其精度与随机森林和元素描述符相当,但略低。值得注意的是,相比GPT-3.5和GPT-4o,LLaMA 3表现更优。该工作凸显了LLM的多功能性,表明这类生成模型有望突破传统应用,用于复杂物理现象建模,为化学、材料科学等领域的未来研究提供新路径。

原文摘要 · Abstract (English)

We demonstrate the ability of large language models (LLMs) to perform material and molecular property regression tasks, a significant deviation from the conventional LLM use case. We benchmark the Large Language Model Meta AI (LLaMA) 3 on several molecular properties in the QM9 dataset and 28 materials properties. Only composition-based input strings are used as the model input and we fine tune on only the generative loss. We broadly find that LLaMA 3, when fine-tuned using the SMILES representation of molecules, provides useful regression results which can rival standard materials property prediction models like random forest or fully connected neural networks on the QM9 dataset. Not surprisingly, LLaMA 3 errors are 5-10x higher than those of the state-of-the-art models that were trained using far more granular representation of molecules (e.g., atom types and their coordinates) for the same task. Similarly, LLaMA 3 provides comparable, although slightly worse, accuracy relative to random forest and elemental descriptors when using just compound chemical description on our set of 28 materials properties. Interestingly, LLaMA 3 provides improved predictions compared to GPT-3.5 and GPT-4o. This work highlights the versatility of LLMs, suggesting that LLM-like generative models can potentially transcend their traditional applications to tackle complex physical phenomena, thus paving the way for future research and applications in chemistry, materials science and other scientific domains.

材料预测大模型回归任务

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