LE-CapsNet: A Light and Enhanced Capsule Network
Capsule Network (CapsNet) classifier has several advantages over CNNs, including better detection of images containing overlapping categories and higher accuracy on transformed images. Despite the advantages, CapsNet is slow due to its different structure. In addition, CapsNet is resource-hungry, includes many parameters and lags in accuracy compared to CNNs. In this work, we propose LE-CapsNet as a light, enhanced and more accurate variant of CapsNet. Using 3.8M weights, LECapsNet obtains 76.73% accuracy on the CIFAR-10 dataset while performing inference 4x faster than CapsNet. In addition, our proposed network is more robust at detecting images with affine transformations compared to CapsNet. We achieve 94.3% accuracy on the AffNIST dataset (compared to CapsNet 90.52%).
Code (0)
등록된 구현이 없습니다.
Similar Papers 제목 키워드 기반
Quick-CapsNet (QCN): A fast alternative to Capsule Networks
The basic computational unit in Capsule Network (CapsNet) is a capsule (vs. neurons in Convolutional Neural Networks (CNNs)). A capsule is a set of neurons, which form a vector. CapsNet is used for supervised classificat…
DL-CapsNet: A Deep and Light Capsule Network
Capsule Network (CapsNet) is among the promising classifiers and a possible successor of the classifiers built based on Convolutional Neural Network (CNN). CapsNet is more accurate than CNNs in detecting images with over…
E2-Capsule Neural Networks for Facial Expression Recognition Using AU-Aware Attention
Capsule neural network is a new and popular technique in deep learning. However, the traditional capsule neural network does not extract features sufficiently before the dynamic routing between the capsules. In this pape…
Facial Expression RecognitionFacial Expression Recognition (FER)PT-CapsNet: A Novel Prediction-Tuning Capsule Network Suitable for Deeper Architectures
Capsule Networks (CapsNets) create internal representations by parsing inputs into various instances at different resolution levels via a two-phase process -- part-whole transformation and hierarchical component rout…
object-detectionObject DetectionSemantic SegmentationQuantum Capsule Networks
Capsule networks, which incorporate the paradigms of connectionism and symbolism, have brought fresh insights into artificial intelligence. The capsule, as the building block of capsule networks, is a group of neurons re…
Quantum Machine Learning