Renyi Drift-Controlled Fine-Tuning
Implementation & benchmark of arXiv:2608.19320 — Quantum Rényi-Jarzynski Equality
Source paper: Quantum Rényi-Jarzynski Equality arXiv:2608.19320 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Add a Renyi divergence penalty between the current network output distribution and a frozen reference distribution representing the pretrained model, a teacher, or a retained-data equilibrium. The Renyi order k becomes a control parameter: k greater than 1 strongly penalizes examples on which the new model assigns disproportionately more probability than the reference, while orders below 1 emphasize support mismatch and low-probability regions. Sweep or anneal k and detect a transition between low-drift and high-task-fit minima rather than treating the regularization coefficient as the only control parameter.
Formulas
Mathematical statement
For each minibatch, let p_theta(y|x) be the trainable model distribution and q(y|x) the frozen reference distribution. The Renyi divergence of order k, for k greater than 0 and k not equal to 1, is D_k(p||q) = [1/(k-1)] log sum_y p(y)^k q(y)^(1-k), with the limit D_KL as k approaches 1. Use J(theta;k) = L_task(theta) + lambda D_k(p_theta||q). The weighting factor p(y)^k q(y)^(1-k) changes exponentially with the log-ratio log[p(y)/q(y)], so changing k can move the global minimizer between competing basins. A transition point k_c occurs when two minimized branches have equal objective value, J(theta_A;k_c) = J(theta_B;k_c), accompanied by a jump in the selected solution or a sharp change in curvature.
Implementation notes
First, integrate the method into supervised fine-tuning, continual learning, or policy distillation. Freeze a reference network q, usually the pre-fine-tuning checkpoint or an exponential-moving-average teacher. For each minibatch, compute student logits, reference logits under no-gradient mode, and the task loss. Convert both logits to probabilities with a common temperature. Use log-space arithmetic: logp = log_softmax(student_logits divided by T), logq = log_softmax(reference_logits divided by T), logS = logsumexp(k times logp plus (1-k) times logq over classes), and renyi = mean(logS) divided by (k-1). For k close to 1, use KL divergence to avoid numerical cancellation. The update is task_loss plus lambda times renyi, followed by backpropagation through the student only. The paper supplies the divergence construction and the prediction that competing minima can exchange dominance as k changes; estimate basin structure, branch objective values, Hessian trace, gradient norms, and output drift empirically. The first cheap experiment should fine-tune a pretrained ResNet-18 on CIFAR-10 with 20 percent label corruption while retaining the original model behavior on a clean replay subset. Compare KL, k values 0.5, 2, 4, and 8, and no regularizer at matched lambda values over five seeds. Warm-start sweeps in both increasing and decreasing k directions. The quantitative prediction is a reproducible kink or hysteresis-like jump in clean-set KL and task accuracy near a critical k_c, especially when two lambda-dependent solutions coexist. Locate k_c from equal minimized objective values and require agreement within 20 percent across seeds. Increasing k should also disproportionately reduce the upper quantiles of log[p_theta divided by q], whereas k below 1 should reduce support-loss events where the student probability is near zero.
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.