Probing the Structure and Functional Properties of the Dropout-Induced Correlated Variability in Convolutional Neural Networks

Xu Pan, Ruben Coen-Cagli, Odelia Schwartz

Research output: Contribution to journalArticlepeer-review

Abstract

Computational neuroscience studies have shown that the structure of neural variability to an unchanged stimulus affects the amount of information encoded. Some artificial deep neural networks, such as those with Monte Carlo dropout layers, also have variable responses when the input is fixed. However, the structure of the trial-by-trial neural covariance in neural networks with dropout has not been studied, and its role in de-coding accuracy is unknown. We studied the above questions in a con-volutional neural network model with dropout in both the training and testing phases. We found that trial-by-trial correlation between neurons (i.e., noise correlation) is positive and low dimensional. Neurons that are close in a feature map have larger noise correlation. These properties are surprisingly similar to the findings in the visual cortex. We further ana-lyzed the alignment of the main axes of the covariance matrix. We found that different images share a common trial-by-trial noise covariance sub-space, and they are aligned with the global signal covariance. This evi-dence that the noise covariance is aligned with signal covariance suggests that noise covariance in dropout neural networks reduces network accu-racy, which we further verified directly with a trial-shuffling procedure commonly used in neuroscience. These findings highlight a previously overlooked aspect of dropout layers that can affect network performance. Such dropout networks could also potentially be a computational model of neural variability.

Original languageEnglish (US)
Pages (from-to)621-644
Number of pages24
JournalNeural computation
Volume36
Issue number4
DOIs
StatePublished - Apr 2024

ASJC Scopus subject areas

  • Arts and Humanities (miscellaneous)
  • Cognitive Neuroscience

Fingerprint

Dive into the research topics of 'Probing the Structure and Functional Properties of the Dropout-Induced Correlated Variability in Convolutional Neural Networks'. Together they form a unique fingerprint.

Cite this