← Search

Dhruv Choudhary

5 accepted papers

2025

SpinQuant: LLM Quantization with Learned Rotations

ICLR 2025poster

Post-training quantization (PTQ) techniques applied to weights, activations, and the KV cache greatly reduce memory usage, latency, and power consumption of Large Language Models (LLMs), but may lead to large quantization errors when outliers are present. Rotating activation or weight matrices helps…

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

Frequency-aware SGD for Efficient Embedding Learning with Provable Benefits

ICLR 2022poster

Embedding learning has found widespread applications in recommendation systems and natural language modeling, among other domains. To learn quality embeddings efficiently, adaptive learning rate algorithms have demonstrated superior empirical performance over SGD, largely accredited to their token-d…

Cited by 5SourcePDFScholar
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