most citedRevisiting Neural Retrieval on Accelerators

19 citations · 39 across the 5 of their papers we have counts for

collaborators

5 papers

cs.LG2023★ 19 cited

Revisiting Neural Retrieval on Accelerators

Jiaqi Zhai, Zhaojie Gong, Yueming Wang +4

Retrieval finds a small number of relevant candidates from a large corpus for information retrieval and recommendation applications. A key component of retrieval is to model (user,…

cs.IR2023

MTrainS: Improving DLRM training efficiency using heterogeneous memories

Hiwot Tadese Kassa, Paul Johnson, Jason Akers +7

Recommendation models are very large, requiring terabytes (TB) of memory during training. In pursuit of better quality, the model size and complexity grow over time, which requires…

cs.CV2021★ 4 cited

Cross-Region Domain Adaptation for Class-level Alignment

Zhijie Wang, Xing Liu, Masanori Suganuma +1

Semantic segmentation requires a lot of training data, which necessitates costly annotation. There have been many studies on unsupervised domain adaptation (UDA) from one domain to…

cs.DC2021★ 3 cited

Software-Hardware Co-design for Fast and Scalable Training of Deep Learning Recommendation Models

Dheevatsa Mudigere, Yuchen Hao, Jianyu Huang +50

Deep learning recommendation models (DLRMs) are used across many business-critical services at Facebook and are the single largest AI application in terms of infrastructure demand…

cs.LG2021★ 13 cited

TT-Rec: Tensor Train Compression for Deep Learning Recommendation Models

Chunxing Yin, Bilge Acun, Xing Liu +1

The memory capacity of embedding tables in deep learning recommendation models (DLRMs) is increasing dramatically from tens of GBs to TBs across the industry. Given the fast growth…