Generalization in fully-connected neural networks for time series forecasting

Anastasia Borovykh*, Cornelis W. Oosterlee, Sander M. Bohté

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

Abstract

In this paper we study the generalization capabilities of fully-connected neural networks trained in the context of time series forecasting. Time series do not satisfy the typical assumption in statistical learning theory of the data being i.i.d. samples from some data-generating distribution. We use the input and weight Hessians, that is the smoothness of the learned function with respect to the input and the width of the minimum in weight space, to quantify a network's ability to generalize to unseen data. While such generalization metrics have been studied extensively in the i.i.d. setting of for example image recognition, here we empirically validate their use in the task of time series forecasting. Furthermore we discuss how one can control the generalization capability of the network by means of the training process using the learning rate, batch size and the number of training iterations as controls. Using these hyperparameters one can efficiently control the complexity of the output function without imposing explicit constraints.

Original languageEnglish
Article number101020
JournalJournal of Computational Science
Volume36
DOIs
Publication statusPublished - Sept 2019
Externally publishedYes

Bibliographical note

Publisher Copyright:
© 2019 Elsevier B.V.

Keywords

  • Deep learning
  • Forecasting
  • Generalization
  • Neural networks
  • Time series

Fingerprint

Dive into the research topics of 'Generalization in fully-connected neural networks for time series forecasting'. Together they form a unique fingerprint.

Cite this