Cell Variational Information Bottleneck Network

arXiv:2403.15082v1 Announce Type: new
Abstract: In this work, we propose Cell Variational Information Bottleneck Network (cellVIB), a convolutional neural network using information bottleneck mechanism, which can be combined with the latest feedforward network architecture in an end-to-end training method. Our Cell Variational Information Bottleneck Network is constructed by stacking VIB cells, which generate feature maps with uncertainty. As layers going deeper, the regularization effect will gradually increase, instead of directly adding excessive regular constraints to the output layer of the …

abstract architecture arxiv cells convolutional neural network cs.cv feature generate information maps network network architecture neural network training type work