Capsule Network Is Not More Robust Than Convolutional Network
Jindong Gu, Volker Tresp, Han Hu
Abstract
The Capsule Network is widely believed to be more robust than Convolutional Networks. However, there are no comprehensive comparisons between these two networks, and it is also unknown which components in the CapsNet affect its robustness. In this paper, we first carefully examine the special designs in CapsNet that differ from that of a ConvNet commonly used for image classification. The examination reveals five major new/different components in CapsNet: a transformation process, a dynamic routing layer, a squashing function, a marginal loss other than cross-entropy loss, and an additional class-conditional reconstruction loss for regularization. Along with these major differences, we conduct comprehensive ablation studies on three kinds of robustness, including affine transformation, overlapping digits, and semantic representation. The study reveals that some designs, which are thought critical to CapsNet, actually can harm its robustness, i.e., the dynamic routing layer and the transformation process, while others are beneficial for the robustness. Based on these findings, we propose enhanced ConvNets simply by introducing the essential components behind the CapsNet's success. The proposed simple ConvNets can achieve better robustness than the CapsNet.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 84a24355-fdcb-4757-9b6f-b50648b6fd49Cited by top-tier papers3
- LAS-AT: Adversarial Training with Learnable Attack StrategyXiaojun Jia, Yong Zhang, Baoyuan Wu, Ke Ma et al.CVPR 2022 · 140 citations
- Generating Transferable 3D Adversarial Point Cloud via Random Perturbation FactorizationBangyan He, Jian Liu, Yiming Li, Siyuan Liang et al.AAAI 2023 · 53 citations
- Why Capsule Neural Networks Do Not Scale: Challenging the Dynamic Parse-Tree AssumptionMatthias Mitterreiter, Marcel Koch, Joachim Giesen, Sören LaueAAAI 2023 · 17 citations
Builds on8
- Local Relation Networks for Image RecognitionHan Hu, Zheng Zhang, Zhenda Xie, Stephen LinICCV 2019 · 555 citations
- Capsule Routing via Variational BayesFabio De Sousa Ribeiro, Georgios Leontidis, Stefanos D. KolliasAAAI 2020 · 93 citations
- Capsules with Inverted Dot-Product Attention RoutingYao-Hung Hubert Tsai, Nitish Srivastava, Hanlin Goh, Ruslan SalakhutdinovICLR 2020 · 91 citations
- Detecting and Diagnosing Adversarial Images with Class-Conditional Capsule ReconstructionsYao Qin, Nicholas Frosst, Sara Sabour, Colin Raffel et al.ICLR 2020 · 76 citations
- Interpretable Graph Capsule Networks for Object RecognitionJindong GuAAAI 2021 · 42 citations
Related papers
- Improving the Robustness of Capsule Networks to Image Affine TransformationsJindong Gu, Volker TrespCVPR 2020
- Effective and Efficient Vote Attack on Capsule NetworksJindong Gu, Baoyuan Wu, Volker TrespICLR 2021 · 28 citations
- Enabling Equivariance for Arbitrary Lie GroupsLachlan E. MacDonald, Sameera Ramasinghe, Simon LuceyCVPR 2022 · 11 citations
- PT-CapsNet: A Novel Prediction-Tuning Capsule Network Suitable for Deeper ArchitecturesChenbin Pan, Senem VelipasalarICCV 2021 · 11 citations
- OrthCaps: An Orthogonal CapsNet with Sparse Attention Routing and PruningXinyu Geng, Jiaming Wang, Jiawei Gong, Yuerong Xue et al.CVPR 2024
