Conditional-copula probabilistic head
Implementation & benchmark of arXiv:2607.26256 — Conditional copula representations and extremal bounds for multivariate statistical functionals
Source paper: Conditional copula representations and extremal bounds for multivariate statistical functionals arXiv:2607.26256 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a generic multivariate Gaussian or independently factorized output head with separate marginal quantile models and a conditional copula module. The marginals determine each output's calibrated one-dimensional distribution, while the copula models dependence on the uniformized variables, allowing the network to represent asymmetric correlations and tail co-movement without forcing a particular marginal family.
Formulas
Mathematical statement
For an output vector \(\mathbf{X}=(X_1,\ldots,X_n)\) with marginal CDFs \(F_i\), define the probability-integral transforms \(U_i=F_i(X_i)\); under continuous marginals, \(U_i\sim\operatorname{Uniform}(0,1)\), and \(\mathbf{U}\) has copula \(C\). The paper defines the conditional copula CDF of the final coordinate by \(C_{n\mid 1,\ldots,n-1}(u_n\mid\mathbf{u})=\frac{\partial^{n-1}}{\partial u_1\cdots\partial u_{n-1}}C(u_1,\ldots,u_n)\), where \(\mathbf{u}=(u_1,\ldots,u_{n-1})\). This derivative is the conditional CDF of \(U_n\) given \((U_1,\ldots,U_{n-1})=\mathbf{u}\); differentiating it once more in \(u_n\) gives the conditional density used in likelihood training. The adapted neural model uses conditional CDFs \(H_i(u_i\mid u_{<i},z)\), where \(z\) is the network context, and the joint density factorization \(p(\mathbf{x}\mid z)=\prod_{i=1}^n f_i(x_i\mid z)h_i(u_i\mid u_{<i},z)\), with \(u_i=F_i(x_i\mid z)\), marginal densities \(f_i\), and copula conditional densities \(h_i=\partial H_i/\partial u_i\). The key property is separation: changing \(F_i\) does not intrinsically change the dependence model on the uniform scale.
Implementation notes
Integrate this as the probabilistic output head of a small conditional VAE, trajectory predictor, or model-based RL dynamics network. The backbone produces context z. First create n marginal heads F_i(·|z), using discretized logistic distributions, Gaussian-mixture CDFs, or monotone rational-quadratic splines; each head must return log density f_i(x_i|z), CDF F_i(x_i|z), and inverse CDF F_i^{-1}(u_i|z). Uniformize each observed target with u_i=clip(F_i(x_i|z), epsilon, 1-epsilon). Then create an autoregressive copula head. For i=1,...,n, feed [z,u_1,...,u_{i-1}] to a monotone spline whose input and output domain is [0,1], yielding H_i(u_i|u_<i,z) and h_i=dH_i/du_i. Optimize the negative log likelihood -sum_i[log f_i(x_i|z)+log h_i(u_i|u_<i,z)]. Enforce monotonicity by positive spline widths, heights, and derivatives, and enforce H_i(0)=0 and H_i(1)=1. For generation, sample independent r_i~Uniform(0,1), recursively solve H_i(u_i|u_<i,z)=r_i by bisection or spline inversion, then output x_i=F_i^{-1}(u_i|z). The paper's exact mathematical contribution is the uniformization and conditional-copula separation; spline parameterization, context conditioning, and autoregressive ordering are engineering choices. Estimate calibration with marginal PIT histograms and dependence with rank correlation, tail-dependence estimates, and multivariate energy score. Start on a 2D or 5D synthetic mixture with non-Gaussian marginals and known asymmetric copula, comparing independent heads, a multivariate Gaussian head, and the copula head at equal parameter count. Success means lower joint NLL and energy score at unchanged marginal CRPS, plus better coverage of joint prediction regions and visibly improved probability of simultaneous extreme events.
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 9/10
Built and numerically verified a two-dimensional copula MVP with exact non-Gaussian lognormal/Gamma marginals and Gaussian-copula dependence. The conditional-density derivative matched finite differences with relative error 6.5e-11; PITs were near Uniform(0,1), rank measures matched analytic predictions within 0.003, and observed NLL gains tracked the predicted -0.5 log(1-rho^2) scaling. At rho=0.8, the copula improved NLL from 2.9782 to 2.4657, so the dependence-separation mechanism manifested clearly.
- Agent confidence
- 9/10
- Baseline
- Independent marginal NLL 2.9782 at rho=0.8; no dependence gain at rho=0.
- Idea
- Copula NLL 2.4657 at rho=0.8, observed gain 0.5125 nats versus predicted 0.5108; fitted rho 0.8009. PIT means were approximately 0.5, variances approximately 0.0833, and KS statistics were 0.0033-0.0061.
Limitations: This is a controlled 2D MVP rather than a learned neural conditional spline head: marginals are known analytically, dependence is a constant Gaussian copula rather than context-conditioned or asymmetric/tail-flexible, and no VAE, trajectory model, calibration-region, CRPS, energy-score, or equal-parameter-count neural comparison was tested.
How to run: python3 copula_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.26256)
No significant difference vs. the tuned baseline (+4.0%, p=0.0552, wins 2 / 8 paired seeds).
- Benchmark
- Correlated_multitask_regression
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 0.2637
- Idea mean
- 0.2742
- Effect (Δ)
- 0.0105 (+4.0%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.0552 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±3.8%
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Only the correlated multitask regression custom track was tested, with 1,200 training samples, 500 test samples, 18 epochs, and a six-dimensional conditional Gaussian copula rather than monotone neural splines or non-Gaussian marginals. Energy score, CRPS, calibration histograms, tail-dependence coverage, parameter-count matching, and the built-in tabular/vision/sequence/dynamics tracks were not tested.
How to run: python3 bench_copula_stage2.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
Implemented an end-to-end conditional Gaussian-copula probabilistic head with the same MLP trunk as a diagonal-Gaussian baseline on the structurally matched six-output correlated multitask regression track. The mechanism signature was confirmed: predicted mean off-diagonal residual correlation 0.1143 versus observed 0.1704 (absolute error 0.0561). However, test MSE worsened from 0.26368 to 0.27419, paired delta +0.01051, with permutation p=0.0552; therefore there is no significant win and the system did not work on this bench.
- Baseline
- Best lr=0.006; full 8-seed test MSE mean 0.2636817899, std 0.0127073678; sweep means: lr=0.001 0.3392473, lr=0.003 0.2891158, lr=0.006 0.2687482.
- Idea
- Best lr=0.006; full 8-seed test MSE mean 0.2741883807, std 0.0090002103; idea sweep means: lr=0.001 0.3560834, lr=0.003 0.3174563, lr=0.006 0.2741884; paired delta idea-baseline +0.0105066, p=0.0552, 2/8 idea wins.
- Mechanism signature
- Confirmed: predicted copula off-diagonal correlation mean 0.1142624 versus observed trained-model residual correlation mean 0.1704001; absolute error 0.0561378.
Benchmark result
No significant difference vs. the tuned baseline (+4.0%, p=0.0552, wins 2 / 8 paired seeds).
- Benchmark
- Correlated_multitask_regression
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 0.2637
- Idea mean
- 0.2742
- Effect (Δ)
- 0.0105 (+4.0%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.0552 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±3.8%
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Only the correlated multitask regression custom track was tested, with 1,200 training samples, 500 test samples, 18 epochs, and a six-dimensional conditional Gaussian copula rather than monotone neural splines or non-Gaussian marginals. Energy score, CRPS, calibration histograms, tail-dependence coverage, parameter-count matching, and the built-in tabular/vision/sequence/dynamics tracks were not tested.
How to run: python3 bench_copula_stage2.py
Artifacts
- bench_copula_stage2.py 6.5 KB View Raw
- bench_report.json 5.7 KB View Raw
- copula_experiment.py 5.4 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-08-31T174639.md 4.3 KB View
- results.json 4.4 KB View Raw