SNI-GNN: SmartNIC-Assisted Full-Graph GNN Training with In-Network Embedding Prediction
Guofan Yu, Sitian Chen, Zhenheng Tang, Xiaowen Chu, Amelie Chi Zhou
cs.LG, cs.DC
Submitted: 2026-08-06
Updated: 2026-08-10
Comments: Camera-ready version accepted at ICDE 2026. 14 pages, 18 figures
License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
Terminology
Sources
- How Powerful are Graph Neural Networks?
- Graph Neural Network Training Systems: A Performance Comparison of Full-Graph and Mini-Batch
- FragNet: A Graph Neural Network for Molecular Property Prediction with Four Levels of Interpretability
- PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication
- Open Graph Benchmark: Datasets for Machine Learning on Graphs
- FreshGNN: Reducing Memory Access via Stable Historical Embeddings for Graph Neural Network Training
- GPUs, CPUs, and... NICs: Rethinking the Network's Role in Serving Complex AI Pipelines
- DisDP: Disaggregating Compute, Network, and Storage for Model-Sharded Data-Parallel Training
- Semi-Supervised Classification with Graph Convolutional Networks
- Graph Attention Networks
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks