REVIEW 3 major objections 5 minor 13 references
Joint-stochastic-approximation Autoencoders with Application to Semi-supervised Learning
T0 review · 3 major / 5 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read A new autoencoder family trains deep generative models by directly maximizing data log-likelihood, and its discrete-latent version matches state-of-the-art semi-supervised classifiers on MNIST and SVHN.
desk verdict A useful but under-supported SSL extension of the JSA framework; the empirical claims rest on 10-step MCMC draws that are likely not mixing, and Algorithm 1 doesn't match the stated objective. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The engine is joint stochastic approximation (JSA). The learning problem is written as finding a root of a system of equations: the generative gradient $\mathbb{E}_{p_\theta(h|x)}[\frac{\partial}{\partial\theta}\log p_\theta(x,h)] = 0$ and the inference gradient $\mathbb{E}_{p_\theta(h|x)}[\frac{\partial}{\partial\phi}\log q_\phi(h|x)] = 0$. These are expectations under the posterior, so the Robbins-Monro stochastic approximation algorithm can solve them by sampling from a Markov kernel whose invariant distribution is $p(z;\lambda)$. The paper uses a Metropolis independence sampler with proposal $q_\phi(h|x)$ from the inference network; computing the importance ratio cancels the intractable marginal $p_\theta(x)$, and the accept/reject step makes the inference network a valid proposal even when it mismatches the true posterior.
What would settle it
Train the MNIST JAE while logging the fraction of proposed samples accepted by the Metropolis independence sampler and the per-unit flip rate in the 60-dimensional Bernoulli latent space. Low acceptance rates or no flips within the ten warm-up steps would show that the gradient estimates are not drawn from the posterior, contradicting the maximum-likelihood interpretation; high acceptance with fast mixing would support the claim.
Extended reading notes
Core claim
The central claim is that maximum-likelihood learning for a deep directed generative model can be cast as a stochastic approximation root-finding problem over paired generative and inference parameters, and that solving it yields an autoencoder which directly maximizes the data log-likelihood while simultaneously minimizing $\mathrm{KL}(p_\theta(h|x) \| q_\phi(h|x))$. The authors prove consistency in the nonparametric limit: when both models have infinite capacity and the data size grows, the generative model recovers the true data distribution and the inference model recovers the exact posterior. Empirically, they show that the Metropolis independence sampler compensates for encoder/decoder mismatch, that the same update rules work for discrete and continuous variables, and that a JAE with a 60-dimensional Bernoulli prior on MNIST and a 220-dimensional Bernoulli prior on SVHN matches the semi-supervised classification performance of continuous-latent DGMs. They interpret this as first evidence that discrete latent variable models can succeed in semi-supervised learning.
Load-bearing premise
The training procedure assumes that ten warm-up sampling steps per data point bring the sampler close enough to the posterior that the minibatch gradients approximate the true maximum-likelihood direction; if the chain is still far from equilibrium, the implemented algorithm is not actually maximizing the data log-likelihood.
Editorial extensions
If this is right
- JAE training optimizes the data log-likelihood directly, so the fitted generative model inherits the statistical consistency properties of maximum likelihood instead of optimizing a surrogate bound.
- Discrete latent variables become usable without gradient tricks such as REINFORCE, Gumbel-softmax, or straight-through estimators, making Bernoulli or categorical codes practical in semi-supervised learning.
- Encoder/decoder structure mismatch does not bias the learned generative model, because the Metropolis independence sampler corrects the proposal distribution rather than being forced to match the posterior exactly.
- Semi-supervised JAEs can disentangle class and style in the latent space and perform class-conditional generation with discrete codes, as shown by the MNIST traversal experiments.
- The same learning criterion extends naturally to semi-supervised objectives by adding supervised likelihood terms and a discriminative classification term $\alpha \log q_\phi(y|x)$.
Reading between the lines
- One testable extension is applying JAEs to text or categorical observations, where VAEs are known to suffer posterior collapse; the paper's context-free-grammar experiment hints that the MIS mechanism may avoid that failure mode.
- A direct diagnostic that would strengthen or weaken the central claim is measuring the acceptance rate and mixing speed of the Metropolis independence sampler during training, which the paper does not report.
- The nonparametric consistency argument suggests JAEs with continuous latent spaces should also improve over VAEs on likelihood-based evaluation; the binary MNIST log-likelihood comparison points in that direction but does not establish a clear advantage.
- If the sampling assumption holds, the practical payoff is that latent-space choices can be driven by data structure rather than by the availability of gradient estimators, which would simplify model design for clustered, categorical, or symbolic data.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper introduces Joint-stochastic-approximation autoencoders (JAEs), a family of deep directed generative models trained by stochastic approximation. The JSA learning rule pairs maximum-likelihood estimation of the generative model with minimization of the inclusive KL divergence between the posterior and an inference model; the required posterior expectations are obtained by a Metropolis independence sampler using the inference model as proposal. The authors extend this framework to semi-supervised learning by adding a classification term and report experiments on synthetic factor analysis, a Gaussian mixture, discrete sequence data, and MNIST/SVHN semi-supervised benchmarks with Bernoulli latent spaces. The paper claims that JAEs with discrete latent variables achieve performance comparable to continuous-latent DGMs and that the method is robust to encoder/decoder structure mismatch.
Significance. If the central claim is valid, the paper offers a principled alternative to VAE and GAN training that is directly tied to the data log-likelihood, handles discrete latents without gradient estimators, and is robust to mismatch between the inference and generative models. The derivation in Section 3.2 is self-contained: the gradient equations are obtained from log-likelihood and inclusive KL, and the SA formulation is reasonable. The synthetic factor-analysis experiment is a nice controlled demonstration that MIS can compensate for a misspecified inference model. However, the main empirical claim on MNIST/SVHN rests on the assumption that the MIS chains reach stationarity within the 10-step warm-up used in Algorithm 1, and the paper provides no diagnostics to support this. Because that assumption is load-bearing for the claim that the implemented method maximizes the data log-likelihood, the significance of the empirical results is currently conditional on additional evidence.
major comments (3)
- [Algorithm 1, Appendix B.4-B.5] The gradient updates in Algorithm 1 require samples from the posteriors pθ(y,h|x) and pθ(h|x,y). In the implementation described in Appendices B.4 and B.5, the MIS chain is restarted for every data point with a 10-step warm-up, using the factorized inference model qφ(y|x)qφ(h|x) as proposal. For a 60-dimensional Bernoulli space (MNIST) or 220-dimensional space (SVHN), ten component-wise Metropolis steps are very unlikely to have reached stationarity, particularly early in training when qφ is far from pθ. The paper reports no acceptance rates, effective sample sizes, or sensitivity to warm-up length. If the chains are biased, the θ update estimates E_qφ[∂θ log pθ(x,h)] rather than E_pθ[∂θ log pθ(x,h)], so the claim that the implemented algorithm 'directly maximizes the data log-likelihood' is not established. Please provide chain diagnostics and a warm-up sensitivity study, or substantially soften the claim.
- [Section 3.2, Proposition 2] Proposition 2 is a nonparametric consistency statement that assumes infinite capacity and that the KL divergences can be driven to zero. It does not provide a convergence theorem for the SA recursion, nor does it address the finite-capacity, finite-sample regime used in the experiments. As stated, the proposition essentially restates the classical consistency of maximum likelihood in the infinite limit and does not quantify the bias introduced by finite MCMC sampling. The paper would be strengthened by a convergence argument for the SA update under verifiable conditions, or by an explicit statement that the theoretical guarantee is limited to the idealized nonparametric limit.
- [Section 3.3, Eq. (4), and abstract] The abstract and Section 3.2 state that JSA learning directly maximizes the data log-likelihood, but the semi-supervised objective in Eq. (4) includes the term -α Σ_{(x,y)∈L} log qφ(y|x), and the SVHN experiments in Appendix B.5 add further regularizers (entropy minimization and a pseudo-loss). The θ-update in Algorithm 1 is indeed a likelihood gradient, so the core mechanism is intact, but the overall implemented objective is a regularized hybrid rather than the raw data log-likelihood. The paper should clarify this distinction in the abstract and Section 3.3 to avoid overstating the maximum-likelihood claim.
minor comments (5)
- [Section 3.2] There is a typo in the phrase 'minimizes the inclusive KL divergence the between the posteriori and the inference model'; 'the between' should be removed and 'posteriori' should be 'posterior'.
- [Section 4.1] The sentence 'the decoder qφ(h|x) is implemented as a 2d Gaussian' appears to refer to the inference network (encoder), not the decoder; please correct the terminology.
- [Table 1 and Section 4.4] The asterisks in Table 1 are not defined, and the text says the SVHN error is around 6.80% on RGB images while the table lists 6.22±0.55 for JAE+Bernoulli; the relationship between the grayscale and RGB results and the meaning of the asterisks should be stated explicitly.
- [Table 2 caption] The caption contains a typo: 'Evalution' should be 'Evaluation'.
- [References] Several references have inconsistent formatting, e.g., 'arXiv: Artificial Intelligence' for Che et al. and 'arXiv: Machine Learning' for Goodfellow et al.; please unify the reference style.
Circularity Check
No significant circularity: JAE's likelihood and inclusive-KL derivation is re-derived in-paper, and benchmark results are external comparisons.
full rationale
The paper's central derivation is self-contained. Section 3.2 begins from the exact identity ∂/∂θ log pθ(x) = E_{pθ(h|x)}[∂/∂θ log pθ(x,h)], defines the joint optimization in Eq. (2), derives the root-finding system Eq. (3), and verifies it fits the stochastic-approximation form of Eq. (1). The Metropolis independence sampler is constructed with target p(z;λ) and proposal q(z;λ), so the accept/reject step is standard MCMC rather than an imported conclusion. Proposition 2 is a standard consistency statement under nonparametric limits, not a fitted prediction. The SSL extension in Section 3.3 similarly derives Eq. (5) from Eq. (4) and Algorithm 1. Self-citations to Xu & Ou (2016) and Wang et al. (2017) identify the origin of JSA and the multi-move SA technique, but the load-bearing equations are presented and proved in this paper, so the self-citation is not load-bearing. The empirical claims (Tables 1-2, Figures 1-5) are comparisons against external methods and datasets, not re-statements of fitted inputs. The 10-step MCMC warm-up noted in Appendices B.4-B.5 raises a question about whether the implemented sampler is near stationarity; that is an empirical correctness risk, not definitional circularity, and cannot by itself support a circularity finding under the requirement to exhibit a specific reduction.
Assumptions & free parameters
free parameters (4)
- alpha (labeled-data classification weight) =
not reported
- SVHN extra regularization weights =
not reported
- inference output noise variance =
0.05 to 0.01 (GMM); 0.1 (SVHN)
- MIS warm-up steps =
10
assumptions (5)
- standard math Robbins-Monro stochastic approximation regularity conditions hold for the JSA updates
- domain assumption The MIS chain is at stationarity within the finite training budget
- domain assumption The inference model q_phi covers the posterior support
- domain assumption The generative model family contains the true data distribution
- standard math The empirical distribution converges to the true distribution
Cite this review
Pith. "Pith review of Joint-stochastic-approximation Autoencoders with Application to Semi-supervised Learning." pith.science (2026). https://pith.science/paper/63MYTV44
@misc{pith2026250518558,
author = {Pith},
title = {Pith review of: Joint-stochastic-approximation Autoencoders with Application to Semi-supervised Learning},
year = {2026},
howpublished = {\url{https://pith.science/paper/63MYTV44}},
note = {Machine review of arXiv:2505.18558}
}
read the original abstract
Our examination of existing deep generative models (DGMs), including VAEs and GANs, reveals two problems. First, their capability in handling discrete observations and latent codes is unsatisfactory, though there are interesting efforts. Second, both VAEs and GANs optimize some criteria that are indirectly related to the data likelihood. To address these problems, we formally present Joint-stochastic-approximation (JSA) autoencoders - a new family of algorithms for building deep directed generative models, with application to semi-supervised learning. The JSA learning algorithm directly maximizes the data log-likelihood and simultaneously minimizes the inclusive KL divergence the between the posteriori and the inference model. We provide theoretical results and conduct a series of experiments to show its superiority such as being robust to structure mismatch between encoder and decoder, consistent handling of both discrete and continuous variables. Particularly we empirically show that JSA autoencoders with discrete latent space achieve comparable performance to other state-of-the-art DGMs with continuous latent space in semi-supervised tasks over the widely adopted datasets - MNIST and SVHN. To the best of our knowledge, this is the first demonstration that discrete latent variable models are successfully applied in the challenging semi-supervised tasks.
Figures
Figures from the paper (6 more)
Reference graph
Works this paper leans on
-
[1]
Arjovsky, M., Chintala, S., and Bottou, L. Wasserstein gan. arXiv preprint arXiv:1701.07875,
-
[6]
Adversarially learned inference
Dumoulin, V ., Belghazi, I., Poole, B., Lamb, A., Arjovsky, M., Mastropietro, O., and Courville, A. Adversarially learned inference. arXiv preprint arXiv:1606.00704 ,
-
[8]
Maddison, C. J., Mnih, A., and Teh, Y . W. The concrete distribution: A continuous relaxation of discrete random variables. arXiv preprint arXiv:1611.00712, 2016a. Maddison, C. J., Mnih, A., and Teh, Y . W. The concrete distribution: A continuous relaxation of discrete random variables. arXiv: Learning, 2016b. Makhzani, A., Shlens, J., Jaitly, N., Goodfel...
-
[10]
Virtual adversarial training: a regularization method for supervised and semi-supervised learning
Miyato, T., ichi Maeda, S., Koyama, M., and Ishii, S. Virtual adversarial training: a regularization method for supervised and semi-supervised learning. CoRR, abs/1704.03976,
-
[11]
Symmetric Variational Autoencoder and Connections to Adversarial Learning
Pu, Y ., Chen, L., Dai, S., Wang, W., Li, C., and Carin, L. Symmetric variational autoencoder and connections to adversarial learning. CoRR, abs/1709.01846,
-
[12]
O., Bousquet, O., Gelly, S., and Sch ¨olkopf, B
Tolstikhin, I. O., Bousquet, O., Gelly, S., and Sch ¨olkopf, B. Wasserstein auto-encoders. CoRR, abs/1711.01558,
-
[13]
Joint Stochastic Approximation learning of Helmholtz Machines
Xu, H. and Ou, Z. Joint stochastic approximation learning of helmholtz machines. arXiv preprint arXiv:1603.06170,
-
[1995]
Donahue, J., Kr ¨ahenb¨uhl, P., and Darrell, T. Adversarial feature learning. CoRR, abs/1605.09782,
Show all 13 references
-
[1998]
and Bengio, Y
Bornschein, J. and Bengio, Y . Reweighted wake-sleep. arXiv preprint arXiv:1406.2751,
-
[2014]
Kusner, M. J. and Hern ´andez-Lobato, J. M. Gans for se- quences of discrete elements with the gumbel-softmax distribution. CoRR, abs/1611.04051,
-
[2015]
Adversar- ial variational bayes: Unifying variational autoencoders and generative adversarial networks
Mescheder, L., Nowozin, S., and Geiger, A. Adversar- ial variational bayes: Unifying variational autoencoders and generative adversarial networks. arXiv preprint arXiv:1701.04722,
-
[2016]
Importance weighted autoencoders
Burda, Y ., Grosse, R., and Salakhutdinov, R. Importance weighted autoencoders. arXiv preprint arXiv:1509.00519,
-
[2017]
Bengio, Y ., L´eonard, N., and Courville, A. C. Estimating or propagating gradients through stochastic neurons for conditional computation. CoRR, abs/1308.3432,
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.