HACKOBAR_item
[arXiv]score: 0.20

Mini-Batch Class Composition Bias in Link Prediction

April 30, 2026
arXiv:2604.25978 exposes a critical training flaw in GNN-based link prediction: batch normalization layers enable models to exploit mini-batch class composition as a shortcut heuristic, bypassing genuine structural learning. After correcting this bias, node-class feature alignment improves measurably, revealing systematic overestimation of link predictor generalization. Practitioners deploying GNNs for knowledge graph completion, recommendation, or molecular property prediction should audit their mini-batch sampling strategies immediately. This reframes prior benchmark results as potentially artifactual, demanding revalidation of leading link prediction architectures.
cs.LGcs.AI