r-Deformed Power Divergence Loss
Implementation & benchmark of arXiv:2607.01805 — $r$-deformed $α$-$z$-Rényi relative entropy
Source paper: $r$-deformed $α$-$z$-Rényi relative entropy arXiv:2607.01805 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace cross-entropy or ordinary Renyi loss between a target distribution and a model distribution with the paper's r-deformed alpha-z divergence. The deformation parameter r provides a controllable power-law alternative to the logarithm, allowing experiments that emphasize hard, low-probability target events differently from standard log losses.
Formulas
Mathematical statement
The paper uses the r-logarithm \(\ln_r(x)=(x^{1-r}-1)/(1-r)\) for \(x>0\), with continuous limit \(\ln_1(x)=\log x\). For density operators \(\rho\) and \(\sigma\), its central trace functional is \(S_{\alpha,z}(\rho,\sigma)=\operatorname{tr}\left(\sigma^{(1-\alpha)/(2z)}\rho^{\alpha/z}\sigma^{(1-\alpha)/(2z)}\right)^z\), where \(\alpha\) and \(z\) are real parameters and matrix powers are spectral powers of positive semidefinite matrices. The proposed divergence is \(D_{\alpha,z,r}(\rho\|\sigma)=\ln_r(S_{\alpha,z}(\rho,\sigma))/(\alpha-1)\). For commuting diagonal probability vectors \(y\) and \(p\), use the computable specialization \(D_{\alpha,r}(y\|p)=\ln_r(\sum_i y_i^\alpha p_i^{1-\alpha})/(\alpha-1)\), where \(y_i,p_i\geq 0\) and both distributions sum to one. The paper proves data-processing and divergence properties only in specified parameter ranges; the classifier experiment should therefore record the chosen \((\alpha,z,r)\) and use a parameter region covered by the theorem when reproducing its guarantee.
Implementation notes
Integrate this at the final classifier loss. Let logits \(a\in\mathbb{R}^K\) produce \(p_i=\operatorname{softmax}(a)_i\), and let \(y\) be either a one-hot label or a label-smoothed target distribution. Use the classical commuting specialization rather than implementing quantum matrix powers. Pseudocode: compute \(p=\operatorname{softmax}(a)\); clamp \(p_i\) and \(y_i\) to \([\epsilon,1]\), with \(\epsilon=10^{-8}\); compute \(T=\sum_i y_i^\alpha p_i^{1-\alpha}\); compute \(L=(T^{1-r}-1)/((1-r)(\alpha-1))\); average over the minibatch and backpropagate. Use a \(r=1\) branch that returns \(\log(T)/(\alpha-1)\), and calculate powers through log probabilities to avoid underflow. Begin with \(\alpha\in\{0.5,1.5,2\}\) and \(r\in\{0,0.5,1,1.5\}\). The mathematical computation is the power sum and r-log transform; admissible parameter selection and optimization hyperparameters are empirical. Compare against cross-entropy, ordinary Renyi loss \(\log(T)/(\alpha-1)\), and Tsallis-style loss under identical models and schedules. First run CIFAR-10 with ResNet-18, batch size 128, and SGD or AdamW. Measure validation accuracy, expected calibration error, gradient-norm variance, and robustness to class imbalance and label smoothing. Success is higher accuracy or calibration at equal training FLOPs, or lower gradient variance with no final accuracy loss. Reject configurations producing exploding gradients or singular behavior as \(T\) approaches zero or infinity.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Inconclusive
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 7/10
Built a stable commuting r-deformed power-divergence loss with log-domain power sums and an r=1 branch, plus a deterministic toy classifier experiment. The math check confirmed the r=1 result exactly matches ordinary Rényi loss, showed finite gradients, and demonstrated smoothly different power-law penalties as r varies. On the imbalanced Gaussian task, r=0 matched the best accuracy (0.865 vs 0.860 CE) and improved ECE (0.0626 vs 0.0769), while also reducing gradient variance; this is a promising but small single-seed signal rather than definitive evidence.
- Agent confidence
- 7/10
- Baseline
- Cross-entropy: accuracy 0.8600, ECE 0.07694, gradient variance 0.00606, validation CE 0.6132
- Idea
- r=0, alpha=0.5: accuracy 0.8650, ECE 0.06264, gradient variance 0.00114, validation CE 0.6056; r=0.5: accuracy 0.8600, ECE 0.08247, gradient variance 0.00103
Limitations: Only a small synthetic 3-class problem, one random seed, one alpha value, one label-smoothing setting, optimizer, and schedule were tested. CIFAR-10/ResNet-18, multiple seeds, broader parameter ranges, wall-clock/FLOP comparisons, and theorem-specific data-processing guarantees were not tested.
How to run: python3 run_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.01805)
No significant difference vs. the tuned baseline (+2.7%, p=0.1716, wins 2 / 8 paired seeds).
- Benchmark
- Vision (image classification)
- Model
- cnn_small
- Paired seeds
- 8
- Baseline mean
- 0.7653
- Idea mean
- 0.7856
- Effect (Δ)
- 0.0203 (+2.7%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.1716 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Only the 400-example CIFAR-10 subset, 8 epochs, alpha=0.5, and r=0 were tested. Longer training, broader alpha/r sweeps, label smoothing, calibration, and full CIFAR-10/ResNet-18 were not tested.
How to run: python3 bench_run.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
The r-deformed alpha-divergence loss was implemented and tested end-to-end against cross-entropy with matched cnn_small models, paired seeds, and a shared learning-rate grid. The idea had higher mean test error (0.785625 versus 0.7653125), paired delta +0.0203125, and permutation p=0.17155, so it did not produce a significant win. Its trained-model gradient-norm variance was lower (0.4867 versus 3.6030), confirming the auxiliary mechanism signal without improving the primary metric.
- Baseline
- mean test error 0.7653125; best lr=0.0015; 8 paired seeds
- Idea
- mean test error 0.7856250; alpha=0.5, r=0, best lr=0.003; paired delta +0.0203125; p=0.17155
Benchmark result
No significant difference vs. the tuned baseline (+2.7%, p=0.1716, wins 2 / 8 paired seeds).
- Benchmark
- Vision (image classification)
- Model
- cnn_small
- Paired seeds
- 8
- Baseline mean
- 0.7653
- Idea mean
- 0.7856
- Effect (Δ)
- 0.0203 (+2.7%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.1716 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Only the 400-example CIFAR-10 subset, 8 epochs, alpha=0.5, and r=0 were tested. Longer training, broader alpha/r sweeps, label smoothing, calibration, and full CIFAR-10/ResNet-18 were not tested.
How to run: python3 bench_run.py
Artifacts
- bench_report.json 2.6 KB View Raw
- bench_run.py 6.0 KB View Raw
- report.md 1.4 KB View
- report_bench_2026-09-04T001403.md 3.3 KB View
- results.json 1.7 KB View Raw
- run_experiment.py 5.0 KB View Raw