GP-DPP Adaptive Token Selection
Implementation & benchmark of arXiv:2608.23574 — InfoDPP-PAC: Principled Patch Selection for Whole Slide Image Analysis
Source paper: InfoDPP-PAC: Principled Patch Selection for Whole Slide Image Analysis arXiv:2608.23574 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace fixed-budget token or patch pruning with greedy selection that combines a teacher-derived relevance score and Gaussian-process mutual information. Select an item when it is both relevant and non-redundant, and stop when the largest remaining information gain falls below a calibrated threshold instead of retaining a fixed number of items.
Formulas
Mathematical statement
Let candidate embeddings be x_i in R^d, teacher observations be y_i=f(x_i)+epsilon_i with independent epsilon_i~N(0,sigma_n^2), and k(x,x') be a positive-definite kernel. For observed candidate set X and labels y, the GP posterior mean and covariance are mu_D(x)=K_{x,X}(K_{X,X}+sigma_n^2 I)^(-1)y and k_D(x,x')=k(x,x')-K_{x,X}(K_{X,X}+sigma_n^2 I)^(-1)K_{X,x'}. Here K_{A,B} is the matrix of kernel evaluations between index sets A and B, I is the identity matrix, and D=(X,y). For a selected set S, use the noisy GP mutual information I(f_S;y_S)=1/2 log det(I+sigma_n^(-2)K_{D,S,S}), where K_{D,S,S} is the posterior covariance restricted to S. The conditional information gain of adding item i is Delta_MI(i|S)=1/2 log(1+v_i(S)/sigma_n^2), where v_i(S)=k_D(i,i)-k_D(i,S)(k_D(S,S)+sigma_n^2 I)^(-1)k_D(S,i) is the conditional posterior variance. Combine this with nonnegative teacher relevance r_i in [0,1] using F(S)=sum_{i in S} alpha r_i+beta/2 log det(I+sigma_n^(-2)K_{D,S,S}); alpha,beta>=0. The modular relevance term plus the log-determinant term is monotone submodular, so greedy selection has the standard (1-1/e) approximation for a fixed budget. For adaptive cardinality, stop when max_{i not in S}[alpha r_i+beta Delta_MI(i|S)]<=tau, where tau is selected on validation data to meet a target retention or quality level.
Implementation notes
Integrate this before an expensive attention or MIL block, operating on candidate token or patch embeddings X with shape N by d. Run a frozen teacher once to produce a scalar relevance r_i for every candidate and normalize r_i to [0,1]. Use an RBF kernel k(x,x')=exp(-||x-x'||^2/(2ell^2)) or a cosine kernel, with length scale ell estimated from median pairwise distances on a random subsample. Compute the GP posterior covariance k_D from a small inducing subset of teacher-labelled candidates, using Cholesky factorization rather than an explicit inverse. Greedy pseudocode: initialize S as empty; while |S|<B_max, compute each candidate's conditional variance v_i(S) from the Cholesky factor of K_D[S,S]+sigma_n^2 I, compute gain_i=alpha*r_i+0.5*beta*log(1+v_i(S)/sigma_n^2), choose i*=argmax gain_i, stop if gain_i*<=tau, otherwise append i* to S and update the Cholesky factor by a rank-one extension. Pass only X[S] to the student attention or MIL model and preserve original positions as positional metadata. Compute GP posterior and variance using the stated equations; estimate teacher scores, ell, sigma_n, alpha, beta, and tau empirically. First test ViT-Tiny on ImageNet-100 or a small histopathology MIL benchmark against top-k, uniform random, and k-center selection at equal retained-token counts. Measure accuracy, teacher-score retention, embedding coverage, wall-clock attention FLOPs, and peak activation memory. Success means a better accuracy-versus-token curve than top-k, at least 1.5x lower attention compute, and adaptive retention that decreases on easy examples without degrading hard-example accuracy.
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.