ostris Pretrains 'Primal VAE': Aggressive 2x-Downsample, 1-Channel Latent Design

ostrisai · x · 2026-10-26

Researcher ostris is pretraining a new VAE concept, Primal VAE: only 2x spatial downsampling to 1 channel. Patch-4 the latent and you get 8x at 16 channels — same as most modern VAEs — without deep compression.

Current details: pretraining with only LPIPS and MAE for now (critic coming soon); training with a loss to keep the norm in line with noise, plus EQVAE-style latent flips and rotations to make the latent space easier to diffuse.

Related event: ostris pretrains radical Primal VAE with 1-channel latent(2 posts)→

Original post →

More from Research

Research channel →