Cannistraci-Hebb Channel-wise Dynamic Sparse Training of Convolutional Neural Networks with Contextual Modulation
Abstract
Dynamic Sparse Training (DST) is an effective paradigm for sparse to sparse training of neural network connectivity under a fixed parameter budget. Recent advances in network-science for AI introduced epitopological learning methods, such as Cannistraci-Hebb Training (CHT), demonstrating that network automata applied to the mere network topology enables gradient-free predictions of the sparse connectivity evolution that can trigger significant increase in task performance. However, these methods were currently developed for standard MLP-like fully-connected architectures and need significant rethinking to be extended to convolutional neural networks (CNNs) due to element-wise weight sharing and spatially repeated interactions. In this work, we bridge this gap by introducing epitopological learning DST for CNNs through channel-wise network modeling to address the prohibitive complexity of element-wise modeling, which treats each kernel-channel as a single network node. Results show orders-of-magnitude improvements in time complexity of link-regrowth when applied at channel-level with respect to element-level. We also enhance the channel-wise formulation with lightweight contextual modulation to improve expressivity. Extensive experiments on CIFAR-10, CIFAR-100, and Tiny-ImageNet with ResNet and VGG architectures demonstrate that our method achieves performance comparable to dense training using only 30% of the parameters, which significantly reduces computational cost. Furthermore, our approach improves robustness under noisy inputs. These results highlight the effectiveness of topology-aware sparse training and the benefit of decoupling graph construction from sparse update rules in convolutional networks.