← Search

Keivan Rezaei

10 accepted papers

2026

Revisiting the Past: Data Unlearning with Model State History

ICLR 2026poster

Large language models are trained on massive corpora of web data, which may include private data, copyrighted material, factually inaccurate data, or data that degrades model performance. Eliminating the influence of such problematic datapoints on a model through complete retraining---by repeatedly…

Cited by 0SourcecodeScholar
2025

Localizing Knowledge in Diffusion Transformers

NeurIPS 2025poster

Understanding how knowledge is distributed across the layers of generative models is crucial for improving interpretability, controllability, and adaptation. While prior work has explored knowledge localization in UNet-based architectures, Diffusion Transformer (DiT)-based models remain underexplore…

Cited by 0SourceScholar
2025

RePanda: Pandas-powered Tabular Verification and Reasoning

ACL 2025long

Fact-checking tabular data is essential for ensuring the accuracy of structured information in domains such as journalism, finance, and scientific research. However, existing methods often rely on black-box models with opaque reasoning. We introduce RePanda, a structured fact verification approach t…

2024

Ad Auctions for LLMs via Retrieval Augmented Generation

NeurIPS 2024poster

In the field of computational advertising, the integration of ads into the outputs of large language models (LLMs) presents an opportunity to support these services without compromising content integrity. This paper introduces novel auction mechanisms for ad allocation and pricing within the textual…

Cited by 4SourcePDFScholar
2024

On Mechanistic Knowledge Localization in Text-to-Image Generative Models

ICML 2024poster

Identifying layers within text-to-image models which control visual attributes can facilitate efficient model editing through closed-form updates. Recent work, leveraging causal tracing show that early Stable-Diffusion variants confine knowledge primarily to the first layer of the CLIP text-encoder,…

2024

PRIME: Prioritizing Interpretability in Failure Mode Extraction

ICLR 2024poster

In this work, we study the challenge of providing human-understandable descriptions for failure modes in trained image classification models. Existing works address this problem by first identifying clusters (or directions) of incorrectly classified samples in a latent space and then aiming to provi…

Cited by 4SourcePDFScholar
2024

Robustness of AI-Image Detectors: Fundamental Limits and Practical Attacks

ICLR 2024poster

In light of recent advancements in generative AI models, it has become essential to distinguish genuine content from AI-generated one to prevent the malicious usage of fake materials as authentic ones and vice versa. Various techniques have been introduced for identifying AI-generated images, with w…

2023

Run-off Election: Improved Provable Defense against Data Poisoning Attacks

ICML 2023poster

In data poisoning attacks, an adversary tries to change a model's prediction by adding, modifying, or removing samples in the training data. Recently, *ensemble-based* approaches for obtaining *provable* defenses against data poisoning have been proposed where predictions are done by taking a majori…

Cited by 17SourcePDFScholar