activity
20132022
most citedHighly Scalable Deep Learning Training System with Mixed-Precision: Training ImageNet in Four Minutes

306 citations · 388 across the 10 of their papers we have counts for

collaborators
Showing cs.LGShow all

5 papers · 1 filter

cs.LG2022

Multi-modal Graph Learning for Disease Prediction

Shuai Zheng, Zhenfeng Zhu, Zhizhe Liu +4

Benefiting from the powerful expressive capability of graphs, graph-based approaches have been popularly applied to handle multi-modal medical data and achieved impressive performa…

cs.LG2021★ 1 cited

CETransformer: Casual Effect Estimation via Transformer Based Representation Learning

Zhenyu Guo, Shuai Zheng, Zhizhe Liu +2

Treatment effect estimation, which refers to the estimation of causal effects and aims to measure the strength of the causal relationship, is of great importance in many fields but…

cs.LG2021

Multi-modal Graph Learning for Disease Prediction

Shuai Zheng, Zhenfeng Zhu, Zhizhe Liu +3

Benefiting from the powerful expressive capability of graphs, graph-based approaches have achieved impressive performance in various biomedical applications. Most existing methods…

cs.LG2019★ 9 cited

Distributed Equivalent Substitution Training for Large-Scale Recommender Systems

Haidong Rong, Yangzihao Wang, Feihu Zhou +8

We present Distributed Equivalent Substitution (DES) training, a novel distributed training framework for large-scale recommender systems with dynamic sparse features. DES introduc…

cs.LG2018★ 306 cited

Highly Scalable Deep Learning Training System with Mixed-Precision: Training ImageNet in Four Minutes

Xianyan Jia, Shutao Song, Wei He +11

Synchronized stochastic gradient descent (SGD) optimizers with data parallelism are widely used in training large-scale deep neural networks. Although using larger mini-batch sizes…