← Search

Pratyush Maini

18 accepted papers

2025

Memorization Sinks: Isolating Memorization during LLM Training

ICML 2025poster

Large language models are susceptible to memorizing repeated sequences, posing privacy and copyright concerns. A popular mitigation strategy is to remove memorized information from specific neurons post-hoc. However, such approaches have shown limited success so far. In a controlled setting, we show…

2025

OpenUnlearning: Accelerating LLM Unlearning via Unified Benchmarking of Methods and Metrics

NeurIPS 2025poster

Robust unlearning is crucial for safely deploying large language models (LLMs) in environments where data privacy, model safety, and regulatory compliance must be ensured. Yet the task is inherently challenging, partly due to difficulties in reliably measuring whether unlearning has truly occurred.…

Cited by 0SourcecodeScholar
2025

STAMP Your Content: Proving Dataset Membership via Watermarked Rephrasings

ICML 2025poster

Given how large parts of publicly available text are crawled to pretrain large language models (LLMs), data creators increasingly worry about the inclusion of their proprietary data for model training without attribution or licensing. Their concerns are also shared by benchmark curators whose test-s…

2025

Safety Pretraining: Toward the Next Generation of Safe AI

NeurIPS 2025poster

As large language models (LLMs) are increasingly deployed in high-stakes settings, the risk of generating harmful or toxic content remains a central challenge. Post-hoc alignment methods are brittle: once unsafe patterns are learned during pretraining, they are hard to remove. In this work, we prese…

Cited by 0SourceScholar
2025

Unlocking Post-hoc Dataset Inference with Synthetic Data

ICML 2025poster

The remarkable capabilities of Large Language Models (LLMs) can be mainly attributed to their massive training datasets, which are often scraped from the internet without respecting data owners’ intellectual property rights. Dataset Inference (DI) offers a potential remedy by identifying whether a s…

2024

LLM Dataset Inference: Did you train on my dataset?

NeurIPS 2024poster

The proliferation of large language models (LLMs) in the real world has come with a rise in copyright cases against companies for training their models on unlicensed data from the internet. Recent works have presented methods to identify if individual text sequences were members of the model's train…

2024

Rephrasing the Web: A Recipe for Compute and Data-Efficient Language Modeling

ACL 2024long

Large language models are trained on massive scrapes of the web, which are often unstructured, noisy, and poorly phrased. Current scaling laws show that learning from such data requires an abundance of both compute and data, which grows with the size of the model being trained. This is infeasible bo…

Cited by 57SourcePDFScholar
2024

Rethinking LLM Memorization through the Lens of Adversarial Compression

NeurIPS 2024poster

Large language models (LLMs) trained on web-scale datasets raise substantial concerns regarding permissible data usage. One major question is whether these models "memorize" all their training data or they integrate many data sources in some way more akin to how a human would learn and synthesize i…

Cited by 40SourcePDFScholar
2024

Scaling Laws for Data Filtering-- Data Curation cannot be Compute Agnostic

CVPR 2024poster

Vision-language models (VLMs) are trained for thousands of GPU hours on carefully selected subsets of massive web scrapes. For instance the LAION public dataset retained only about 10 percent of the total crawled data. In recent times data curation has gained prominence with several works developing…

2024

T-MARS: Improving Visual Representations by Circumventing Text Feature Learning

ICLR 2024poster

Large web-crawled multimodal datasets have powered a slew of new methods for learning general-purpose visual representations, advancing the state of the art in computer vision and revolutionizing zero- and few-shot recognition. One crucial decision facing practitioners is how, if at all, to curate t…

2024

Understanding Hallucinations in Diffusion Models through Mode Interpolation

NeurIPS 2024poster

Colloquially speaking, image generation models based upon diffusion processes are frequently said to exhibit ''hallucinations'' samples that could never occur in the training data. But where do such hallucinations come from? In this paper, we study a particular failure mode in diffusion models, whi…

2023

Can Neural Network Memorization Be Localized?

ICML 2023poster

Recent efforts at explaining the interplay of memorization and generalization in deep overparametrized networks have posited that neural networks *memorize* ``hard'' examples in the final few layers of the model. Memorization refers to the ability to correctly predict on *atypical* examples of the t…

2023

Model-tuning Via Prompts Makes NLP Models Adversarially Robust

EMNLP 2023long main

In recent years, NLP practitioners have converged on the following practice: (i) import an off-the-shelf pretrained (masked) language model; (ii) append a multilayer perceptron atop the CLS token's hidden representation (with randomly initialized weights); and (iii) fine-tune the entire model on a d…

Cited by 0SourcecodeScholar
2022

Characterizing Datapoints via Second-Split Forgetting

NeurIPS 2022accept

Researchers investigating example hardness have increasingly focused on the dynamics by which neural networks learn and forget examples throughout training. Popular metrics derived from these dynamics include (i) the epoch at which examples are first correctly classified; (ii) the number of times th…

2022

Perturbation type categorization for multiple adversarial perturbation robustness

UAI 2022poster

Recent works in adversarial robustness have proposed defenses to improve the robustness of a single model against the union of multiple perturbation types. However, these methods still suffer significant trade-offs compared to the ones specifically trained to be robust against a single perturbation…

2021

Dataset Inference: Ownership Resolution in Machine Learning

ICLR 2021spotlight

With increasingly more data and computation involved in their training, machine learning models constitute valuable intellectual property. This has spurred interest in model stealing, which is made more practical by advances in learning with partial, little, or no supervision. Existing defenses foc…

2020

Adversarial Robustness Against the Union of Multiple Perturbation Models

ICML 2020poster

Owing to the susceptibility of deep learning systems to adversarial attacks, there has been a great deal of work in developing (both empirically and certifiably) robust classifiers. While most work has defended against a single type of attack, recent work has looked at defending against multiple per…