Infomax-Reranker / METHOD.md
Angshul's picture
Upload 14 files
d3fd67f verified
|
Raw History Blame Contribute Delete
1.9 kB

Method

For one query with candidate passages $d_1,\dots,d_n$:

  • $r_j$: cross-encoder logit (InfoMax) or z-score fusion of cross-encoder logit and dense cosine (InfoMax+F).
  • $e_j$: all-MiniLM-L6-v2 embedding, centred on the pool mean and renormalised; $q$ the query embedding, treated the same way.
  • $w_j = \mathrm{minmax}(r)_j^{\gamma}$: relevance weight.

Target. A latent query-information vector $z \in \mathbb{R}^r$, with prior $z \sim \mathcal N(0,\Lambda)$, on the aspect basis $U = [,q,\ u_2,\dots,u_r,]$. Here $u_2,\dots,u_r$ are the top right-singular vectors of the relevance-weighted pool embeddings after removing the query direction, and $\Lambda = \mathrm{diag}(1, \rho, s_k^2/s_1^2)$.

Observations. Each passage is a noisy linear measurement of the target:

yj=wj (U⊀ej)⊀z+gj⊀u+Ξ΅j,gj=(Iβˆ’UU⊀)ej.y_j = \sqrt{w_j}\,(U^\top e_j)^\top z + g_j^\top u + \varepsilon_j,\qquad g_j = (I-UU^\top)e_j .

The shared nuisance $u \sim \mathcal N(0,\beta I)$ correlates the noise of passages with similar off-query content (duplicates, same source), and $\varepsilon_j \sim \mathcal N(0,\sigma^2)$.

Selection. Greedy maximisation of the conditional mutual information

I(z;yj∣yS)=12log⁑Var(yj∣yS)Var(yj∣yS,z),I(z; y_j \mid y_S) = \tfrac12 \log \frac{\mathrm{Var}(y_j \mid y_S)}{\mathrm{Var}(y_j \mid y_S, z)},

computed exactly by Schur complements. No probabilities from a language model are needed: in the linear-Gaussian model the posterior entropy depends only on which passages are selected.

Anchoring. The first pick is the top-1 of the base ranking $r$. At each later step, the base ranking's next item $b$ is replaced by the max-gain item $j^\star$ only if $\mathrm{gain}(j^\star) > (1+\delta),\mathrm{gain}(b)$. With $\delta=\infty$ the selector returns the base ranking exactly.

Guarantee. None is claimed. Correlated noise breaks submodularity of the objective, so the greedy rule has no worst-case approximation bound.