# GW Geodesic Mixup

- ID: 2141
- Canonical URL: https://synthcore.org/idea/2141/gw-geodesic-mixup
- API JSON: https://synthcore.org/api/idea/2141.json
- API Markdown: https://synthcore.org/api/idea/2141.md
- Verification status: unverified
- Source: [arXiv:2608.11680](https://arxiv.org/abs/2608.11680)
- Category: training
- Solves: sample-efficiency, generalization, accuracy
- ML areas: graph-nn, data-augmentation, loss, geometry
- Math tags: metric-geometry, optimal-transport, geometry, measure-theory
- Ratings: usefulness 6/10; difficulty 5/10; novelty 6/10

## Idea description

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.

## Mathematical statement

Let X=(I,omega_0,lambda) and Y=(I,omega_1,lambda) be two Z-networks, where omega_0,omega_1:I^2->Z. Let pi be an optimal coupling of their node measures and let rho:I->I times I be measurable with rho_*lambda=pi; write rho=(rho_X,rho_Y). If Z is geodesic, select for almost every pair of matched relational indices a constant-speed geodesic gamma_s:[0,1]->Z from the source value omega_0(rho_X(s)) to the target value omega_1(rho_Y(s)). Define omega_t pointwise by omega_t(s)=gamma_s(t). The paper's measurable-selection construction shows that t->omega_t is a constant-speed geodesic in the nonlinear Lebesgue space. For finite graphs, Pi is the numerical node coupling and an edge pair (i,j) is matched with (k,l) using weight Pi_{ik}Pi_{jl}. In Euclidean Z, Geo_t(z_0,z_1)=(1-t)z_0+t z_1.

## Key formulas

- $$\omega_t(x)=\gamma_x(t),\qquad \gamma_x(0)=\omega_0(x),\quad \gamma_x(1)=\omega_1(x),\quad d_Z(\gamma_x(s),\gamma_x(t))=|s-t|\,d_Z(\omega_0(x),\omega_1(x)).$$
- $$\rho_*\mathscr{L}=\pi,\qquad \rho=(\rho_X,\rho_Y).$$
- $$\omega_t(i,j)=\operatorname{Geo}_t\!\left(\omega_0(i,j),\omega_1(k,l)\right),\qquad (k,l)\sim\Pi_{ik}\Pi_{jl}.$$
- $$\operatorname{Geo}_t(z_0,z_1)=(1-t)z_0+t z_1\quad\text{for }Z=\mathbb{R}^{d},\qquad \mathcal{L}_{\mathrm{mix}}=(1-t)\mathcal{L}(f(X_0),y_0)+t\mathcal{L}(f(X_1),y_1).$$

## Implementation notes

Integrate this into the minibatch data loader for graph classification, molecular prediction, or attributed-network learning. Select two graphs X0 and X1, compute a node coupling Pi using entropic GW or a cheaper feature-based Sinkhorn cost, and sample t from Beta(alpha,alpha), starting with alpha=0.4. Construct a mixed graph with the node set of X0. For each source edge or relational pair (i,j), choose target indices k=argmax Pi[i,:] and l=argmax Pi[j,:] for a deterministic MVP, then set the mixed attribute to Geo_t(omega_0[i,j],omega_1[k,l]). For Euclidean attributes use (1-t)omega_0[i,j]+t omega_1[k,l]; for vector node features use the same correspondence and interpolation. A stochastic version samples (k,l) with probability proportional to Pi[i,k]Pi[j,l], while a soft version averages the resulting attributes. Train on the soft label (1-t)y0+t y1, or use endpoint labels if the task is not linearly compositional. The paper supplies the pointwise constant-speed geodesic construction and requires Z to be geodesic; Pi, discrete matching, and finite-size approximations are estimated numerically. First compare a small GIN against standard graph mixup, permutation augmentation, and no augmentation on synthetic permuted graphs, then MUTAG or ZINC. Measure accuracy, calibration, robustness to node deletion, and whether estimated distances to endpoints scale approximately as t and 1-t. The expected signal is improved interpolation-region accuracy and better validation performance with fewer labeled examples.

## Disclaimer

AI-generated research hypothesis, automatically tested. Not peer-reviewed.