Aitchison Embeddings for Learning Compositional Graph Representations
Pith reviewed 2026-07-01 07:41 UTC · model grok-4.3
The pith
Graph nodes represented as mixtures over latent archetypes via Aitchison geometry produce embeddings that are competitive on standard tasks and interpretable by construction.
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
Nodes are represented as simplex-valued 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.
What carries the argument
Isometric log-ratio (ILR) coordinates that map simplex compositions to Euclidean vectors while exactly preserving Aitchison distances and subcompositional coherence.
If this is right
- Embeddings are interpretable by construction because each coordinate directly encodes relative contributions of the archetypes.
- Subcompositional coherence permits principled removal and renormalization of component subsets while keeping distances well-defined.
- The same geometry supports both fixed and learnable ILR bases without changing the overall optimization setup.
- Performance remains competitive with strong baselines on node classification and link prediction across tested graphs.
Where Pith is reading between the lines
- The subcomposition mechanism could be used to test which groups of archetypes most affect downstream predictions by systematically ablating them.
- If real networks frequently exhibit role-mixture structure, the same Aitchison embedding could be applied to other compositional data such as topic distributions or chemical mixtures.
- Learnable ILR bases might reveal dataset-specific archetype contrasts that are not apparent with fixed bases.
Load-bearing premise
Many networks naturally admit a role-mixture view where nodes are best described as mixtures over latent archetypal factors.
What would settle it
On a graph where node roles cannot be expressed as mixtures over a small fixed set of archetypes, the method would show clear accuracy drops relative to standard non-compositional embeddings on both node classification and link prediction.
Figures
read the original 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 role-mixture 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 simplex-valued 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 post-hoc. Finally, subcompositional coherence enables principled component restriction: removing and renormalizing subsets preserves a well-defined geometry, which we exploit via subcompositional dimensionality removal to probe how archetype groups influence representations and predictions.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes a compositional graph embedding framework based on Aitchison geometry. Nodes are represented as simplex-valued compositions over latent archetypal factors and embedded using isometric log-ratio (ILR) coordinates (both fixed and learnable bases) to enable Euclidean optimization while preserving Aitchison distances and subcompositional coherence. The central empirical claim is that the resulting embeddings achieve competitive performance on node classification and link prediction relative to strong baselines, with the added property of intrinsic explainability rather than post-hoc analysis.
Significance. If the empirical competitiveness and geometric properties hold under scrutiny, the work supplies a principled inductive bias for role-mixture structures in networks. The explicit use of Aitchison geometry and ILR embeddings for compositional data, together with subcompositional dimensionality removal, constitutes a concrete strength that could improve interpretability in graph representation learning.
major comments (1)
- [Abstract] Abstract: the assertion of 'competitive performance with strong baselines' is load-bearing for the central claim yet is presented without any visible quantitative results, error bars, dataset names, or statistical details; this prevents verification that the performance advantage (or parity) is robust rather than an artifact of particular splits or baselines.
minor comments (2)
- [Abstract] The motivation paragraph treats the role-mixture view as a modeling choice rather than deriving it from graph properties; a brief justification or reference to when this inductive bias is expected to be appropriate would improve clarity.
- [Abstract] Notation for the ILR transformation and the distinction between fixed versus learnable bases is introduced without an explicit equation or small illustrative example, making the transition from simplex to Euclidean space harder to follow on first reading.
Simulated Author's Rebuttal
We thank the referee for their thoughtful review and for highlighting the need for greater transparency in the abstract. We agree that the central empirical claim requires concrete support and will revise the abstract accordingly while preserving its brevity. No standing objections remain after addressing this point.
read point-by-point responses
-
Referee: [Abstract] Abstract: the assertion of 'competitive performance with strong baselines' is load-bearing for the central claim yet is presented without any visible quantitative results, error bars, dataset names, or statistical details; this prevents verification that the performance advantage (or parity) is robust rather than an artifact of particular splits or baselines.
Authors: We concur that the abstract should provide sufficient quantitative grounding for the competitiveness claim. In the revised version we will insert concise performance highlights (e.g., mean accuracy or AUC with standard deviation across splits), name the primary datasets (Cora, CiteSeer, PubMed, and the link-prediction benchmarks), and reference the number of runs or statistical tests used, while remaining within abstract length limits. The full tables and error-bar figures already present in the experimental section will be cross-referenced. revision: yes
Circularity Check
No significant circularity detected
full rationale
The derivation applies the established Aitchison geometry and isometric log-ratio (ILR) transformation—standard tools from compositional data analysis—to node representations on the simplex as an explicit modeling choice. Performance and explainability claims rest on downstream empirical evaluation against external baselines on node classification and link prediction tasks, not on any internal redefinition or self-referential fitting. The role-mixture motivation is presented as an inductive bias rather than a derived theorem, and subcompositional coherence follows directly from the properties of the ILR map without reducing predictions to the inputs by construction. No self-citation chains, fitted parameters renamed as predictions, or ansatzes smuggled via prior work appear in the provided derivation chain.
Axiom & Free-Parameter Ledger
axioms (1)
- domain assumption Many networks naturally admit a role-mixture view where nodes are mixtures over latent archetypal factors.
Forward citations
Cited by 1 Pith paper
-
Rank Is Not Capacity: Spectral Occupancy for Latent Graph Models
Spectra defines and controls effective capacity in graph embeddings via the Shannon effective rank of a trace-normalized kernel spectrum, making capacity a post-fit property rather than a pre-training hyperparameter.
Reference graph
Works this paper leans on
-
[1]
Learning Role-based Graph Embeddings
Ahmed, N. K., Rossi, R., Lee, J. B., Willke, T. L., Zhou, R., Kong, X., and Eldardiry, H. Learning role-based graph embeddings.arXiv preprint arXiv:1802.02896,
work page internal anchor Pith review Pith/arXiv arXiv
-
[2]
Mixed membership stochastic blockmodels
URL https://arxiv.org/abs/0705.4485. Aitchison, J. The Statistical Analysis of Compositional Data.Journal of the Royal Statistical Society: Series B (Methodological), 44(2):139–160,
work page internal anchor Pith review Pith/arXiv arXiv
-
[3]
Explainability Techniques for Graph Convolutional Networks
Baldassarre, F. and Azizpour, H. Explainability tech- niques for graph convolutional networks.arXiv preprint arXiv:1905.13686,
work page internal anchor Pith review Pith/arXiv arXiv 1905
-
[4]
doi: 10.1023/A:1023818214614. Epasto, A. and Perozzi, B. Is a Single Embedding Enough? Learning Node Representations that Capture Multiple Social Contexts. InThe World Wide Web Conference, pp. 394–404,
-
[5]
URL https://arxiv. org/abs/2201.05197. Grover, A. and Leskovec, J. node2vec: Scalable Feature Learning for Networks. InProceedings of the 22nd ACM SIGKDD International Conference on Knowledge Dis- covery and Data Mining, pp. 855–864,
-
[6]
URL https: //doi.org/10.1198/016214502388618906
doi: 10.1198/016214502388618906. URL https: //doi.org/10.1198/016214502388618906. Holland, P. W., Laskey, K. B., and Leinhardt, S. Stochastic blockmodels: First steps.Social Net- works, 5(2):109–137,
-
[7]
doi: https://doi.org/10.1016/0378-8733(83)90021-7
ISSN 0378-8733. doi: https://doi.org/10.1016/0378-8733(83)90021-7. URL https://www.sciencedirect.com/ science/article/pii/0378873383900217. Jin, J., Ke, Z. T., and Luo, S. Mixed membership estima- tion for social networks,
-
[8]
URL https://arxiv. org/abs/1708.07852. Kingma, D. P. and Ba, J. Adam: A method for stochastic optimization.arXiv preprint arXiv:1412.6980,
-
[9]
Semi-Supervised Classification with Graph Convolutional Networks
Kipf, T. Semi-supervised classification with graph con- volutional networks.arXiv preprint arXiv:1609.02907,
work page internal anchor Pith review Pith/arXiv arXiv
-
[10]
Lin, C., Sun, G. J., Bulusu, K. C., Dry, J. R., and Hernandez, M. Graph neural networks including sparse interpretabil- ity.arXiv preprint arXiv:2007.00119,
-
[11]
Nakis, N., Celikkanat, A., Boucherie, L., Djurhuus, C., Burmester, F., Holmelund, D
URL https://arxiv.org/abs/2206.03463. Nakis, N., Celikkanat, A., Boucherie, L., Djurhuus, C., Burmester, F., Holmelund, D. M., Frolcov ´a, M., and Mørup, M. Characterizing polarization in social networks using the signed relational latent distance model. In Ruiz, F., Dy, J., and van de Meent, J.-W. (eds.),Proceedings of The 26th International Conference o...
-
[12]
Perozzi, B., Al-Rfou, R., and Skiena, S
URL https://arxiv.org/ abs/2304.10851. Perozzi, B., Al-Rfou, R., and Skiena, S. DeepWalk: On- line Learning of Social Representations. InProceedings of the 20th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 701–710,
-
[13]
Don’t walk, skip! online learning of multi-scale network em- beddings
Perozzi, B., Kulkarni, V ., Chen, H., and Skiena, S. Don’t walk, skip! online learning of multi-scale network em- beddings. InProceedings of the 2017 IEEE/ACM In- ternational Conference on Advances in Social Networks Analysis and Mining, pp. 258–265,
2017
-
[14]
URL https://arxiv. org/abs/2005.07959. Rozemberczki, B., Kiss, O., and Sarkar, R. Karate Club: An API Oriented Open-source Python Framework for Unsupervised Learning on Graphs. InProceedings of the 29th ACM International Conference on Information and Knowledge Management (CIKM ’20), pp. 3125–3132. ACM,
-
[15]
Ying, C., Cai, T., Luo, S., Zheng, S., Ke, G., He, D., Shen, Y ., and Liu, T.-Y
doi: 10.1086/226141. Ying, C., Cai, T., Luo, S., Zheng, S., Ke, G., He, D., Shen, Y ., and Liu, T.-Y . Do transformers really perform badly for graph representation?Advances in neural information processing systems, 34:28877–28888,
-
[16]
=σ(α−g(∥x i −x j∥2)), x i ∈R k−1. Proof. Since ILR is bijective, for any collection {xi}n i=1 ⊂R k−1 there exists a unique collection {zi}n i=1 ⊂∆ k−1 ◦ such thatx i = ILR(zi)for alli, and converselyz i = ILR−1(xi)exists for alli. BecauseILRis an isometry, for alli < j, ∥xi −x j∥2 =∥ILR(z i)−ILR(z j)∥2. Substituting this identity into the respective expre...
2008
-
[17]
Link prediction.For link prediction, we follow the widely adopted evaluation protocol of Perozzi et al
For SLIM-RAA, HM-LDM, MMSBM, and SIMPLEX-EUCLIDEAN, we optimized the negative Bernoulli log-likelihood (matching AICOG up to the model-specific log-odds parameterization) using learning rate 0.05 for 5,000 epochs, so differences are attributable only to the log-odds form. Link prediction.For link prediction, we follow the widely adopted evaluation protoco...
2014
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.