← Search

Arun Kejariwal

4 accepted papers

2024

Layer Compression of Deep Networks with Straight Flows

AAAI 2024technical

Very deep neural networks lead to significantly better performance on various real tasks. However, it usually causes slow inference and is hard to be deployed on real-world devices. How to reduce the number of layers to save memory and to accelerate the inference is an eye-catching topic. In thi…

2022

DreamShard: Generalizable Embedding Table Placement for Recommender Systems

NeurIPS 2022accept

We study embedding table placement for distributed recommender systems, which aims to partition and place the tables on multiple hardware devices (e.g., GPUs) to balance the computation and communication costs. Although prior work has explored learning-based approaches for the device placement of co…

2022

Future gradient descent for adapting the temporal shifting data distribution in online recommendation systems

UAI 2022poster

One of the key challenges of learning an online recommendation model is the temporal domain shift, which causes the mismatch between the training and testing data distribution and hence domain generalization error. To overcome, we propose to learn a meta future gradient generator that forecasts the…

Cited by 8SourcePDFScholar
2022

Harmless Transfer Learning for Item Embeddings

NAACL 2022findings

Learning embedding layers (for classes, words, items, etc.) is a key component of lots of applications, ranging from natural language processing, recommendation systems to electronic health records, etc. However, the frequency of real-world items follows a long-tail distribution in these application…

Cited by 2SourcePDFScholar