Skip to content

Neural Architecture Search Powered by Swarm Intelligence 🐜

License

Notifications You must be signed in to change notification settings

SasoPavlic/AutoDaedalus

Β 
Β 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

AutoDaedalus

Description πŸ“

The design of neural network architecture is becoming more difficult as the complexity of the problems we tackle using machine learning increases. Many variables influence the performance of a neural model, and those variables are often limited by the researcher's prior knowledge and experience. In our master's thesis, we will focus on becoming familiar with evolutionary neural network design, anomaly detection techniques, and a deeper knowledge of autoencoders and their potential for application in unsupervised learning. Our practical objective will be to build a neural architecture search based on swarm intelligence, and construct an autoencoder architecture for anomaly detection in the MNIST dataset.

What it can do? πŸ‘€

  • Construct novel autoencoder's architecture using neural architecture search (NAS) based on ant colony optimization (Swarm intelligence)
  • Allow an unsupervised machine learning algorithm to make decisions that mark the threshold between normal and anomalous data instances.
  • Finds anomalies in MNIST dataset based on configuration parameters

Requirements βœ…

  • CUDA supported GPU
  • CUDA release 11.1
  • Tensorflow 2.4.1
  • Keras 2.3.1
  • Python 3.8
INSTALL ALL

pip3 install -r requirements.txt

Documentation πŸ“˜

Usage πŸ”¨

Configure settings/autoencoder.yaml according to your needs. Run script with Pythonautodaedalus_main.py

Flowchart πŸ“

Flowchart representing AutoDaedalus workflow

Future goals 🌟

  • Adjust input node to accept any dataset automatic (CIFAR-10, Fashion mnist, ...)
  • Split functions into smaller components
  • Improve logging and final results readability

Acknowledgments πŸŽ“

AutoDaedalus was developed under the supervision of doc. dr Saőo Karakatič for the Master degree of Informatics and Technologies of Communication in University of Maribor.

About

Neural Architecture Search Powered by Swarm Intelligence 🐜

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%