pith. sign in

Contrastive Variational Autoencoder Enhances Salient Features

2 Pith papers cite this work. Polarity classification is still indexing.

2 Pith papers citing it
abstract

Variational autoencoders are powerful algorithms for identifying dominant latent structure in a single dataset. In many applications, however, we are interested in modeling latent structure and variation that are enriched in a target dataset compared to some background---e.g. enriched in patients compared to the general population. Contrastive learning is a principled framework to capture such enriched variation between the target and background, but state-of-the-art contrastive methods are limited to linear models. In this paper, we introduce the contrastive variational autoencoder (cVAE), which combines the benefits of contrastive learning with the power of deep generative models. The cVAE is designed to identify and enhance salient latent features. The cVAE is trained on two related but unpaired datasets, one of which has minimal contribution from the salient latent features. The cVAE explicitly models latent features that are shared between the datasets, as well as those that are enriched in one dataset relative to the other, which allows the algorithm to isolate and enhance the salient latent features. The algorithm is straightforward to implement, has a similar run-time to the standard VAE, and is robust to noise and dataset purity. We conduct experiments across diverse types of data, including gene expression and facial images, showing that the cVAE effectively uncovers latent structure that is salient in a particular analysis.

fields

cs.CL 1 cs.LG 1

years

2025 1 2024 1

verdicts

UNVERDICTED 2

representative citing papers

Frictional Q-Learning

cs.LG · 2025-09-24 · unverdicted · novelty 7.0

Frictional Q-Learning encodes supported actions as tangent directions on an action manifold using a contrastive variational autoencoder to reduce extrapolation errors in off-policy reinforcement learning.

citing papers explorer

Showing 2 of 2 citing papers.

  • Frictional Q-Learning cs.LG · 2025-09-24 · unverdicted · none · ref 1 · internal anchor

    Frictional Q-Learning encodes supported actions as tangent directions on an action manifold using a contrastive variational autoencoder to reduce extrapolation errors in off-policy reinforcement learning.

  • SimSiam Naming Game: A Unified Approach for Representation Learning and Emergent Communication cs.CL · 2024-10-29 · unverdicted · none · ref 2 · internal anchor

    SSNG replaces sampling-based updates in MHNG with symmetric self-supervised representation alignment using Gumbel-Softmax for discrete messages, yielding higher linear-probe classification accuracy on CIFAR-10 and ImageNet-100 than referential, reconstruction, or MHNG baselines.