---
title: Variational Probabilistic Quantization (VPQ)
url: https://www.emergentmind.com/topics/variational-probabilistic-quantization-vpq
type: topic
---

# Variational Probabilistic Quantization (VPQ)

Searching arXiv for papers using or closely related to “Variational Probabilistic Quantization” and the specific arXiv IDs provided.
Search terms used: “Variational Probabilistic Quantization”, “Probabilistic Weight Fixing quantization”, “Variational Bayesian Quantization”, “SQ-VAE stochastic quantization”, and the exact arXiv IDs 2309.13575, 2002.08158, 2205.07547, 1306.6922, 2510.02048, 2512.10602, 2511.06863.
Variational Probabilistic Quantization (VPQ) is not a single universally standardized algorithm in the arXiv literature reviewed here. Rather, the label has been applied to several variational or probabilistic constructions in which quantization, discretization, or field quantization is mediated by uncertainty models, latent distributions, stochastic relaxations, or variational objectives. In neural compression and model quantization, VPQ has denoted uncertainty-aware clustering of weights, stochastic or variational discretization of latent variables, and quantization of variational parameters in Bayesian neural networks; in communications it has denoted probabilistic neural encoders for secret common-randomness extraction; and in stochastic field theory it has been used for the Stochastic Variational Method as an alternative quantization scheme for the complex Klein–Gordon field [2309.13575], [2205.07547], [2002.08158], [2512.10602], [2510.02048], [1306.6922], [2511.06863].

## 1. Scope of the term and recurring variational structure

Across these usages, the recurring motif is that quantization is not treated as a purely deterministic nearest-neighbor or thresholding operation. Instead, the discrete or low-cardinality representation is informed by posterior uncertainty, stochastic sampling, variational bounds, or stochastic action principles. In some cases the target of quantization is a neural-network weight tensor; in others it is a latent code, a variational parameter, a common-randomness symbol, or a field configuration [2309.13575], [2205.07547], [2512.10602], [2510.02048], [1306.6922].

| Usage | Quantized object | Variational/probabilistic mechanism |
|---|---|---|
| PWFN / neural weight sharing | Network weights \(w_i\) | Mean-field Gaussian posterior \(q(w;\theta)=\prod_i \mathcal N(w_i\mid \mu_i,\sigma_i^2)\) |
| SQ-VAE | Discrete latent code indices \(s\) | Softmax posterior with temperature \(\tau\) and self-annealed stochastic quantization |
| VBQ | Continuous latent coordinates \(z_i\) | Quantile-space discrete search weighted by posterior uncertainty |
| QBNN VPQ | Variational parameters \(\mu,\log\sigma^2\) | Quantizers inserted inside the SVI ELBO with STE |
| Secret-CR VPQ | Encoder outputs \(W,V\in\{1,\dots,|\mathcal W|\}\) | Variational bounds on \(I(W;Z)\) and adversarial training |
| SVM as quantization scheme | Stochastic field modes \(C_{i,\mathbf k}\) | Stochastic action principle leading to the functional Schrödinger equation |
| VAEVQ | Continuous visual latents and codebook | VAE ELBO plus RCS and DCR around nearest-neighbor quantization |

A common misconception is to read VPQ as if it named one settled technique. The surveyed literature indicates the opposite: the same acronym has been used for distinct research programs. Another common source of confusion is the meaning of “quantization” itself. In [1306.6922], quantization means field quantization in the sense of quantum theory; in [2309.13575], [2205.07547], [2002.08158], [2512.10602], and [2511.06863], it means compression-oriented discretization or low-precision representation; and in [2510.02048] it means mapping correlated observations to discrete random variables with agreement, uniformity, and secrecy constraints.

## 2. Weight-space VPQ: probabilistic weight fixing and uncertainty-aware clustering

In [2309.13575], the method is introduced as “Probabilistic Weight Fixing” (PWFN) and is based on a fully factorized Gaussian variational posterior over network weights,
$$
q(w;\theta)=\prod_{i=1}^N \mathcal N(w_i\mid \mu_i,\sigma_i^2),
$$
with likelihood \(p(D\mid w)\). The usual ELBO is written as
$$
\mathrm{ELBO}(\theta)=\mathbb E_{w\sim q(w;\theta)}[\log p(D\mid w)]-KL[q(w;\theta)\|p(w)],
$$
but in practice the prior KL is absorbed into a regularizer. To prevent \(\sigma_i\to 0\), the training loss adds a “noise-resilience” term
$$
L_{\rm REG}(\sigma)=\sum_{i=1}^N \ell(\sigma_i),\qquad
\ell(\sigma)=
\begin{cases}
S-\sigma,& \sigma<S\\
0,& \text{otherwise},
\end{cases}
$$
so that
$$
L(\theta)=-\mathbb E_{w\sim q(w;\theta)}[\log p(D\mid w)] + \alpha\,L_{\rm REG}(\sigma).
$$
In experiments, \(\alpha=2^{-11}\) and \(S=0.05\).

The central quantization mechanism is a separate “fixing” stage applied after or between variational updates. A codebook \(c=\{c_1,\dots,c_k\}\), initially chosen as powers-of-two and additive combinations, is used to fix each \(\mu_i\) to a cluster center according to the distance-in-\(\sigma\) metric
$$
D_{\rm prob}(i,j)=\frac{|\mu_i-c_j|}{\sigma_i}.
$$
The paper presents this as a variational relaxation of discrete assignments: rather than introducing hard latent indicators \(z_i\in\{1,\dots,k\}\) during training, all weights remain Gaussian and clustering is deferred to the fixing step. The description explicitly notes that one may view this as the \(\tau\to 0\) limit of a Gumbel-Softmax relaxation,
$$
p(z_{ij}=1\mid \mu_i,\sigma_i)\propto \exp\!\left(-\frac{|\mu_i-c_j|}{\sigma_i\cdot \tau}\right),
$$
while the implemented method uses the direct \( |\mu_i-c_j|/\sigma_i \) criterion.

The initialization is also uncertainty-aware. The means \(\mu_i\) are initialized to the pre-trained point weights. The \(\sigma_i\) are initialized by a small parabolic function of the distance from the nearest power-of-two:
$$
\sigma_i^{(0)}=(0.05)^2
\cdot \frac{|\mu_i-2^{x_i}|}{2^{x_i}}
\cdot \frac{|2^{x_i+1}-\mu_i|}{2^{x_i+1}},
$$
for \(2^{x_i}\le \mu_i\le 2^{x_i+1}\), followed by clamping to \([2^{-30},0.05]\).

The fixing procedure alternates \(T=9\) rounds; each round fixes a fraction \(p_t\) of the weights and retrains the remaining free weights for \(3\) epochs, for a total of \(27\) epochs. The optimizer is SGD with momentum \(0.9\), learning rate \(0.001\), batch-size \(128\), and \(4\times\)RTX8000 GPUs. The threshold is \(\delta=1\) in units of \(\sigma\), and the schedule \(p_t\) follows Subia-Waud & Dasmahapatra 2022. No architecture-specific tuning is required beyond choosing the same \(\alpha,\delta\), and the method is reported to work on ResNets-18/34/50, DenseNet-161, and DeiT-Small/Tiny.

Empirically, the method “demonstrates superior compressibility and higher accuracy compared to state-of-the-art methods” on both ResNet models and transformer-based architectures [2309.13575]. On ImageNet, the reported Top-1 results are \(70.0\) for ResNet-18 with \(155\) unique values, \(74.3\) for ResNet-34 with \(154\) unique values, \(77.5\) for ResNet-50 with \(325\) unique values, \(78.1\) for DeiT-Small with \(356\) unique values, \(71.2\) for DeiT-Tiny with \(296\) unique values, and \(77.6\) for DenseNet-161

Source: https://www.emergentmind.com/topics/variational-probabilistic-quantization-vpq