# A Particle Swarm Optimization-based Flexible Convolutional Auto-Encoder for Image Classification

arXiv: https://arxiv.org/abs/1712.05042 · HTML: https://mlsanity.com/paper/1712.05042/

## Key facts

- Cited by 11 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 13 Dec 2017.

## Built on by (most cited first, 11 of 11)

- [Homogeneous Architecture Augmentation for Neural Predictor](https://mlsanity.com/paper/2107.13153/) (2021)
- [Multi-objective Search of Robust Neural Architectures against Multiple Types of Adversarial Attacks](https://mlsanity.com/paper/2101.06507/) (2021)
- [Accelerating Evolutionary Neural Architecture Search via Multi-Fidelity Evaluation](https://mlsanity.com/paper/2108.04541/) (2021)
- [An algorithmic framework for the optimization of deep neural networks architectures and hyperparameters](https://mlsanity.com/paper/2303.12797/) (2023)
- [EGANS: Evolutionary Generative Adversarial Network Search for Zero-Shot Learning](https://mlsanity.com/paper/2308.09915/) (2023)
- [A Pairwise Comparison Relation-assisted Multi-objective Evolutionary Neural Architecture Search Method with Multi-population Mechanism](https://mlsanity.com/paper/2407.15600/) (2024)
- [Quantum autoencoders for image classification](https://mlsanity.com/paper/2502.15254/) (2025)
- [Knowledge Transfer based Evolutionary Deep Neural Network for Intelligent Fault Diagnosis](https://mlsanity.com/paper/2109.13479/) (2021)
- [Automating Neural Architecture Design without Search](https://mlsanity.com/paper/2204.11838/) (2022)
- [Early Diagnosis of Retinal Blood Vessel Damage via Deep Learning-Powered Collective Intelligence Models](https://mlsanity.com/paper/2210.09449/) (2022)
- [Compression-Driven Anomaly Detection in Brain MRI Using an Interpretable Quantum Autoencoder](https://mlsanity.com/paper/2606.27411/) (2026)

## Abstract (arXiv)

Convolutional auto-encoders have shown their remarkable performance in stacking to deep convolutional neural networks for classifying image data during past several years. However, they are unable to construct the state-of-the-art convolutional neural networks due to their intrinsic architectures. In this regard, we propose a flexible convolutional auto-encoder by eliminating the constraints on the numbers of convolutional layers and pooling layers from the traditional convolutional auto-encoder. We also design an architecture discovery method by using particle swarm optimization, which is capable of automatically searching for the optimal architectures of the proposed flexible convolutional auto-encoder with much less computational resource and without any manual intervention. We use the designed architecture optimization algorithm to test the proposed flexible convolutional auto-encoder through utilizing one graphic processing unit card on four extensively used image classification datasets. Experimental results show that our work in this paper significantly outperform the peer competitors including the state-of-the-art algorithm.