Skip to content

ybwang119/elec4544

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

elec4544

This reposiry is for HKU elec4544. We test the idea of adding more channels with group convolution. Here is the abstract.

It has been a consensus that group convolution could reduce the computational cost but no direct research could show that such an operator may increase or deteriorate the image classification accuracy. Inspired by the ResNeXt, we focused on the relationship between wider channels and more groups in the modification of classic networks, i.e., AlexNet and VGG-16. It turns out that with our modification, the top-1 classification accuracy of AlexNet increased by 2.22% and 1.88% on Flowers and MiniImageNet respectively, and for VGG-16 our modified models achieved 2.06%, 1.48% and 3.40% more validation accuracy on CIFAR-10, Flowers and MiniImageNet respectively. We also designed the ablation study to illustrate the necessity of adopting group convolution when attempting to enlarge the VGG-16 network architecture. Our experiment proves that group convolution is not only a way to reduce the computational cost but also a way to avoid over-fitting, accelerate the training process, and enhance the generalization capability.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages