arXiv:2503.07655cs.LGcs.AI2025-03被引 2

用跨标记注意力融合分子文本与图结构,提升药物发现中的分子语言建模效果。

GraphT5: Unified Molecular Graph-Language Modeling via Multi-Modal Cross-Token Attention

  • 设计跨标记注意力机制,对齐SMILES序列与分子图的细粒度信息
  • 在分子描述生成与IUPAC命名预测任务中超越现有方法
  • 适合需要融合化学结构与文本信息的研究者

分子语言建模任务(如分子描述生成)在理解分子特性以支持药物发现或材料合成方面具有潜力。与通常使用分子图预测性质不同,大多数分子语言建模方法仍依赖于1维的SMILES序列,因为该任务需通过Transformer模型生成多标记序列。然而,如何将包含分子结构和空间信息的图数据与文本数据有效融合成为关键挑战。单纯输入SMILES文本和2D分子图而未解决两者在不同模态下的对齐与表征问题,难以充分利用分子结构知识。为此,我们提出GraphT5,一种融合1维SMILES文本与2维分子图表示的多模态框架。核心创新是引入跨标记注意力模块,捕捉SMILES与分子图间在细粒度标记层面的隐含关联,从而增强分子语言建模能力。大量实验包括分子描述生成、IUPAC名称预测及案例分析表明,GraphT5优于最新基线方法,验证了其在充分利用1维文本与2维图表示方面的有效性。

原文摘要 · Abstract (English)

Molecular language modeling tasks such as molecule captioning have been recognized for their potential to further understand molecular properties that can aid drug discovery or material synthesis based on chemical reactions. Unlike the common use of molecule graphs in predicting molecular properties, most methods in molecular language modeling rely heavily on SMILES sequences. This preference is because the task involves generating a sequence of multiple tokens using transformer-based models. Therefore, a main challenge is determining how to integrate graph data, which contains structural and spatial information about molecules, with text data. In addition, simply using both 1D SMILES text and 2D graph as inputs without addressing how they align and represent the molecule structure in different modalities makes it challenging to fully utilize structural knowledge about molecules. To this end, we propose GraphT5, a multi-modal framework that integrates 1D SMILES text and 2D graph representations of molecules for molecular language modeling. Specifically, we introduce a novel cross-token attention module in GraphT5 to bridge the gap arising from the fundamental differences between the two modalities of molecule representations. Cross-token attention exploits implicit information between SMILES and graphs of molecules, resulting from their interactions at a fine-grained token level that benefits molecular language modeling. Extensive experiments including molecule captioning, IUPAC name prediction tasks, and case studies show that our GraphT5 outperforms the latest baseline approaches, which validates the effectiveness of our GraphT5 in sufficiently utilizing 1D SMILES text and 2D graph representations.

分子建模多模态注意力机制

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