跳到正文
arXiv:cs.LG· Hohyun Kim, Hyesung Kim, Min-hwan Oh, Seunggeun Lee·· 10 小时前

SNACK:面向概率图生成的序列化表示框架

SNACK: A Sequential Notation Framework for Probabilistic Graph Generation

AI 导读

SNACK 是一种将图线性化为字符串的概率图生成表示方法,通过编码下三角邻接矩阵的非零项,建立合法序列与有序邻接矩阵的双射,并用非法 logit 掩码保证 100% 合法生成序列。

正文

View PDF HTML (experimental)

Abstract:We introduce SNACK, a string representation for probabilistic graph generation that enables sequence models to generate and model graphs with discrete node and edge features. SNACK linearizes graphs by encoding the nonzero entries of the lower-triangular adjacency matrix, establishing a bijective correspondence between valid sequences and ordered adjacency matrices. This design enables invalid logit masking to enforce structural constraints--such as chemical valence and ordering rules--guaranteeing 100% valid generated sequences. SNACK also enables principled string-based GFlowNet training by explicitly accounting for node-ordering symmetries. Across standard graph-generation tasks and molecular distribution-learning benchmarks, SNACK achieves competitive sample quality, while substantially improving the throughput of GFlowNet training relative to graph-based generation. Probing analyses further show that models trained on SNACK encode meaningful local and global graph structure despite receiving only sequential supervision. These results position SNACK as a practical framework for probabilistic graph generation with deep sequence models.
Comments: 30 pages; Accepted at NeurIPS 2026
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:2610.19781 [stat.ML]
  (or arXiv:2610.19781v1 [stat.ML] for this version)
  https://doi.org/10.48550/arXiv.2610.19781

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Hohyun Kim [view email]
[v1] Wed, 14 Oct 2026 08:30:23 UTC (634 KB)

来源:arXiv:cs.LG · arxiv.org