activity
20142021
most citedOn Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima

571 citations · 1.8k across the 19 of their papers we have counts for

collaborators
Showing cs.LGShow all

8 papers · 1 filter

cs.LG2021★ 2 cited

Differentiable NAS Framework and Application to Ads CTR Prediction

Ravi Krishna, Aravind Kalaiah, Bichen Wu +4

Neural architecture search (NAS) methods aim to automatically find the optimal deep neural network (DNN) architecture as measured by a given objective function, typically some comb…

cs.LG2021

Low-Precision Hardware Architectures Meet Recommendation Model Inference at Scale

Zhaoxia, Deng, Jongsoo Park +17

Tremendous success of machine learning (ML) and the unabated growth in ML model complexity motivated many ML-specific designs in both CPU and accelerator architectures to speed up…

cs.LG2021★ 20 cited

FBGEMM: Enabling High-Performance Low-Precision Deep Learning Inference

Daya Khudia, Jianyu Huang, Protonu Basu +4

Deep learning models typically use single-precision (FP32) floating point data types for representing activations and weights, but a slew of recent research work has shown that com…

cs.LG2019★ 66 cited

A Study of BFLOAT16 for Deep Learning Training

Dhiraj Kalamkar, Dheevatsa Mudigere, Naveen Mellempudi +16

This paper presents the first comprehensive empirical study demonstrating the efficacy of the Brain Floating Point (BFLOAT16) half-precision format for Deep Learning training acros…

cs.LG2018★ 81 cited

Deep Learning Inference in Facebook Data Centers: Characterization, Performance Optimizations and Hardware Implications

Jongsoo Park, Maxim Naumov, Protonu Basu +25

The application of deep learning techniques resulted in remarkable improvement of machine learning models. In this paper provides detailed characterizations of deep learning models…

cs.LG2018★ 41 cited

Bandana: Using Non-volatile Memory for Storing Deep Learning Models

Assaf Eisenman, Maxim Naumov, Darryl Gardner +5

Typical large-scale recommender systems use deep learning models that are stored on a large amount of DRAM. These models often rely on embeddings, which consume most of the require…

cs.LG2016★ 571 cited

On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima

Nitish Shirish Keskar, Dheevatsa Mudigere, Jorge Nocedal +2

The stochastic gradient descent (SGD) method and its variants are algorithms of choice for many Deep Learning tasks. These methods operate in a small-batch regime wherein a fractio…

cs.LG2016★ 6 cited

Distributed Hessian-Free Optimization for Deep Neural Network

Xi He, Dheevatsa Mudigere, Mikhail Smelyanskiy +1

Training deep neural network is a high dimensional and a highly non-convex optimization problem. Stochastic gradient descent (SGD) algorithm and it's variations are the current sta…