Enhancing Distance-Based Graph Autoencoders with Structural Penalties for Dynamic Graph Embedding
Authors: Aleksandar Tomčić, Miloš Savić, Miloš Radovanović
Organizations: Department of Mathematics and Informatics, Faculty of Sciences, University of Novi Sad, Serbia Trg Dositeja Obradovi´ca 4, 21000 Novi Sad, Serbia
Graph autoencoders (GAEs) are widely used for learning representations of dynamic graphs. However, their optimisation objectives typically do not take structural heterogeneity across nodes into account. We propose three distance-based GAE variants that incorporate structural penalties into the reconstruction loss. All variants share a two-layer Graph Convolutional Network encoder and a Euclidean-distance decoder trained with distance-based reconstruction objectives. We extend sparsity-corrected loss with two node-level regularization terms: (i) a hub penalty based on degree centrality, and (ii) a penalty based on Natural Community Local Intrinsic Dimensionality (NC-LID). The paper is motivated by prior evidence linking high NC-LID to reduced embedding quality. The proposed methods are designed to emphasize reconstruction errors for structurally ambiguous nodes. Experiments on multiple dynamic graph data sets show that incorporating NC-LID-based regularization consistently improves reconstruction performance over the baseline without structural regularization and the method using hub-aware regularization. These findings highlight NC-LID as a useful structural signal for enhancing distance-based graph autoencoders in dynamic settings.
Figures & tables
Dataset
Res.
Snaps.
Max nodes
Max edges
radoslaw-email
month
9
151
1,675
ia-hospital-ward-proximity-attr
day
5
54
492
ia-contacts_hypertext2009
day
3
102
1,062
ia-enron-employees
3-months
13
140
823
ia-primary-school-proximity-attr
day
2
241
5,923
fb-forum
month
6
815
5,838
Table 1: Experimental datasets. Max nodes and max edges refer to the largest individual snapshot.
Table 2
d
Basic
Hub
LID
fb-forum
10
0.3754
0.3562
0.4098
25
0.5797
0.5708
0.5925
50
0.6374
0.6277
0.6431
100
0.6522
0.6439
0.6622
200
0.6638
0.6600
0.6681⋆
Table 4: Mean reconstruction F1 for fb-forum , email-Eu-core-temporal , CollegeMsg , and fb-messages .
Basic
Hub
LID
Dataset
F1
d⋆
F1
d⋆
F1
d⋆
radoslaw-email
0.6099
100
0.6000
100
0.6229
100
hospital-ward
0.8207
200
0.7968
100
0.8222
200
hypertext2009
0.7290
100
0.7197
100
0.7214
100
enron-employees
0.8098
100
0.7999
100
0.8140
100
primary-school
0.7718
200
0.7706
200
0.7706
100
Table 5: Best mean reconstruction F1 per dataset and model. d⋆ is the optimal dimension. Bold marks the winner per dataset. Win count tallies best F1 victories across all 9 datasets.
Embedding structured objects into Euclidean spaces has enabled a wide range of successful machine learning applications. Such objects include words, documents, image patches, time series, and graph nodes. In contrast, embedding entire graphs remains a challenging problem. Existing methods either sustain the original order of the graph nodes or match the output nodes to the input ones, both of which create scalability issues. In this work, we propose a graph representation as a cloud of hyperballs, which allows us to define a specific, typically unique, node ordering. Based on this representation, we propose GeoGAE, an autoencoder, in which the Transformer encoder translates a hyperball cloud into a graph-level embedding, and the Transformer decoder translates the graph-level embedding back into the graph. This formulation enables the model to capture both the global graph structure and local relational patterns. We evaluate our method on multiple graph datasets, spanning various domains. The results demonstrate effectiveness of our method in encoding and reconstructing graphs from their embeddings.
Radosław Nowak, Anna Bielawska, Bogusz Stefańczyk +2
Institute of Theoretical and Applied Informatics Polish Academy of Sciences · IDEAS Research Institute · Faculty of Mathematics, Informatics and Mechanics Warsaw University of Technology
Dynamic graph learning aims to capture evolving structural and semantic patterns in real-world systems, such as fraud detection and recommender systems. Due to the scarcity of labeled data in real-world dynamic graphs, recent studies have introduced generative or contrastive paradigms (e.g., masked graph autoencoders or graph contrastive learning) to generate task-agnostic graph embeddings. However, these methods typically rely on complex edge-level reconstruction objectives and tailored graph augmentation strategies. This incurs substantial computational overhead when scaling to large-scale dynamic graphs. In this paper, we propose SG-JEPA, a joint spiking embedding predictive architecture for large-scale dynamic graphs. In contrast to existing self-supervised methods, SG-JEPA partitions nodes into context and target sets along the temporal dimension to learn embeddings that are predictive of each other via additional spatial-temporal information. Furthermore, through encoding sequential inputs into coarse-to-fine spike count embeddings, spiking neurons enable SG-JEPA to adapt to the varying computational constraints of downstream tasks. Extensive experiments demonstrate that SG-JEPA achieves competitive or even superior performance over discriminative baselines on node classification, while effectively scaling to the dynamic graph with 13 million edges. SG-JEPA avoids the complex machinery (negative sampling, graph augmentations, edge-level reconstruction, etc.), resulting in superior training efficiency and memory scalability compared with prior self-supervised dynamic graph baselines.
Huizhe Zhang, Yuchang Zhu, Huazhen Zhong +2
Sun Yat-sen University Guangzhou, China · Sun Yat-sen University Zhuhai, China
We propose MABLE (Masked Autoencoding with Bi-Lipschitz Decoding for Embeddings and Graph Metric Learning), a self-supervised framework for learning node and graph embeddings from large, heterogeneous graphs, demonstrated here on geospatial mineral-exploration data. MABLE combines masked reconstruction with fixed cosine-similarity losses that align matched augmented views while keeping unpaired embeddings well spread. A bi-Lipschitz feature decoder ties a low-dimensional reconstruction component of each node embedding to feature similarity, while matched-node consistency shapes the remaining context used by graph pooling. Lipschitz-controlled pooling helps stabilize graph-level representations under perturbations of retained node embeddings, while augmentation alignment trains robustness to masking, node dropping, and sampling variation. Across local copper and regional Arabian Shield studies, MABLE embeddings provide complementary downstream signal and produce coherent embedding-derived layers for hypothesis generation without learned discriminators or hard-negative selection.
Yaniv Shulman, Shaghayegh Akbarpour, Jack B. Muir
Fleet Space Technologies, Adelaide, Australia · Research School of Earth Sciences, Australian National University, Acton, Australia