Coloring Deep CNN Layers with Activation Hue Loss
Abstract
This paper proposes a novel hue-like angular parameter to model the structure of deep convolutional neural network (CNN) activation space, referred to as the {\em activation hue}, for the purpose of regularizing models for more effective learning. The activation hue generalizes the notion of color hue angle in standard 3-channel RGB intensity space to -channel activation space. A series of observations based on nearest neighbor indexing of activation vectors with pre-trained networks indicate that class-informative activations are concentrated about an angle in both the image plane and in multi-channel activation space. A regularization term in the form of hue-like angular labels is proposed to complement standard one-hot loss. Training from scratch using combined one-hot + activation hue loss improves classification performance modestly for a wide variety of classification tasks, including ImageNet.
Cite
@article{arxiv.2310.03911,
title = {Coloring Deep CNN Layers with Activation Hue Loss},
author = {Louis-François Bouchard and Mohsen Ben Lazreg and Matthew Toews},
journal= {arXiv preprint arXiv:2310.03911},
year = {2023}
}