mlsanityData to 14 Oct 2026

NodeJEPA: Structure-Conditioned Latent Prediction for Node-Level Graph Self-Supervised Learning

arXiv 2608.04381 ↗First posted 5 Aug 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Self-supervised learning on graphs is largely shaped by contrastive methods that depend on carefully designed augmentations, and by generative methods that reconstruct node attributes in the input space. Both paradigms can entangle representations with low-level input statistics rather than with relational structure. Joint-embedding predictive architectures (JEPA) instead learn by predicting latent targets rather than reconstructing inputs. Recent work has explored this idea for graph-level representation learning, but how to design JEPA-style objectives for node-level tasks, and which structural signals the predictor should condition on, remains less clear. We present NodeJEPA, a joint-embedding predictive architecture for node-level graph self-supervised learning. NodeJEPA masks structure-aware k-hop ego-subgraphs and trains a context encoder to predict the latent representations of the masked nodes. These targets come from an EMA-updated target encoder with stop-gradient. A structure-conditioned predictor integrates spectral and centrality descriptors through cross-attention. Variance, covariance, and Laplacian spectral regularizers help stabilize the embedding geometry, and an optional curriculum gradually increases masking difficulty during training. Because prediction occurs in latent space, NodeJEPA does not rely on input reconstruction or hand-crafted graph augmentations. We evaluate NodeJEPA on standard node classification benchmarks under linear probing and fine-tuning protocols, and conduct ablations on masking, prediction, and regularization design choices. Our study offers a practical recipe for node-level JEPA-style latent prediction on graphs, and clarifies when structural conditioning helps representation learning. Code, configurations, and evaluation scripts are publicly available at https://github.com/OliverZ-dot/Node-Jepa.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 38 of its 53 references that are in the map. First posted on arXiv 5 Aug 2026.

Builds on

38 of 53 references.

References · 38

  1. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  2. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  3. 2017Graph Attention Networkscited by 8,923
  4. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  5. 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
  6. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  7. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
  8. 2017Inductive Representation Learning on Large Graphscited by 6,320
  9. 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
  10. 2018How Powerful are Graph Neural Networks?cited by 4,393
  11. 2017Neural Message Passing for Quantum Chemistrycited by 3,487
  12. 2019A Comprehensive Survey on Graph Neural Networkscited by 2,919
26 more
  1. 2020Exploring Simple Siamese Representation Learningcited by 2,866
  2. 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
  3. 2016node2vec: Scalable Feature Learning for Networkscited by 1,918
  4. 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
  5. 2014DeepWalk: Online Learning of Social Representationscited by 1,547
  6. 2016Variational Graph Auto-Encoderscited by 1,382
  7. 2018Deep Graph Infomaxcited by 1,106
  8. 2016Geometric deep learning: going beyond Euclidean datacited by 1,048
  9. 2018Pitfalls of Graph Neural Network Evaluationcited by 1,007
  10. 2019Strategies for Pre-training Graph Neural Networkscited by 971
  11. 2020Geom-GCN: Geometric Graph Convolutional Networkscited by 826
  12. 2020Contrastive Multi-View Representation Learning on Graphscited by 652
  13. 2019Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networkscited by 541
  14. 2020Deep Graph Contrastive Representation Learningcited by 539
  15. 2020Graph Contrastive Learning with Adaptive Augmentationcited by 519
  16. 2020GCC: Graph Contrastive Coding for Graph Neural Network Pre-Trainingcited by 463
  17. 2022GraphMAE: Self-Supervised Masked Graph Autoencoderscited by 448
  18. 2019InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximizationcited by 441
  19. 2021Graph Contrastive Learning Automatedcited by 287
  20. 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 254
  21. 2021Large-Scale Representation Learning on Graphs via Bootstrappingcited by 186
  22. 2021Self-Supervised Learning of Graph Neural Networks: A Unified Reviewcited by 184
  23. 2021From Canonical Correlation Analysis to Self-supervised Graph Neural Networkscited by 141
  24. 2023GraphMAE2: A Decoding-Enhanced Masked Self-Supervised Graph Learnercited by 132
  25. 2020Sub-graph Contrast for Scalable Self-Supervised Graph Representation Learningcited by 90
  26. 2023Graph-level Representation Learning with Joint-Embedding Predictive Architecturescited by 22

Built on by

Not cited yet.