Replace uniform minibatch sampling by a fixed-size determinantal point process whose similarity matrix is a diffusion kernel on the training-data k-NN graph. The sampler repels nearby or redundant examples while preserving multiple diffusion modes, so a small batch should cover intrinsic data geometry and provide lower-variance estimates of losses and gradients.
Replace the arbitrary labelled target graph in each flow-matching pair by the node permutation that minimizes its Euclidean distance to the source graph. Construct the conditional interpolation and velocity target using this aligned representative, so the model learns transport in graph quotient space rather than wasting capacity on node-order mismatches.
Use a minibatch-level transport plan that jointly decides which source graph should be paired with which target graph and how each target should be node-aligned. This can reduce total flow-matching displacement beyond per-example matching, producing shorter and less conflicting training trajectories without modifying the architecture.
Generate relational training examples by first coupling two graphs or kernels and then interpolating every matched pair of node or edge attributes along a geodesic in Z. Unlike ordinary mixup, this preserves the intrinsic Gromov-Wasserstein correspondence and produces constant-speed paths in the relational metric when Z is geodesic.