Since their introduction by Sabour et al., capsule networks have been extended to 2D semantic segmentation with the introduction of convolutional capsules. While extended further to 3D semantic segmentation when mixed with Convolutional Neural Networks (CNNs), no capsule-only network (to the best of our knowledge) has been able to reach CNNs’ accuracy on multilabel segmentation tasks. In this work, we propose OnlyCaps-Net, the first competitive capsule-only network for 2D and 3D multi-label semantic segmentation. OnlyCaps-Net improves both capsules’ accuracy and inference speed by replacing Sabour et al. squashing with the introduction of two novel squashing functions, i.e. softsquash or unitsquash, and the iterative routing with a new parameter free single pass routing, i.e. unit routing. Additionally, OnlyCaps-Net introduces a new parameter efficient convolutional capsule type, i.e. depthwise separable convolutional capsule.
CITATION STYLE
Bonheur, S., Thaler, F., Pienn, M., Olschewski, H., Bischof, H., & Urschler, M. (2022). OnlyCaps-Net, a Capsule only Based Neural Network for 2D and 3D Semantic Segmentation. In Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics) (Vol. 13435 LNCS, pp. 340–349). Springer Science and Business Media Deutschland GmbH. https://doi.org/10.1007/978-3-031-16443-9_33
Mendeley helps you to discover research relevant for your work.