Unverified 2026

GP-DPP Adaptive Token Selection

Implementation & benchmark of arXiv:2608.23574 — InfoDPP-PAC: Principled Patch Selection for Whole Slide Image Analysis

Usefulness6/10
Difficulty5/10
Novelty5/10

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

$$\mu_{\mathcal{D}}({\bm{x}})={\bm{K}}_{{\bm{x}},\mathcal{X}}({\bm{K}}_{\mathcal{X},\mathcal{X}}+\sigma_{n}^{2}{\bm{I}})^{-1}{\bm{y}}$$
$$k_{\mathcal{D}}({\bm{x}},{\bm{x}}^{\prime})=k({\bm{x}},{\bm{x}}^{\prime})-{m{K}}_{{\bm{x}},\mathcal{X}}({\bm{K}}_{\mathcal{X},\mathcal{X}}+\sigma_{n}^{2}{\bm{I}})^{-1}{\bm{K}}_{\mathcal{X},{\bm{x}}^{\prime}}$$
$$F(S)=\sum_{i\in S}\alpha r_i+\frac{\beta}{2}\log\det\left(I+\sigma_n^{-2}K_{\mathcal D,S,S}\right),\qquad \Delta_{\mathrm{MI}}(i\mid S)=\frac{1}{2}\log\left(1+\frac{v_i(S)}{\sigma_n^2}\right)$$
$$v_i(S)=k_{\mathcal D}(i,i)-k_{\mathcal D}(i,S)\left(k_{\mathcal D}(S,S)+\sigma_n^2I\right)^{-1}k_{\mathcal D}(S,i),\qquad \text{stop if }\max_{i\notin S}\left[\alpha r_i+\beta\Delta_{\mathrm{MI}}(i\mid S)\right]\leq\tau$$

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.