[edit]
Developing Autoencoder: Incremental Bottleneck Expansion Leads to an Informed Latent Space
Proceedings of the First Workshop on NeuroAI Multimodal Intelligence @ AAAI 2026, PMLR 308:58-66, 2026.
Abstract
Representation learning models, such as autoencoders (AEs), can effectively extract meaningful and generalizable features from natural image data. However, the learned latent features are often mixed or distributed across all bottleneck units, making interpretation difficult. Previous work has sought to address this by explicitly optimizing for feature separation or ordering. We propose a biologically inspired progressive learning scheme, the Developing Autoencoder (Dev-AE), which incrementally expands the representational capacity. Increasing the size of the bottleneck layer over training epochs forces the Dev-AE to first learn compressed, low-dimensional representations before expanding into progressively higher-dimensional feature spaces. Comparing the latent space organization in Dev-AEs with that in standard AEs and PCA-initialized AEs (PCA-AE), we observe improved feature ordering and higher activation sparsity. Moreover, Dev-AEs show better classification performance based on the learned encodings, with units added in the final increment contributing the most. Our findings indicate that an incremental latent space expansion fosters ordered, sparse, and more diverse representations, leading to more efficient use of representational capacity and improved classification accuracy, thereby offering a promising route toward interpretable and compact encodings.