My implemenation of ABC-Net. Currently I finish the ABC-Net on alexnet architecture on imagenet dataset, but the model does NOT converge. I am considering use some small dataset which is relative easy to debug.
IF YOU FIND THE MISTAKEs I MADE IN THE IMPLEMNATION THAT CAUSE THE DISCONVERGENCY, PLEASE LET ME KNOW. thanks: )
- ABC-Net in resnet-18 architecture on cifar10 dataset
Considering some details are NOT specified in the paper, I make modifications as follow:
- the way to solve alpha.: The lstsq(scipy.linalg.lstsq) method is adopted.
- STE attach to each Binary Base OR full precision Weight: fp Weight
- gradient of W(see following Notes): keep as the original paper
- alexnet architecture: basicly adopting the modification in Xnor-Net excepting kernel size to keep identity with pretrained model.
I find some preoblem of this paper reported in the notes.