Certified Approximation for Interpretable Representer Landmarks
Organizations: Department of Computer Science Purdue University West Lafayette, IN 47907, USA
Abstract
Representer explanations rank the training landmarks that most influence a self-supervised representation. At scale, this ranking rests on up to four stacked approximations of the empirical neural tangent kernel (eNTK). These are random output heads, a parameter sketch, landmark sampling and a coefficient fit. Existing analyses bound each approximation separately, but none certifies the top- set against their combined error. We introduce CAIRN (Certified Approximation for Interpretable Representer laNdmarks), a framework that carries this error through to the ranking. We derive the exact variance of the sketched multi-head eNTK, which matches measurement within where Johnson-Lindenstrauss bounds err by up to . This yields a high-probability top- certificate for a fixed coefficient fit, alongside exact residual-trace certificates for discarded spectral mass. An exact product-variance identity separates kernel error from fit variability and identifies when a larger kernel budget can still sharpen a ranking. Stochastic Lanczos Quadrature (SLQ) estimates the effective dimension within and guides the landmark budget without dense eigendecomposition. We show that residual mass does not control class coverage, and residual-greedy selection cuts the worst coverage excess of -means++ from to ( on the sketched eNTK). Cross-view initializers outperform principal-component initialization in five (AUI) to all six (CSI) settings. Against the KREPES Gauss-Newton solver, CAIRN converges to faster, trails by at most points and gains up to points on MNIST. Together, these results make the reliability of representer explanations measurable and show where approximation budgets are best spent.
Figures & tables
| Accuracy (%) | Time to converge (s) | Identifiability | ||||||
|---|---|---|---|---|---|---|---|---|
| Dataset | Objective | CAIRN | KREPES | CAIRN | KREPES | Speedup | CAIRN / KREPES | |
| Adult | Barlow Twins | 84.10 0.01 | 84.01 | +0.09 | 0.994 | 10.5 | 10.5 | 83 / 73 |
| Adult | SimCLR | 84.08 | 84.35 | -0.27 | 2.04 | 7.76 | 3.8 | 97 / 93 |
| Adult | VICReg | 84.02 | 84.33 | -0.31 | 1.57 | 17.9 | 11.3 | 83 / 57 |
| MNIST | Barlow Twins | 96.30 | 96.31 | -0.01 | 1.39 | 7.90 | 5.7 | 100 / 100 |
| MNIST | SimCLR | 96.10 0.03 | 94.49 | +1.61 | 6.10 | 20.4 | 3.3 | 100 / 100 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Definition |
|---|---|
| Network, empirical NTK and the sketched kernel | |
| Frozen backbone network with parameters ; the eNTK is taken at this network and it is never retrained | |
| Generic inputs | |
| Number of network outputs | |
| Number of network parameters, zero-padded to a power of two for the SRHT (the unpadded count is written out, e.g. ) | |
| Jacobian of the network output with respect to the parameters at , zero-padded | |
| Symbol | Definition |
|---|---|
| Influence scores and ranking certificates | |
| , | Landmark set and its size |
| -th landmark, | |
| Query (test) point whose landmark ranking is explained | |
| Fitted Nyström coefficient matrix and its initialization; row is | |
| Coefficient displacement | |
| Symbol | Definition |
|---|---|
| Landmark selection, residual and budget | |
| , | Unlabeled data set and the number of candidate points |
| Landmark pool size ( on Adult, on MNIST) | |
| , | Selection feature matrix and its feature dimension |
| Gram matrix of the selection features | |
| -head sketched feature map , so | |
| Symbol | Definition |
|---|---|
| Coefficient initialization | |
| Output (latent) dimension of the coefficient fit ( ); also the number of retained eigenpairs | |
| Leading eigenvectors and eigenvalues of the decomposed matrix | |
| Eigenvalue jitter; relative, (KREPES’s own PCI uses absolute ) | |
| PCI | Principal-component initialization of the centered (dense or Lanczos eigensolver) |
| Approximate (Ritz) eigenpair of a symmetric matrix ( Proposition 3.9 ) | |
| Symbol | Definition |
|---|---|
| Training, evaluation and identifiability | |
| Latent ; its two-view versions and | |
| Bias of the kernel encoder (initialized to ) and its initial value; is the latent at | |
| (Alg. 3 ) | Self-supervised loss (BT, SimCLR, VICReg, BYOL) and its batch value |
| Initializer, number of epochs and early-stopping patience (Alg. 3 ) | |
| (Alg. 1 ) | KREPES bias and its initial value |
| Setting | Value |
|---|---|
| Output dimension | |
| Epochs / patience | / |
| Batch size | |
| Learning rate / weight decay | / |
| BT off-diagonal weight | |
| VICReg |
| Contribution | Leave-one-out | CAIRN seed | ||||||
|---|---|---|---|---|---|---|---|---|
| Dataset | Objective | CAIRN | KREPES | CAIRN | KREPES | CAIRN | KREPES | stability |
| Adult | Barlow Twins | 7 | 23 | 83 | 70 | 63 | 73 | 1.00 |
| Adult | SimCLR | 97 | 70 | 83 | 87 | 90 | 93 | 1.00 |
| Adult | VICReg | 0 | 57 | 83 | 57 | 67 | 33 | 1.00 |
| MNIST | Barlow Twins | 100 | 100 | 100 | 100 | 93 | 87 | 0.97 |
| MNIST | SimCLR | 100 | 100 | 100 | 0 | 100 | 0 | 1.00 |
| Random | -means++ | DPP | -center | PC-NTK | PC | |
|---|---|---|---|---|---|---|
| MNIST | 0.526 | 0.514 | 0.511 | 0.665 | 0.691 | 0.648 |
| Adult | 2.012 | 1.981 | 1.988 | 3.327 | 4.168 | 2.065 |
| Dataset | Loss | Acc (%) | Classes seen | |
|---|---|---|---|---|
| Adult | BT | 81.59 | 9 | 2/2 |
| SimCLR | 81.98 | 2 | 2/2 | |
| VICReg | 81.43 | 21 | 2/2 | |
| MNIST | BT | 90.64 | 91 | 10/10 |
| SimCLR | 93.30 | 73 | 10/10 | |
| VICReg | 93.23 | 41 | 10/10 |