Landmark Distance-Profile Adapter
Implementation & benchmark of arXiv:2607.28981 — Distance Profile Embedding for Independence and Conditional Independence Testing of Random Objects
Source paper: Distance Profile Embedding for Independence and Conditional Independence Testing of Random Objects arXiv:2607.28981 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Add a metric-aware front end that represents an arbitrary object x by its distances to a fixed set of reference objects rather than forcing x into a Euclidean or Hilbert embedding. Feed the resulting profile through a learned projection and concatenate it with the ordinary neural representation. This should be useful for graphs, trees, distributions, and sets where generic vectorization loses geometry or requires an expensive object-specific encoder.
Formulas
Mathematical statement
Definition 1 defines the distance profile embedding as \(\Phi_X(X):\Omega\to L_2(\Omega_X,\lambda_X)\), with \(\Phi_X(x)(u)=d_X(u,x)\). Here \((\Omega_X,d_X)\) is the object metric space, \(u\in\Omega_X\) is a reference object, \(\lambda_X\) is a measure over reference objects, and \(L_2(\Omega_X,\lambda_X)\) is the space of square-integrable real functions. The paper claims that this map is injective and preserves full distributional information under mild conditions, without requiring an isometric Hilbert embedding of \(\Omega_X\). Approximate the function by an empirical landmark measure \(\hat\lambda_K=\sum_{j=1}^K w_j\delta_{a_j}\), where \(a_j\) are reference objects, \(w_j\geq 0\), and \(\sum_j w_j=1\). The finite DPE is \(\hat\Phi_K(x)=[\sqrt{w_1}d_X(a_1,x),\ldots,\sqrt{w_K}d_X(a_K,x)]\), whose Euclidean inner product approximates the L2 inner product \(\langle\Phi_X(x),\Phi_X(y)\rangle_{L_2}=\int_{\Omega_X}d_X(u,x)d_X(u,y)\,d\lambda_X(u)\). The transferable property is geometric and distributional identifiability through distance profiles; the finite approximation is improved by increasing K or selecting more representative landmarks.
Implementation notes
Integrate the adapter immediately after object loading and before the main neural encoder. Inputs are objects x from a metric space with a callable distance function d_X(x,y); examples include graph edit distance, Wasserstein distance between empirical distributions, tree distance, or a point-cloud metric. First construct a landmark bank A={a_1,...,a_K} from the training set. Use uniform weights w_j=1/K initially, then optionally use stratified weights that approximate the training distribution. For each minibatch object x_i, compute D[i,j]=d_X(x_i,a_j), form P[i,j]=sqrt(w_j)*D[i,j], and standardize each landmark coordinate using training-set mean and variance. Pass P through a two-layer projection MLP and concatenate the projected profile with any existing object features.
Pseudocode:
landmarks = select_diverse_training_objects(K)
mu, sigma = estimate_coordinate_stats(landmarks, training_subset)
for x in minibatch:
p = [sqrt(w[j]) * distance(x, landmarks[j]) for j in range(K)]
p = (p - mu) / (sigma + eps)
h = projection_mlp(p)
prediction = network(h, other_inputs)
loss = task_loss(prediction, target)
backpropagate(loss)
The paper supplies the infinite-dimensional map and its injective, distribution-preserving rationale; the finite landmark approximation, coordinate normalization, landmark selection, and projection are engineering choices. Start with K=64 or 128 and compare random landmarks against farthest-point sampling under d_X. The first cheap experiment should use a small graph-classification benchmark with graph edit distance or a shortest-path-based graph metric, comparing the DPE adapter against a bag-of-statistics MLP and a standard graph encoder at matched parameter count. Success means higher test accuracy or calibration at equal training FLOPs, better performance when training data are reduced, and graceful scaling as K increases. Also test whether reducing K from 128 to 32 causes less degradation than reducing the dimension of a learned baseline embedding.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.