• Home
  • Search
  • VDNN: virtualized deep neural networks for scalable, memory-efficient neural network design
  • Cite Icon210
  • https://doi.org/10.5555/3195638.3195660Copy DOI Icon

VDNN: virtualized deep neural networks for scalable, memory-efficient neural network design

  • Oct 15, 2016
  • Minsoo Rhu +4 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

The most widely used machine learning frameworks require users to carefully tune their memory usage so that the deep neural network (DNN) fits into the DRAM capacity of a GPU. This restriction hampers a researcher's flexibility to study different machine learning algorithms, forcing them to either use a less desirable network architecture or parallelize the processing across multiple GPUs. We propose a runtime memory manager that virtualizes the memory usage of DNNs such that both GPU and CPU memory can simultaneously be utilized for training larger DNNs. Our virtualized DNN (vDNN) reduces the average GPU memory usage of AlexNet by up to 89%, OverFeat by 91%, and GoogLeNet by 95%, a significant reduction in memory requirements of DNNs. Similar experiments on VGG-16, one of the deepest and memory hungry DNNs to date, demonstrate the memory-efficiency of our proposal. vDNN enables VGG-16 with batch size 256 (requiring 28 GB of memory) to be trained on a single NVIDIA Titan X GPU card containing 12 GB of memory, with 18% performance loss compared to a hypothetical, oracular GPU with enough memory to hold the entire DNN.

Similar Papers
  • Research Article
  • Citations8

Noise-robust bioimpedance approach for cardiac output measurement

  • Jul 01, 2019
  • Physiological Measurement
  • Ethan K Murphy +4
  • Research Article

Water scarcity prediction for global region using machine learning

  • Jan 01, 2020
  • International Journal of Water
  • Shubra Jain +2
  • Research Article
  • Citations21

MoDNN: Memory Optimal Deep Neural Network Training on Graphics Processing Units

  • Mar 01, 2019
  • IEEE Transactions on Parallel and Distributed Systems
  • Xiaoming Chen +3
  • Research Article
  • Citations5

Application of Machine Learning to Interpret Steady-State Drainage Relative Permeability Experiments

  • Mar 22, 2023
  • SPE Reservoir Evaluation & Engineering
  • Eric Sonny Mathew +4
  • Research Article
  • Citations1

Machine Learning Approaches for Automated Diagnosis of Cardiovascular Diseases: A Review of Electrocardiogram Data Applications.

  • Sep 12, 2024
  • Cardiology in review
  • Abdelhakim Elmassaoudi +2
  • Research Article
  • Citations2

P.115 Diagnostic performance of machine learning based MR algorithm vs conventional MR images for predicting the likelihood of brain tumors

  • Jun 01, 2022
  • Canadian Journal of Neurological Sciences / Journal Canadien des Sciences Neurologiques
  • S Chatterjee +5
  • Research Article

Throughput optimization of optical-NOMA system using ML algorithms for diverse order QAM transmission schemes

  • Jun 26, 2025
  • Journal of Optical Communications
  • Rajneesh Pareek +2
  • Research Article
  • Citations87

Prediction of Future Terrorist Activities Using Deep Neural Networks

  • Apr 22, 2020
  • Complexity
  • M Irfan Uddin +7
  • Research Article
  • Citations4

Hybrid generative regression-based deep intelligence to predict the risk of chronic disease

  • Nov 11, 2021
  • International Journal of Intelligent Computing and Cybernetics
  • Sandeep Kumar Hegde +1
  • Dissertation
  • Citations1

Acoustic feature-based sentiment analysis of call center data

  • Dec 01, 2017
  • Zeshan Peng
  • Research Article
  • Citations3

Explainable AI ‐Driven Firewall Evaluation: Empowering Cybersecurity Decision‐Making for Optimal Network Defense

  • Apr 04, 2025
  • SECURITY AND PRIVACY
  • Zaheen Fatima +4
  • PDF
  • Research Article
  • Citations12

Maize Kernel Abortion Recognition and Classification Using Binary Classification Machine Learning Algorithms and Deep Convolutional Neural Networks

  • Aug 31, 2020
  • AI
  • Lovemore Chipindu +4
  • Conference Article
  • Citations62

A Review of In-Memory Computing Architectures for Machine Learning Applications

  • Sep 07, 2020
  • Sathwika Bavikadi +4
  • Book Chapter
  • Citations43

The Technological Elements of Artificial Intelligence

  • Jan 01, 2019
  • Matt Taddy
  • Book Chapter
  • Citations1

Handwritten Digit Recognition Using Very Deep Convolutional Neural Network

  • Jan 01, 2022
  • M Dhilsath Fathima +2
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.