ADD CapsNet example#8674
ADD CapsNet example#8674Soonhwan-Kwon wants to merge 7 commits intoapache:masterfrom ai-adv-lab:master
Conversation
|
We're planning to make this capsule net example as gluon also. Is there anything to do please let me know. |
| Sara Sabour, Nicholas Frosst, Geoffrey E Hinton. Dynamic Routing Between Capsules. NIPS 2017 | ||
| - The Current best test error is 0.5% | ||
|
|
||
| Due to the permission issue, this example is maintained in this [repository](https://github.com/samsungsds-rnd/capsnet.mxnet) separately. |
There was a problem hiding this comment.
If you mean the last line, Pull Request can takes days to accept, meanwhile some follow-up changes can be updated to our temporary repository https://github.com/samsungsds-rnd/capsnet.mxnet.
|
Does the results match the original paper? |
|
The original paper's result was 0.25% and we expect close accuracy(0.34%) can be achieved by adding data augmentation(scale shift) like other implementations of other platforms(keras, tensorflow) which achieved 0.34 % (the average) result. The result of 0.25 is hard to achieve but we expect close enough accuracy can be obtained soon. |
|
I expect that there are many users of MXNet finding implementation of CapsNet. We're in the middle of implementing data augmentation and will make effort to follow up the accuracy soon. |
|
@piiswrong We added data augmentation(shifted up to 2 pixels) as on paper, and fixed structure by blocking gradient on both bias(as constant) and inputs_hat before routing ends. Now it achieves test error =0.44% at epoch 40. |
|
@piiswrong
※ tensorflow implementation's batch size is not 100 but 128 but MXNet and Keras implementation 's batchsize are 100. Sorry for repeating update but would you mind noticing us that it can be merged or not? I'm afraid that there is no response since 2 days ago, and it seems that it is not acceptable as example for MXNet. |
Description
This example is MXNet implementation of CapsNet:
Sara Sabour, Nicholas Frosst, Geoffrey E Hinton. Dynamic Routing Between Capsules. NIPS 2017
The current best test error is 0.5%
Checklist
Essentials
make lint)Changes