Aitchison Embeddings for Learning Compositional Graph Representations
Nikolaos Nakis, Chrysoula Kosma, Panagiotis Promponas, Michail Chatzianastasis, Giannis Nikolentzos
Abstract
Representation learning is central to graph machine learning, powering tasks such as link prediction and node classification. However, most graph embeddings are hard to interpret, offering limited insight into how learned features relate to graph structure. Many networks naturally admit a rolemixture view, where nodes are best described as mixtures over latent archetypal factors. Motivated by this structure, we propose a compositional graph embedding framework grounded in Aitchison geometry, the canonical geometry for comparing mixtures. Nodes are represented as simplexvalued compositions and embedded via isometric log-ratio (ILR) coordinates, which preserve Aitchison distances while enabling unconstrained optimization in Euclidean space. This yields intrinsically interpretable embeddings whose geometry reflects relative trade-offs among archetypes and supports coherent behavior under component restriction; we consider both fixed and learnable ILR bases. Across node classification and link prediction, our method achieves competitive performance with strong baselines while providing explainability by construction rather than posthoc. Finally, subcompositional coherence enables principled component restriction: removing and renormalizing subsets preserves a welldefined geometry, which we exploit via subcompositional dimensionality removal to probe how archetype groups influence representations and predictions. Code is available here: https: //github.com/Nicknakis/AICoG
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on2
Related papers
- Graph Geometry Interaction LearningShichao Zhu, Shirui Pan, Chuan Zhou, Jia Wu et al.NeurIPS 2020 · 117 citations
- Tree-Structured Orthonormal Decomposition of the Aitchison SimplexDaisuke Yamada, Qijun Zhang, Travis Pence, Barbara Bendlin et al.ICML 2026
- ProgNet: Program-Grounded Evidence Composition for Interpretable Graph ClassificationMinseok Jeon, Seunghyun Park, Jun-Gi JangKDD 2026
- Improving Compositional Generalization using Iterated Learning and Simplicial EmbeddingsYi Ren, Samuel Lavoie, Michael Galkin, Danica J. Sutherland et al.NeurIPS 2023 · 21 citations
- Geometric Graph Representation Learning via Maximizing Rate ReductionXiaotian Han, Zhimeng Jiang, Ninghao Liu, Qingquan Song et al.WWW 2022 · 24 citations
