mlsanityData to 7 Oct 2026

Multi-scale Graph Autoregressive Modeling: Molecular Property Prediction via Next Token Prediction

arXiv 2601.02530 ↗First posted 5 Jan 2026cs.LG q-bio.QMcited by 0

Not on the map. Why?

Abstract (arXiv)

We present Connection-Aware Motif Sequencing (CamS), a graph-to-sequence representation that enables decoder-only Transformers to learn molecular graphs via standard next-token prediction (NTP). For molecular property prediction, SMILES-based NTP scales well but lacks explicit topology, whereas graph-native masked modeling captures connectivity but risks disrupting the pivotal chemical details (e.g., activity cliffs). CamS bridges this gap by serializing molecular graphs into structure-rich causal sequences. CamS first mines data-driven connection-aware motifs. It then serializes motifs via scaffold-rooted breadth-first search (BFS) to establish a stable core-to-periphery order. Crucially, CamS enables hierarchical modeling by concatenating sequences from fine to coarse motif scales, allowing the model to condition global scaffolds on dense, uncorrupted local structural evidence. We instantiate CamS-LLaMA by pre-training a vanilla LLaMA backbone on CamS sequences. It achieves state-of-the-art performance on MoleculeNet and the activity-cliff benchmark MoleculeACE, outperforming both SMILES-based language models and strong graph baselines. Interpretability analysis confirms that our multi-scale causal serialization effectively drives attention toward cliff-determining differences.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 37 of its 75 references that are in the map. First posted on arXiv 5 Jan 2026.

Builds on

37 of 75 references.

References · 37

  1. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  2. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
  3. 2017Graph Attention Networkscited by 8,854
  4. 2017Inductive Representation Learning on Large Graphscited by 6,284
  5. 2017Neural Message Passing for Quantum Chemistrycited by 3,460
  6. 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,801
  7. 2020Graph Contrastive Learning with Augmentationscited by 1,203
  8. 2018Deep Graph Infomaxcited by 1,101
  9. 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
  10. 2019Strategies for Pre-training Graph Neural Networkscited by 964
  11. 2024Emu3: Next-Token Prediction is All You Needcited by 761
  12. 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 647
25 more
  1. 2022GraphMAE: Self-Supervised Masked Graph Autoencoderscited by 443
  2. 2020ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Predictioncited by 332
  3. 2021Graph Contrastive Learning Automatedcited by 287
  4. 2021Pre-training Molecular Graph Representation with 3D Geometrycited by 248
  5. 2021ChemRL-GEM: Geometry Enhanced Molecular Representation Learning for Property Predictioncited by 204
  6. 2021Motif-based Graph Self-Supervised Learning for Molecular Property Predictioncited by 159
  7. 20213D Infomax improves GNNs for Molecular Property Predictioncited by 146
  8. 2021Self-supervised Graph-level Representation Learning with Local and Global Structurecited by 125
  9. 2022Should You Mask 15% in Masked Language Modeling?cited by 117
  10. 2022SELFIES and the future of molecular string representationscited by 76
  11. 2024A Comprehensive Survey of Foundation Models in Medicinecited by 56
  12. 2021Molecule Generation by Principal Subgraph Mining and Assemblingcited by 53
  13. 2023Rethinking Tokenizer and Decoder in Masked Graph Modeling for Moleculescited by 49
  14. 2023De Novo Molecular Generation via Connection-aware Motif Miningcited by 45
  15. 2021Molformer: Motif-based Transformer on 3D Heterogeneous Molecular Graphscited by 44
  16. 2025Nature Language Model: Deciphering the Language of Nature for Scientific Discoverycited by 35
  17. 2024Graph Transformers: A Surveycited by 32
  18. 2024HIGHT: Hierarchical Graph Tokenization for Molecule-Language Alignmentcited by 17
  19. 2025UniMoMo: Unified Generative Modeling of 3D Molecules for De Novo Binder Designcited by 11
  20. 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learningcited by 10
  21. 2024ChemFM as a Scaling Law Guided Foundation Model Pre-trained on Informative Chemicalscited by 10
  22. 2024Expressivity and Generalization: Fragment-Biases for Molecular GNNscited by 8
  23. 2024Representing Molecules as Random Walks Over Interpretable Grammarscited by 7
  24. 2024Where to Mask: Structure-Guided Masking for Graph Masked Autoencoderscited by 7
  25. 2024GraphBPE: Molecular Graphs Meet Byte-Pair Encodingcited by 7

Built on by

Not cited yet.