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

306 citations · 564 across the 35 of their papers we have counts for

collaborators
Showing cs.LGShow all

7 papers · 1 filter

cs.LG2023

Tribe or Not? Critical Inspection of Group Differences Using TribalGram

Yongsu Ahn, Muheng Yan, Yu-Ru Lin +2

With the rise of AI and data mining techniques, group profiling and group-level analysis have been increasingly used in many domains including policy making and direct marketing. I…

cs.LG2023

TAPAS: Fast and Automatic Derivation of Tensor Parallel Strategies for Large Neural Networks

Ziji Shi, Le Jiang, Ang Wang +6

Tensor parallelism is an essential technique for distributed training of large neural networks. However, automatically determining an optimal tensor parallel strategy is challengin…

cs.LG2021★ 18 cited

M6-10T: A Sharing-Delinking Paradigm for Efficient Multi-Trillion Parameter Pretraining

Junyang Lin, An Yang, Jinze Bai +9

Recent expeditious developments in deep learning algorithms, distributed training, and even hardware design for large models have enabled training extreme-scale models, say GPT-3 a…

cs.LG2021

SINGA-Easy: An Easy-to-Use Framework for MultiModal Analysis

Naili Xing, Sai Ho Yeung, Chenghao Cai +7

Deep learning has achieved great success in a wide spectrum of multimedia applications such as image classification, natural language processing and multimodal data analysis. Recen…

cs.LG2021★ 14 cited

M6-T: Exploring Sparse Expert Models and Beyond

An Yang, Junyang Lin, Rui Men +12

Mixture-of-Experts (MoE) models can achieve promising results with outrageous large amount of parameters but constant computation cost, and thus it has become a trend in model scal…

cs.LG2019★ 5 cited

Progressive Fashion Attribute Extraction

Sandeep Singh Adhikari, Sukhneer Singh, Anoop Rajagopal +1

Extracting fashion attributes from images of people wearing clothing/fashion accessories is a very hard multi-class classification problem. Most often, even catalogues of fashion d…

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…