arXiv · 2007.00024
Data-driven Regularization via Racecar Training for Generalizing Neural Networks
Abstract
We propose a novel training approach for improving the generalization in neural networks. We show that in contrast to regular constraints for orthogonality, our approach represents a {\em data-dependent} orthogonality constraint, and is closely related to singular value decompositions of the weight matrices. We also show how our formulation is easy to realize in practical network architectures via a reverse pass, which aims for reconstructing the full sequence of internal states of the network. Despite being a surprisingly simple change, we demonstrate that this forward-backward training approach, which we refer to as {\em racecar} training, leads to significantly more generic features being extracted from a given data set. Networks trained with our approach show more balanced mutual information between input and output throughout all layers, yield improved explainability and, exhibit improved performance for a variety of tasks and task transfers.
Explore related subjects
Keep this discovery
You Xie, Nils Thuerey. 2020-06-30. Data-driven Regularization via Racecar Training for Generalizing Neural Networks. https://arxiv.org/abs/2007.00024
Cite the original work for its findings. Save a collection to share your selection of sources.