System and method for batch-normalized recurrent highway networks
Embodiments of the present disclosure relate to a recurrent framework based on Recurrent Highway Networks (RHNs) for sequence modeling using batch normalization. In certain embodiments, constraints within the RHNs are relaxed to reduce or avoid gradient vanishing or exploding by normalizing the current transition units in highway layers.
1. A computer-implemented method for image object recognition and captioning, comprising:
loading a pre-trained model into a multi-layer Recurrent Highway Network;
receiving a digital image input on which to perform object recognition, the digital image input containing one or more objects;
batch normalizing the digital image input by applying one or more non-linear transforms to the digital image input to generate transform outputs from the multi-layer Recurrent Highway Network; and
combining the transform outputs to generate a final output from the multi-layer Recurrent Highway Network, wherein the final output comprises a caption describing the one or more objects recognized in the digital image input.
2. The method of claim 1 wherein the pre-trained model contains parameters for the computing network nodes.
3. The method of claim 1 wherein the multi-layer neural network is a batch-normalized Recurrent Highway Network.
4. The method of claim 3 wherein the constraints within the Recurrent Highway Network are relaxed to reduce or avoid gradient vanishing or exploding by normalizing the current transition units in highway layers.
5. The method of claim 4 , wherein a gradient norm clipping strategy is adopted to improve language semantics of the caption.
6. A computer-implemented method for image object recognition and captioning, comprising:
receiving a digital image input on which to perform object recognition, the digital image input containing one or more objects;
batch normalizing the digital image input by applying one or more non-linear transforms to the digital image input to generate transform outputs from a multi-layer Recurrent Highway Network, wherein batch normalizing occurs at one or more highway layers of the multi-layer Recurrent Highway Network, wherein the one or more highway layers comprise a transform gate and a carry gate, and wherein the transform gate is coupled to the carry gate; and
combining the transform outputs to generate a final output, wherein the final output comprises a caption describing the one or more objects in the digital image input.
7. The computer-implemented method of claim 6 , wherein the digital image input is an output of a prior processing loop in the recurrent highway network.
8. The computer-implemented method of claim 6 , wherein batch normalizing the digital image input comprises extracting a mean and a variance across each channel and spatial location.
9. The computer-implemented method of claim 8 , wherein batch normalizing the digital image input comprises subtracting the mean of the digital image input and dividing by the variance of the digital image input.