activity
20182021
most citedLarge Batch Optimization for Deep Learning: Training BERT in 76 minutes

162 citations · 423 across the 11 of their papers we have counts for

collaborators

11 papers

cs.CV2021★ 12 cited

A Simple Single-Scale Vision Transformer for Object Localization and Instance Segmentation

Wuyang Chen, Xianzhi Du, Fan Yang +8

This work presents a simple vision transformer design as a strong baseline for object localization and instance segmentation tasks. Transformers recently demonstrate competitive pe…

cs.LG2021★ 3 cited

Speeding up Deep Model Training by Sharing Weights and Then Unsharing

Shuo Yang, Le Hou, Xiaodan Song +2

We propose a simple and efficient approach for training the BERT model. Our approach exploits the special structure of BERT that contains a stack of repeated modules (i.e., transfo…

cs.NI2021★ 48 cited

A new communication paradigm: from bit accuracy to semantic fidelity

Guangming Shi, Dahua Gao, Xiaodan Song +5

Wireless communication has achieved great success in the past several decades. The challenge is of improving bandwidth with limited spectrum and power consumption, which however ha…

cs.CV2020★ 2 cited

Efficient Scale-Permuted Backbone with Learned Resource Distribution

Xianzhi Du, Tsung-Yi Lin, Pengchong Jin +4

Recently, SpineNet has demonstrated promising results on object detection and image classification over ResNet model. However, it is unclear if the improvement adds up when combini…

cs.LG2020★ 5 cited

Go Wide, Then Narrow: Efficient Training of Deep Thin Networks

Denny Zhou, Mao Ye, Chen Chen +6

For deploying a deep learning model into production, it needs to be both accurate and compact to meet the latency and memory constraints. This usually results in a network that is…

cs.CL2020★ 94 cited

MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devices

Zhiqing Sun, Hongkun Yu, Xiaodan Song +3

Natural Language Processing (NLP) has recently achieved great success by using huge pre-trained models with hundreds of millions of parameters. However, these models suffer from he…

cs.CV2020★ 39 cited

BigNAS: Scaling Up Neural Architecture Search with Big Single-Stage Models

Jiahui Yu, Pengchong Jin, Hanxiao Liu +7

Neural architecture search (NAS) has shown promising results discovering models that are both accurate and fast. For NAS, training a one-shot model has become a popular strategy to…

cs.CV2019★ 21 cited

SpineNet: Learning Scale-Permuted Backbone for Recognition and Localization

Xianzhi Du, Tsung-Yi Lin, Pengchong Jin +5

Convolutional neural networks typically encode an input image into a series of intermediate features with decreasing resolutions. While this structure is suited to classification t…

eess.IV2019★ 29 cited

High Resolution Medical Image Analysis with Spatial Partitioning

Le Hou, Youlong Cheng, Noam Shazeer +6

Medical images such as 3D computerized tomography (CT) scans and pathology images, have hundreds of millions or billions of voxels/pixels. It is infeasible to train CNN models dire…

cs.LG2019★ 162 cited

Large Batch Optimization for Deep Learning: Training BERT in 76 minutes

Yang You, Jing Li, Sashank Reddi +7

Training large deep neural networks on massive datasets is computationally very challenging. There has been recent surge in interest in using large batch stochastic optimization me…

cs.MM2018★ 8 cited

A practical convolutional neural network as loop filter for intra frame

Xiaodan Song, Jiabao Yao, Lulu Zhou +4

Loop filters are used in video coding to remove artifacts or improve performance. Recent advances in deploying convolutional neural network (CNN) to replace traditional loop filter…