arXiv:2608.03855cs.LG2026-08

让化学分子式与科学文本共享同一表示空间,提升模型理解能力

Bi-semantic Chemical Embedder for Joint Representation Learning of SMILES and Natural Language

论文配图:Bi-semantic Chemical Embedder for Joint Representation Learning of SMILES and Natural Language
图 1 · 摘自论文原文
  • 用两阶段训练融合分子结构与自然语言语义
  • 在21.9亿词的化学文献上训练,性能优于通用模型
  • 适合需要跨模态理解的药物研发与化学分析场景

Transformer模型已革新自然语言处理,基于文本的分子表示如SMILES也成功将这些架构引入化学领域。然而,领域自适应预训练常导致模型过度拟合化学语法,严重遗忘其基础语义能力。为此,我们提出CheMatE,一种面向化学的嵌入模型,可在同一表示空间中联合捕捉分子结构与领域特定自然语言。基于ModernBERT架构,CheMatE通过两阶段训练学习双语义表征:先在包含10.4B和11.5B词元的细粒度网页与ChemPile构建的长上下文科学文档语料上进行持续掩码语言建模(MLM),再通过算法生成的SMILES-文本对进行马特罗什卡对比学习(采用多重负样本排名损失,MNRL)。该设计使模型接触富含SMILES的科学文献,实现双语义理解。我们在多个下游任务上评估,涵盖分子性质预测与科学语言理解。结果表明,结合定制语料与分步训练策略,可获得强泛化、高度可迁移的表征。通过在单一文本框架中有效统一结构与上下文信号,CheMatE在专业化学模型与通用语言模型基准上均表现优异。

原文摘要 · Abstract (English)

Transformer models have revolutionized natural language processing (NLP), and text-based molecular representations like SMILES have successfully extended these architectures to chemistry. However, domain-adaptive pre-training often causes models to overfit to chemical syntax, catastrophically forgetting their foundational semantic capabilities. To address this challenge, we introduce CheMatE, a chemistry-oriented embedding model that jointly captures molecular structure and domain-specific natural language within the same representation space. Built on a ModernBERT backbone, CheMatE learns bi-semantic representations through a two-stage training procedure: continued masked language modeling (MLM) followed by a Matryoshka contrastive learning stage via Multiple Negative Ranking Loss (MNRL). First, we train the model using MLM on a novel, large-scale corpus of SMILES-annotated, long-context scientific documents that were constructed and curated from FineWeb and ChemPile (comprising 10.4B and 11.5B tokens, respectively). Subsequently, the model undergoes contrastive learning using a synthetic dataset of SMILES-text pairs algorithmically derived from our original training corpus. This design exposes the model to SMILES-enriched scientific literature, enabling bi-semantic understanding. We evaluate CheMatE across a range of downstream tasks covering molecular property prediction and scientific language understanding. Our results demonstrate that coupling our custom-curated datasets with this sequential training strategy yields robust, highly transferable representations. By effectively unifying structural and contextual signals within a single text-based framework, CheMatE achieves competitive performance across both specialized chemistry models and general-purpose language model baselines.

化学表示双语义TransformerSMILES

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