← Search

Prajwal Singhania

3 accepted papers

2024

Be like a Goldfish, Don't Memorize! Mitigating Memorization in Generative LLMs

NeurIPS 2024poster

Large language models can memorize and repeat their training data, causing privacy and copyright risks. To mitigate memorization, we introduce a subtle modification to the next-token training objective that we call the goldfish loss. During training, a randomly sampled subsets of tokens are excluded…

2024

Loki: Low-rank Keys for Efficient Sparse Attention

NeurIPS 2024poster

Inference on large language models (LLMs) can be expensive in terms of the compute and memory costs involved, especially when long sequence lengths are used. In particular, the self-attention mechanism used in LLM inference contributes significantly to these costs, which has sparked an interest in a…

2021

Alternating Direction Method of Multipliers for Quantization

AISTATS 2021poster

Quantization of the parameters of machine learning models, such as deep neural networks, requires solving constrained optimization problems, where the constraint set is formed by the Cartesian product of many simple discrete sets. For such optimization problems, we study the performance of the Alter…

Cited by 12SourcePDFScholar