Divide-and-conquer routing for learning heterogeneous individualized capsules

用于学习异构个性化胶囊的分治路由

阅读:1

Abstract

Capsule Networks (CapsNets) have demonstrated an enhanced ability to capture spatial relationships and preserve hierarchical feature representations compared to Convolutional Neural Networks (CNNs). However, the dynamic routing mechanism in CapsNets introduces substantial computational costs and limits scalability. In this paper, we propose a divide-and-conquer routing algorithm that groups primary capsules, enabling the model to leverage independent feature subspaces for more precise and efficient feature learning. By partitioning the primary capsules, the initialization of coupling coefficients is aligned with the hierarchical structure of the capsules, addressing the limitations of existing initialization strategies that either disrupt feature aggregation or lead to excessively small activation values. Additionally, the grouped routing mechanism simplifies the iterative process, reducing computational overhead and improving scalability. Extensive experiments on benchmark image classification datasets demonstrate that our approach consistently outperforms the original dynamic routing algorithm as well as other state-of-the-art routing strategies, resulting in improved feature learning and classification accuracy. Our code is available at: https://github.com/rqfzpy/DC-CapsNet.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。