xinhen / genetic-cnn Goto Github PK

Genetic Algorithm for Convolutional Neural Networks

CNN architecture exploration using Genetic Algorithm as discussed in the following paper: Genetic CNN

Figure 1: Adapted from Genetic CNN paper. A two-stage network with 4 and 5 nodes at first and second stage respectively. The default input and output nodes are shown in red and green colour respectively. The node connections within light blue region are learned via Genetic Algorithm. The default input node (in red) will be connected to each node without any predecessor in an encoded region. Likewise, the node without successor will be connected to default output node (in green). Also, the nodes without connections will be dropped from the graph (e.g. node B2 in stage 2). Moreover, if a node has more than one parent, they will be summed element-wise before feeding as input to that layer. Within each stage, the number of convolutional filters is a constant and the spatial resolution remains unchanged. The pooling layers are added to down-sample the dataset at each stage.

Tools Required

Python 3.5 is used during development and following libraries are required to run the code provided in the notebook:

• Tensorflow
• DEAP
• py-dag

Note: If you see mistakes or want to suggest changes, please submit a pull request.