Chaotic CNN for Limited Data Image Classification

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

A new method enhances Convolutional Neural Network (CNN) performance in limited data scenarios by applying chaos-based feature transformations. This technique uses nonlinear transformations from logistic, skew tent, and sine maps on normalized feature vectors before the classification layer, which reshapes the feature space and improves class separability. The approach was tested on MNIST, Fashion-MNIST, and CIFAR-10 datasets with varying CNN depths and limited data. It consistently improved performance over standalone CNNs, achieving a maximum gain of 5.43% on MNIST with a skew tent map and a 3-layer CNN at 40 samples per class. A 9.11% gain was observed on Fashion-MNIST using a sine map with a 3-layer CNN at 50 samples per class, and 7.47% on CIFAR-10 with a skew tent map at 200 samples per class. The method is computationally efficient and requires no new trainable parameters.

Key takeaway

For AI Engineers building image classification models with limited datasets, integrating chaos-based feature transformations can significantly boost generalization without adding model complexity. You should consider applying logistic, skew tent, or sine maps to your CNN's feature vectors to improve class separability, especially when working with small sample sizes like 40-200 images per class. This approach offers a practical and computationally efficient way to enhance model performance.

Key insights

Chaos-based nonlinear transformations enhance CNN generalization and class separability with limited data.

Principles

Method

Apply nonlinear transformations using logistic, skew tent, or sine maps to normalized feature vectors before the CNN's classification layer to reshape the feature space.

In practice

Topics

Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.