Appendix B — Glossary#
Term |
Definition |
|---|---|
Batch size (GNN training) |
Number of query edges in one mini-batch during GNN forward + backward pass. Larger batches → higher GPU utilisation but more memory. |
Batch size (XGBoost) |
Number of embedding vectors loaded per DataIter call in batched mode. Separate from GNN batch size. |
cuGraph-PyG |
NVIDIA’s cuGraph integration with PyTorch Geometric, enabling GPU-accelerated graph sampling and message passing for large graphs. |
DDP (DistributedDataParallel) |
PyTorch’s multi-GPU training wrapper. Each process runs a full model replica; gradients are synchronised using NCCL after each backward pass. |
Edge embedding |
The fixed-size float vector produced by the GNN for each transaction edge. It encodes both the structural graph context and raw node/edge features. Passed to XGBoost for classification. |
Edge type |
A labelled triplet |
Ensemble (Triton) |
A Triton model that chains multiple sub-models in sequence, passing outputs of one as inputs to the next. |
ExtMemQuantileDMatrix |
An XGBoost data structure that transparently pages training data through disk during tree building. Requires XGBoost ≥ 3.0. |
FIL (Forest Inference Library) |
RAPIDS library for GPU-accelerated tree model inference. Used by the |
GAT (Graph Attention Network) |
A GNN variant that computes per-edge attention weights during message passing, allowing the model to focus on the most informative neighbours. |
GNN (Graph Neural Network) |
A class of neural network that operates directly on graph-structured data by iteratively aggregating information from neighbouring nodes. |
GraphSAGE (SAGE) |
A scalable GNN algorithm that samples a fixed number of neighbours per node during training and aggregates their features using a learnable aggregator (typically mean). |
Hybrid inference |
An inference mode enabled by |
Heterogeneous graph |
A graph with multiple node types (e.g., accounts and merchants) and multiple edge types. This pipeline always operates on heterogeneous graphs. |
Hidden channels |
The width (number of dimensions) of the intermediate and output feature vectors produced by each GNN layer. |
Model repository |
The directory structure that Triton reads models from. Each subdirectory is one model; numbered subdirectories (1/, 2/) are model versions. |
NCCL |
NVIDIA Collective Communications Library. Provides high-bandwidth GPU-to-GPU communication primitives used by PyTorch DDP for gradient synchronisation. |
Neighbourhood sampling |
During GNN training, instead of using the full graph neighbourhood (which could be millions of nodes), a fixed number of neighbours ( |
Node type |
A category of entity in the graph. Defined implicitly by the filename in |
PR-AUC |
Area Under the Precision-Recall curve. Summarises classifier quality across all possible decision thresholds. Preferred over ROC-AUC for class-imbalanced datasets (like fraud, where 1–5% of edges are fraudulent). |
Relationship name (rel) |
The middle part of the edge type triple. E.g., in |
Reverse edge |
For every forward edge |
SVS (Shapley Value Sampling) |
A Captum attribution method that estimates Shapley values using random permutation sampling. Used in |
Threshold (decision threshold) |
The value of P(fraud) at which a transaction is flagged as fraudulent. Stored in |
torchrun |
PyTorch’s distributed training launcher. Spawns one process per GPU and sets |
WholeGraph |
NVIDIA’s distributed graph storage library. Partitions node feature tensors across GPUs and provides transparent remote-access semantics during neighbourhood sampling. |
XGBoost |
Extreme Gradient Boosting. An ensemble learning algorithm that builds a sequence of decision trees, each correcting the residuals of the previous ones. Used as the final fraud classifier on top of GNN-derived embeddings. |