Unverified 2026

Uniform likelihood confidence head

Implementation & benchmark of arXiv:2608.02507 — Beyond Modern Asymptotics for Log-Likelihood Ratios in Logistic Regression

Usefulness5/10
Difficulty5/10
Novelty6/10

Source paper: Beyond Modern Asymptotics for Log-Likelihood Ratios in Logistic Regression arXiv:2608.02507 ⓘ · analyzed Aug 31, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Freeze a neural backbone and replace heuristic last-layer uncertainty with a confidence region derived from the paper's uniform logistic likelihood-ratio bound. For a binary head, accept a prediction only when every head parameter in the confidence region gives the same label; otherwise abstain or request an additional label. The threshold also gives a principled stopping rule for fine-tuning the head.

Formulas

$$\mathbf{P}_{\theta^\star}(Y_i=1)=\sigma(\langle x_i,\theta^\star\rangle),\qquad \sigma(t)=\frac{1}{1+e^{-t}},\qquad Y_i\in\{-1,1\}.$$
$$\Lambda_n^{\log}(\theta^\star)=\log\frac{\sup_{\theta\in\mathbf{R}^{d}}\prod_{i=1}^{n}\sigma\bigl(Y_i\langle x_i,\theta\rangle\bigr)}{\prod_{i=1}^{n}\sigma\bigl(Y_i\langle x_i,\theta^\star\rangle\bigr)}.$$
$$q_{1-\delta}(\Lambda_n^{\log})\ \lesssim\ d\log\left(\frac{\mathrm{e}n}{d}\right)+\log\left(\frac{1}{\delta}\right),\qquad n\ge d\ge3.$$
$$\mathcal{C}_\delta=\left\{\theta:\ell(\hat\theta)-\ell(\theta)\le \tau_\delta\right\},\quad \ell(\theta)=\sum_{i=1}^{n}\log\sigma(Y_i\langle x_i,\theta\rangle),\quad \tau_\delta=C\left[d\log\left(\frac{\mathrm{e}n}{d}\right)+\log\left(\frac{1}{\delta}\right)\right].$$

Mathematical statement

The model has deterministic feature vectors x_i in R^d, labels Y_i in {-1,1}, and target head parameter theta-star in R^d, with P_theta-star(Y_i=1)=sigma(<x_i,theta-star>) and sigma(t)=1/(1+exp(-t)). The paper defines Lambda_n^log(theta-star)=log(sup_theta product_i sigma(Y_i<x_i,theta>) divided by product_i sigma(Y_i<x_i,theta-star>)). Its main result states that for arbitrary fixed design and n >= d >= 3, the worst-case (1-delta) quantile is, up to universal constants, d log(e n/d)+log(1/delta); for iid Gaussian design in the regime n approximately at least d+log(1/delta), the scale improves to d+log(1/delta). Define ell(theta)=sum_i log sigma(Y_i<x_i,theta>) and theta-hat as a maximizer. The adapted confidence set is C_delta={theta: ell(theta-hat)-ell(theta) <= tau_delta}, with tau_delta=C[d log(e n/d)+log(1/delta)], where C is the universal-constant factor from the theorem. Under the correctly specified frozen-head model, theta-star belongs to this set with probability at least approximately 1-delta, uniformly over fixed features and target parameters. Because the region is defined by likelihood loss rather than an inverse Hessian, it remains usable near separation or ill-conditioning.

Implementation notes

(1) Integration point: take a pretrained binary classifier, freeze its backbone, and expose the penultimate representation h_i in R^(d-1). Append a bias so x_i=[h_i;1] in R^d, and train only a logistic head theta. Run the confidence computation after each head-training epoch and at inference; the backbone forward pass is unchanged.

(2) Pseudocode:

inputs: frozen features X={x_i}, labels y_i in {-1,+1}, delta
fit theta_hat = argmax_theta sum_i log sigmoid(y_i * x_i^T theta)
tau = C * (d * log(e*n/d) + log(1/delta))
for test feature x:
    theta_min = argmin_{theta: ell(theta_hat)-ell(theta) <= tau} x^T theta
    theta_max = argmax_{theta: ell(theta_hat)-ell(theta) <= tau} x^T theta
    if theta_min >= 0: predict +1
    else if theta_max <= 0: predict -1
    else: abstain

The constrained objectives are convex because negative logistic log-likelihood is convex. Solve each with projected gradient or a primal-dual solver. For a cheaper approximation, sample directions u_j on the unit sphere, find the largest r_j satisfying ell(theta-hat)-ell(theta-hat+r_j u_j)=tau, and compute the minimum and maximum sampled test logit.

(3) The paper supplies the d log(e n/d)+log(1/delta) threshold shape and design-uniform guarantee. Estimate the unknown constant C conservatively through Monte Carlo on synthetic logistic data over dimensions, sample sizes, target norms, and nearly separable designs. If whitened features behave approximately as iid Gaussian, also test tau=C[d+log(1/delta)]. The theorem assumes the representation is fixed independently of the labels used for calibration; reserve a fresh calibration split if the backbone was trained on the same task.

(4) First cheap experiment: use CIFAR-10 restricted to two classes, extract frozen ResNet-18 features, and compare this confidence set with inverse-Hessian ellipsoids and temperature scaling at n in {100,300,1000}. Generate repeated synthetic labels from known logistic heads to measure empirical coverage, then evaluate selective accuracy and abstention rate on real labels. Stress-test duplicate, low-rank, and nearly separable features. Success means coverage at least 1-delta under ill-conditioning and higher selective accuracy than Hessian confidence at matched coverage; a secondary signal is stable confidence when n is only modestly larger than d.

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.