An Achromatic Approach of Compressing CNN Filters by Clustering Pattern-Specific Receptive Fields (Data-2040-Final-Project)
Group Member: Guanzhong Chen, Shiyu Liu, Guansu Niu, Cangcheng Tang, Zhi Wang
Medium: Initial Blog Post, Midway Blog Post, Final Blog Post
Project on Next Journal
Screencast
In studies of image recognition, there are many gray-scale pictures, such as chest radiographs. Currently, the idea behind training those images is to apply models that are essentially designed for training color pictures, such as ResNet and DenseNet. This can lead to many redundant parameters during the process. Therefore, this project aimed to discover a methodology to modify the models trained on colored images and to apply them to gray-scale images.
Merging and clustering notebooks are called DenseNet_merge_first_layer.ipynb
for merging the first layer and DenseNet_merge_multipule_layer.ipynb
for merge the first two layers.
This project used CIFAR-10 dataset which you probably known already. It consists of 60,000 32 by 32 pixels color images in 10 classes, with 6,000 images per class. Also The training to testing ratio is 5:1.
Model | Image Type | Val Accuracy | Number of Parameters |
---|---|---|---|
Color Model | Color Images | 91.16% | 995,230 |
Color Model | Grayscale Images | 83.20% | 995,230 |
Modified 1st Layer | Grayscale Images | 85.43% | 964,237 |
Modified 1st & 2nd Layer | Grayscale Images | 84.83% | 950,046 |
Notes: The third and fourth models were trained for 1 epoch on the batch normalization layer only.