An Input Skip Can Reparameterize a VAE Without Improving the Generator
Organizations: University of California at Los Angeles · NASA Jet Propulsion Laboratory
Abstract
An input skip can reduce reconstruction error while leaving a model's distribution of new samples unchanged. For a Gaussian variational autoencoder (VAE) with constant input skip , a model without a skip can be written in skip form by multiplying its decoder output by and its noise variance by . The distribution of observations and latent variables and its exact generative scores stay the same, while reconstruction mean squared error (MSE) is multiplied by . Separate training can produce different generators because optimization depends on the parameter coordinates. In a Gaussian experiment with known density, changing skip and observation variance together reduces MSE but worsens energy score and the estimate of negative log density from importance sampling. On seven annual Deep-ocean Assessment and Reporting of Tsunamis (DART) records, energy score differences at the same observation variance have both signs on four original records and favor the skip on three records from 2018. Probabilistic principal component analysis (PPCA) has lower mean energy scores than every neural DART fit. With the encoder and decoder kept the same, estimating observation variance from training reconstruction error can also worsen energy score. Assessing generation requires more than reconstruction error.
Figures & tables
| Skip coefficient | Base variance: | High variance: |
|---|---|---|
| No skip, base variance | No skip, high variance | |
| Skip, base variance | Skip, high variance |
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| Case | Train | Validation | Test | Mean | Scale | |
| Gaussian | 8192 | 2048 | 2048 | 0 | 1 | 0.06157 |
| 23461/2022 | 544 | 13 | 17 | 2488.8962 | 1.481 | 0.0001 |
| 44402/2016 | 1153 | 34 | 35 | 2613.8301 | 1.518 | 0.0001 |
| 46407/2011 | 983 | 26 | 27 | 3266.7902 | 1.48 | 0.0001 |
| 52403/2020 | 339 | 17 | 16 | 4433.8446 | 0.7036 | 0.0001 |
| 44402/2018 | 1150 | 390 | 369 | 2615.7276 | 0.33 | 0.0001 |
| Case | Skip | Variance | Raw MSE | MSE | Expected MSE | NLL/ | KL/ | Negative ELBO/ | Negative IW/ |
|---|---|---|---|---|---|---|---|---|---|
| Gaussian | Base | 0.0616 | 0.0616 | 0.06425 | 0.04692 | 0.09338 | 0.1403 | 0.1332 | |
| Gaussian | Base | 0.002496 | 0.06239 | 0.06498 | 0.05287 | 0.09343 | 0.1463 | 0.1373 | |
| Gaussian | High | 0.1985 | 0.1985 | 0.2182 | 1.205 | 0.01483 | 1.22 | 1.22 | |
| Gaussian | High | 0.003683 | 0.09209 | 0.1316 | 1.177 | 0.02497 | 1.202 | 1.201 | |
| 23461/2022 | Base | 0.0008939 | 0.0008939 | 0.0008945 | 0.7864 | 1.447 | 2.233 | 2.065 | |
| 23461/2022 | Base | 0.000102 | 0.002551 | 0.002552 | 9.073 | 1.372 | 10.44 | 10.06 |
| Case | Model | Negative log score | Energy score |
|---|---|---|---|
| Gaussian | True Gaussian | 0.1255 | 5.717 |
| Gaussian | PPCA | 0.1266 | 5.72 |
| Gaussian | Diagonal Gaussian | 0.9087 | 5.793 |
| Case | Skip | Variance | Seed | Step | Raw MSE | MSE/ | Negative IW | KL | Energy score | ESS |
|---|---|---|---|---|---|---|---|---|---|---|
| Gaussian | Base | 42 | 5000 | 0.06162 | 0.06162 | 0.1335 | 17.94 | 5.723 | 41.3 | |
| Gaussian | Base | 1729 | 5250 | 0.0617 | 0.0617 | 0.1334 | 18.06 | 5.724 | 35.68 | |
| Gaussian | Base | 31415 | 7750 | 0.06149 | 0.06149 | 0.1326 | 17.79 | 5.721 | 52.9 | |
| Gaussian | Base | 42 | 10000 | 0.002494 | 0.06236 | 0.1371 | 17.9 | 5.719 | 31.33 | |
| Gaussian | Base | 1729 | 6000 | 0.002497 | 0.06243 | 0.1376 | 17.96 | 5.722 | 29.33 | |
| Gaussian | Base | 31415 | 7500 | 0.002496 | 0.06239 | 0.1371 | 17.96 | 5.724 | 29.24 |
| Case | Skip vs no skip Base variance | Skip vs no skip High variance | Skip, high variance vs no skip, base variance |
|---|---|---|---|
| Gaussian | |||
| 23461/2022 | |||
| 44402/2016 | |||
| 46407/2011 | |||
| 52403/2020 |
| Case | Observation variance | ||||
|---|---|---|---|---|---|
| Skip | No skip | ||||
| Gaussian | Base | Base | |||
| Gaussian | High | High | |||
| Gaussian | High | Base | |||
| 23461/2022 | Base | Base | |||
| 23461/2022 | High | High | |||
| Record | Variance | Statistic | Full windows | After mean subtraction | Time increments |
|---|---|---|---|---|---|
| 23461/2022 | Base | Median and range | 0.369 [-1.21, 1.5] | 0.0255 [-0.0553, 0.297] | 0.00303 [-0.00289, 0.0262] |
| 23461/2022 | Base | Maximum range width | 0.1585 | 0.03316 | 0.004819 |
| 23461/2022 | High | Median and range | 0.83 [0.773, 3.4] | 0.0822 [-0.0275, 0.0868] | 0.0111 [0.000783, 0.0154] |
| 23461/2022 | High | Maximum range width | 0.4628 | 0.03795 | 0.005022 |
| 44402/2016 | Base | Median and range | -0.153 [-1.27, 0.849] | -0.0814 [-0.144, 0.032] | -0.00857 [-0.0311, 0.0141] |
| 44402/2016 | Base | Maximum range width | 0.2093 | 0.02672 | 0.004928 |
| Original variance | Fits | Changes in distance terms | Score after replacement |
|---|---|---|---|
| 42 | Lower | ||
| 42 | Higher |
| Record | Skip | Original variance | Full windows | After mean subtraction | Time increments | |
|---|---|---|---|---|---|---|
| 23461/2022 | Base | 3.32 [2.14, 4.09] | -5.13 [-6.91, -2.27] | -2.11 [-3.1, -0.803] | -0.395 [-0.559, -0.141] | |
| 23461/2022 | Base | 3.53 [2.63, 5.27] | -5.15 [-11, -3.29] | -2.25 [-4.58, -1.3] | -0.566 [-1.11, -0.32] | |
| 23461/2022 | High | 5.18 [3.94, 6.08] | 32.9 [31, 35.2] | 13.9 [13.7, 14.4] | 2.48 [2.22, 2.88] | |
| 23461/2022 | High | 5.74 [4.84, 8.83] | 30.4 [30.2, 35.7] | 13.7 [12.1, 18.8] | 3.7 [2.97, 4.72] | |
| 44402/2016 | Base | 3.45 [3.1, 5.24] | -5.3 [-9.19, -4.46] | -2.22 [-3.63, -1.67] | -0.105 [-0.246, -0.0806] | |
| 44402/2016 | Base | 4.1 [2.66, 6.45] | -5.23 [-10.7, -2.68] | -2.39 [-3.57, -1.3] | -0.281 [-0.395, -0.172] |
| Record | Variance | Evaluation | Steps | Full windows | After mean subtraction | Time increments |
|---|---|---|---|---|---|---|
| 44402/2018 | Base | Validation weights | 6000–10000 | -0.359 [-0.748, -0.128] | -0.385 [-0.519, -0.298] | -0.0368 [-0.055, -0.0353] |
| 44402/2018 | Base | Equal window weights | 10000 | -0.337 [-0.768, -0.119] | -0.346 [-0.479, -0.296] | -0.035 [-0.0522, -0.0334] |
| 44402/2018 | High | Validation weights | 10000–10000 | -0.262 [-0.33, -0.241] | -0.239 [-0.37, -0.201] | -0.0288 [-0.0431, -0.0213] |
| 44402/2018 | High | Equal window weights | 10000 | -0.222 [-0.322, -0.219] | -0.227 [-0.363, -0.151] | -0.0267 [-0.0419, -0.0157] |
| 21415/2018 | Base | Validation weights | 8000–10000 | -0.328 [-0.334, -0.255] | -0.406 [-0.409, -0.317] | -0.06 [-0.0612, -0.0448] |
| 21415/2018 | Base | Equal window weights | 10000 | -0.299 [-0.35, -0.211] | -0.371 [-0.377, -0.331] | -0.0555 [-0.0562, -0.0473] |
Variational Autoencoders
Variational autoencoders are probabilistic encoder–decoder models for learning latent representations, generating data, and estimating uncertainty.
See all 210 papers