arXiv · 2610.09242
Pooling Representation Autoencoders for Efficient Diffusion
Abstract
Representation Autoencoders (RAEs) generate images from pre-trained visual fea- tures, but their dense token grids make generative modeling expensive. Motivated by local feature correlations, we introduce PoolDINO, a learned affine pooling operator that merges neighboring tokens. Training the pooling operator jointly with the RGB decoder preserves the standard two-stage RAE procedure without a separate feature auto-encoder. On ImageNet-256, 4x token compression retains comparable generation quality under internal guidance, while 16x compression trades some quality for greater efficiency. At a fixed budget of 100 sampling steps, latent-sampling throughput increases by 3.7x and 9.0x, respectively, relative to the unpooled baseline. Classification and dense prediction evaluations show that comparable guided generation quality can coexist with weaker performance on other tasks.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Ramón Calvo-González, Youssef Saied, François Fleuret. 2026-10-07. Pooling Representation Autoencoders for Efficient Diffusion. https://arxiv.org/abs/2610.09242
Cite the original work for its findings. Save a collection to share your selection of sources.