论文

化学生成与基础模型的高阶分子语法

Higher-Order Molecular Grammars for Generative and Foundation Models in Chemistry

精选理由

北大团队推出HGR框架,让分子模型能直接处理环系统等高阶拓扑,生成100%有效分子,性能全面领先。

研究人员提出HGR框架,将分子提升为组合复形并解析为上下文无关高阶语法的产生式规则序列。团队构建包含118万分子的RingDiv基准测试集,引入环多样性指数(RDI)量化环系统覆盖率。在分子生成任务中,基于HGR的模型构造100%有效分子,在五个生成基准测试的FCD指标上均排名第一。在表示学习中,HGR-FM在七个MoleculeNet基准测试中取得最高平均AUC,在探测和完全微调协议下分别比最强基线提高8.3和3.3个AUC点。

原文 · arXiv cs.LG

Higher-Order Molecular Grammars for Generative and Foundation Models in Chemistry

Molecular learning models are strongly shaped by their underlying representations. Yet standard sequential and graph formalisms struggle to explicitly encode higher-order topology, such as ring systems and recurring motifs. Existing higher-order representations can capture these structures directly, but they are often computationally demanding and difficult to decode into valid molecules. Here, we introduce Higher-order Grammar Representation (HGR), a principled, topology-aware framework that lifts molecules to combinatorial complexes and parses each complex into a compact sequence of production rules under a context-free higher-order grammar. By serialising higher-order topology into rule sequences, HGR makes these structures directly compatible with standard sequence models, avoiding the computational overhead of explicit higher-order encodings while preserving topological expressiveness. To reduce benchmark bias towards simple ring systems, we construct RingDiv, a ring-enriched benchmark containing 1.18 million molecules, including the curated RingDiv300k subset, and introduce the ring diversity index (RDI) to quantify ring-system coverage. In molecular generation, HGR-based models uniquely combine 100% validity by construction with leading distributional alignment, ranking first in FCD on all five generation benchmarks. In representation learning, HGR-FM achieves the highest mean AUC across seven MoleculeNet benchmarks under both transfer protocols, improving on the strongest baseline by 8.3 and 3.3 AUC points under probing and full fine-tuning, respectively. Collectively, these results establish HGR as an efficient higher-order representation for molecular generation and transferable representation learning.