Learning Discourse-level Diversity for Neural Dialog Models using Conditional Variational Autoencoders

Citation

One-Line Takeaway

This paper is one of the clearest bridges from classical CVAEs to text generation: it treats dialogue response generation as a conditional one-to-many problem, uses a latent variable for discourse-level variation, and adds practical losses to keep the latent from being ignored.

Technical Core

  • Let cc denote dialogue context, xx the target response, and zz a latent discourse variable. The core model is pθ(x,z∣c)=pθ(x∣z,c) pθ(z∣c).p_\theta(x, z \mid c) = p_\theta(x \mid z, c)\, p_\theta(z \mid c).
  • Training introduces a recognition model qϕ(z∣x,c)q_\phi(z \mid x, c) and optimizes the conditional ELBO LCVAE=Eqϕ(z∣x,c)[log⁡pθ(x∣z,c)]−KL ⁣(qϕ(z∣x,c) ∥ pθ(z∣c)).\mathcal{L}_{\mathrm{CVAE}} = \mathbb{E}_{q_\phi(z \mid x,c)}[\log p_\theta(x \mid z,c)] - \mathrm{KL}\!\left(q_\phi(z \mid x,c)\,\|\,p_\theta(z \mid c)\right).
  • In the knowledge-guided version, the model also predicts a linguistic label yy such as a dialogue act, adding an auxiliary term Eqϕ(z∣x,c,y)[log⁡pθ(y∣z,c)].\mathbb{E}_{q_\phi(z \mid x,c,y)}[\log p_\theta(y \mid z,c)].
  • To prevent latent underuse, the paper augments training with a bag-of-words reconstruction term Eqϕ(z∣x,c,y)[log⁡pθ(xbow∣z,c)],\mathbb{E}_{q_\phi(z \mid x,c,y)}[\log p_\theta(x_{\mathrm{bow}} \mid z,c)], which forces zz to carry response-level content not reducible to local autoregressive token prediction.

Method

  • Encode dialogue history with utterance encoders plus a recurrent context encoder to obtain cc.
  • Parameterize both the prior pθ(z∣c)p_\theta(z \mid c) and posterior qϕ(z∣x,c)q_\phi(z \mid x,c) as Gaussian networks.
  • Decode the response with a GRU conditioned on both cc and sampled zz.
  • In kgCVAE, inject explicit linguistic features such as dialogue acts and train a network to predict them from (z,c)(z,c).
  • Use KL annealing and the bag-of-words loss to reduce the vanishing-latent problem that otherwise appears with the recurrent decoder.

Empirical Evaluation

  • Experiments are on the Switchboard dialogue corpus with multiple-reference evaluation assembled to reflect the one-to-many nature of dialogue.
  • The main empirical claim is that CVAE and especially kgCVAE increase discourse-level response diversity while keeping outputs relevant and coherent.
  • The bag-of-words loss is also evaluated separately as a training device for keeping the latent informative.

Assumptions And Limits

The learned latent is still an unconstrained continuous discourse variable, not an identifiable or task-grounded strategy label. The setting is dialogue generation rather than general language-model factorization, and the paper depends on practical heuristics such as KL annealing and auxiliary losses rather than a clean theorem for latent semantics.

Relevance To This Project

  • Similarities:
    • The paper uses the same conditional latent decomposition that underlies the repo's CVAE methodology: p(z∣x),p(y∣x,z),q(z∣x,y).p(z \mid x), \qquad p(y \mid x, z), \qquad q(z \mid x, y).
    • It explicitly treats conditional text generation as a one-to-many mapping and uses zz to capture high-level variation beyond what the condition already determines.
    • It directly grapples with the latent-underuse problem in a sequence decoder, which is central to the repo's methodology.
  • Differences:
    • The latent here is mainly a discourse-intent or response-style variable, not a latent reasoning strategy attached to problem solving.
    • The model is trained as a dialogue generator from data, whereas the repo often starts from an already-strong entangled generator and asks how to factorize or post-train it.
    • The paper's main anti-collapse tools are KL annealing, bag-of-words prediction, and optional dialogue-act supervision; the repo studies a broader family of ELBO-adjacent objectives aimed at making latent modes semantically sharper and more reusable.

Comparison Notes

Sohn et al. (2015) is the clean conditional-VAE predecessor in structured prediction, while this paper transports the same idea into autoregressive text generation. Bowman et al. (2016) is the unconditional sentence-VAE ancestor that makes the sequence-collapse problem vivid. Fu et al. (2019) later studies a cleaner scheduling fix for the same latent-vanishing issue in text VAEs.

Source Note

Retrieval: alphaXiv MCP get_paper_content on arXiv:1703.10960; ACL Anthology page metadata at https://aclanthology.org/P17-1061/; no local cache or downloaded source artifacts were created.

Verified from source: title, authors, venue, the conditional factorization p(x,z∣c)=p(x∣z,c)p(z∣c)p(x,z \mid c)=p(x \mid z,c)p(z \mid c), the use of posterior inference q(z∣x,c)q(z \mid x,c), the knowledge-guided dialogue-act auxiliary prediction, the bag-of-words loss, and the Switchboard evaluation framing were checked against source material.

Built with LogoFlowershow