Header menu link for other important links
X
Retrospective Loss: Looking Back to Improve Training of Deep Neural Networks
S. Jandial, A. Chopra, M. Sarkar, P. Gupta, B. Krishnamurthy,
Published in Association for Computing Machinery
2020
Pages: 1123 - 1131
Abstract
Deep neural networks (DNNs) are powerful learning machines that have enabled breakthroughs in several domains. In this work, we introduce a new retrospective loss to improve the training of deep neural network models by utilizing the prior experience available in past model states during training. Minimizing the retrospective loss, along with the task-specific loss, pushes the parameter state at the current training step towards the optimal parameter state while pulling it away from the parameter state at a previous training step. Although a simple idea, we analyze the method as well as to conduct comprehensive sets of experiments across domains - images, speech, text, and graphs - to show that the proposed loss results in improved performance across input domains, tasks, and architectures. © 2020 ACM.