Uniform likelihood confidence head
Implementation & benchmark of arXiv:2608.02507 — Beyond Modern Asymptotics for Log-Likelihood Ratios in Logistic Regression
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
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.