HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

A Theoretically Grounded Application of Dropout in Recurrent Neural Networks

Yarin Gal Zoubin Ghahramani

A Theoretically Grounded Application of Dropout in Recurrent Neural Networks

Abstract

Recurrent neural networks (RNNs) stand at the forefront of many recent developments in deep learning. Yet a major difficulty with these models is their tendency to overfit, with dropout shown to fail when applied to recurrent layers. Recent results at the intersection of Bayesian modelling and deep learning offer a Bayesian interpretation of common deep learning techniques such as dropout. This grounding of dropout in approximate Bayesian inference suggests an extension of the theoretical results, offering insights into the use of dropout with RNN models. We apply this new variational inference based dropout technique in LSTM and GRU models, assessing it on language modelling and sentiment analysis tasks. The new approach outperforms existing techniques, and to the best of our knowledge improves on the single model state-of-the-art in language modelling with the Penn Treebank (73.4 test perplexity). This extends our arsenal of variational tools in deep learning.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
language-modelling-on-penn-treebank-wordGal & Ghahramani (2016) - Variational LSTM (medium)
Test perplexity: 79.7
Validation perplexity: 81.9
language-modelling-on-penn-treebank-wordGal & Ghahramani (2016) - Variational LSTM (large)
Test perplexity: 75.2
Validation perplexity: 77.9

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
A Theoretically Grounded Application of Dropout in Recurrent Neural Networks | Papers | HyperAI