TensorX
返回文献探索

Paper · arXiv 2411.00871

LLaMo: Large Language Model-based Molecular Graph Assistant

Jinyoung Park, Minseong Bae, Dohwan Ko, Hyunwoo J. Kim

22 upvotesOctober 31, 2024arXiv 预印本
AI 摘要

LLaMo, a large language model for molecular graphs, uses a multi-level graph projector and cross-attention to achieve top performance in tasks like molecular description, property prediction, and IUPAC name generation.

Large Language ModelsLarge Vision-Language ModelsLLaMomolecular graph assistantend-to-end trainedmulti-level graph projectorgraph tokensGNN layermotif representationscross-attention mechanismmachine-generated molecular graph instruction datainstruction-tuningmolecular description generationproperty predictionIUPAC name prediction

Abstract

Large Language Models (LLMs) have demonstrated remarkable generalization and instruction-following capabilities with instruction tuning. The advancements in LLMs and instruction tuning have led to the development of Large Vision-Language Models (LVLMs). However, the competency of the LLMs and instruction tuning have been less explored in the molecular domain. Thus, we propose LLaMo: Large Language Model-based Molecular graph assistant, which is an end-to-end trained large molecular graph-language model. To bridge the discrepancy between the language and graph modalities, we present the multi-level graph projector that transforms graph representations into graph tokens by abstracting the output representations of each GNN layer and motif representations with the cross-attention mechanism. We also introduce machine-generated molecular graph instruction data to instruction-tune the large molecular graph-language model for general-purpose molecule and language understanding. Our extensive experiments demonstrate that LLaMo shows the best performance on diverse tasks, such as molecular description generation, property prediction, and IUPAC name prediction. The code of LLaMo is available at https://github.com/mlvlab/LLaMo.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
LLaMo: Large Language Model-based Molecular Graph Assistant | TensorX