← Search

Vikash Sehwag

21 accepted papers

2026

PropensityBench: Evaluating Latent Safety Risks in Large Language Models via an Agentic Approach

ICLR 2026poster

Recent advances in Large Language Models (LLMs) have sparked concerns over their potential to acquire and misuse dangerous capabilities, posing frontier risks to society. Current safety evaluations primarily test for what a model *can* do---its capabilities---without assessing what it *would* do if…

Cited by 0SourcecodeScholar
2025

Adapting to Evolving Adversaries with Regularized Continual Robust Training

ICML 2025poster

Robust training methods typically defend against specific attack types, such as $\ell_p$ attacks with fixed budgets, and rarely account for the fact that defenders may encounter new attacks over time. A natural solution is to adapt the defended model to new adversaries as they arise via fine-tuning…

2025

Argus: A Compact and Versatile Foundation Model for Vision

CVPR 2025poster

While existing vision and multi-modal foundation models can handle multiple computer vision tasks, they often suffer from significant limitations, including huge demand for data and computational resources during training and inconsistent performance across vision tasks at deployment time. To addres…

Cited by 0SourcePDFScholar
2025

CO-SPY: Combining Semantic and Pixel Features to Detect Synthetic Images by AI

CVPR 2025poster

With the rapid advancement of generative AI, it is now possible to synthesize high-quality images in a few seconds. Despite the power of these technologies, they raise significant concerns regarding misuse. Current efforts to distinguish between real and AI-generated images may lack generalization,…

2025

FineGRAIN: Evaluating Failure Modes of Text-to-Image Models with Vision Language Model Judges

NeurIPS 2025spotlight

Text-to-image (T2I) models are capable of generating visually impressive images, yet they often fail to accurately capture specific attributes in user prompts, such as the correct number of objects with the specified colors. The diversity of such errors underscores the need for a hierarchical evalua…

Cited by 0SourceScholar
2025

How to Evaluate and Mitigate IP Infringement in Visual Generative AI?

ICML 2025poster

The popularity of visual generative AI models like DALL-E 3, Stable Diffusion XL, Stable Video Diffusion, and Sora has been increasing. Through extensive evaluation, we discovered that the state-of-the-art visual generative models can generate content that bears a striking resemblance to characters…

Cited by 0SourcePDFScholar
2025

Stretching Each Dollar: Diffusion Training from Scratch on a Micro-Budget

CVPR 2025poster

As scaling laws in generative AI push performance, they simultaneously concentrate the development of these models among actors with large computational resources. With a focus on text-to-image (T2I) generative models, we aim to unlock this bottleneck by demonstrating very low-cost training of large…

2024

A New Linear Scaling Rule for Private Adaptive Hyperparameter Optimization

ICML 2024poster

An open problem in differentially private deep learning is hyperparameter optimization (HPO). DP-SGD introduces new hyperparameters and complicates existing ones, forcing researchers to painstakingly tune hyperparameters with hundreds of trials, which in turn makes it impossible to account for the p…

Cited by 5SourcePDFScholar
2024

Finding a needle in a haystack: A Black-Box Approach to Invisible Watermark Detection

ECCV 2024poster

"In this paper, we propose WaterMark Detector (), the first invisible watermark detection method under a black-box and annotation-free setting. is capable of detecting arbitrary watermarks within a given detection dataset using a clean non-watermarked dataset as a reference, without relying on speci…

Cited by 4SourcePDFScholar
2024

How to Trace Latent Generative Model Generated Images without Artificial Watermark?

ICML 2024poster

Latent generative models (e.g., Stable Diffusion) have become more and more popular, but concerns have arisen regarding potential misuse related to images generated by these models. It is, therefore, necessary to analyze the origin of images by inferring if a particular image was generated by a spec…

2024

JailbreakBench: An Open Robustness Benchmark for Jailbreaking Large Language Models

NeurIPS 2024poster

Jailbreak attacks cause large language models (LLMs) to generate harmful, unethical, or otherwise objectionable content. Evaluating these attacks presents a number of challenges, which the current collection of benchmarks and evaluation techniques do not adequately address. First, there is no clear…

2023

Differentially Private Image Classification by Learning Priors from Random Processes

NeurIPS 2023spotlight

In privacy-preserving machine learning, differentially private stochastic gradient descent (DP-SGD) performs worse than SGD due to per-sample gradient clipping and noise addition. A recent focus in private learning research is improving the performance of DP-SGD on private data by incorporating prio…

2023

MultiRobustBench: Benchmarking Robustness Against Multiple Attacks

ICML 2023poster

The bulk of existing research in defending against adversarial examples focuses on defending against a single (typically bounded $\ell_p$-norm) attack, but for a practical setting, machine learning (ML) models should be robust to a wide variety of attacks. In this paper, we present the first unified…

2023

Uncovering Adversarial Risks of Test-Time Adaptation

ICML 2023poster

Recently, test-time adaptation (TTA) has been proposed as a promising solution for addressing distribution shifts. It allows a base model to adapt to an unforeseen distribution during inference by leveraging the information from the batch of (unlabeled) test data. However, we uncover a novel securit…

Cited by 11SourcePDFScholar
2022

Generating High Fidelity Data From Low-Density Regions Using Diffusion Models

CVPR 2022poster

Our work focuses on addressing sample deficiency from low-density regions of data manifold in common image datasets. We leverage diffusion process based generative models to synthesize novel images from low-density regions. We observe that uniform sampling from diffusion models predominantly samples…

Cited by 71PDFcodeScholar
2022

Robust Learning Meets Generative Models: Can Proxy Distributions Improve Adversarial Robustness?

ICLR 2022poster

While additional training data improves the robustness of deep neural networks against adversarial examples, it presents the challenge of curating a large number of specific real-world samples. We circumvent this challenge by using additional data from proxy distributions learned by advanced genera…

2022

Understanding Robust Learning through the Lens of Representation Similarities

NeurIPS 2022accept

Representation learning, \textit{i.e.} the generation of representations useful for downstream applications, is a task of fundamental importance that underlies much of the success of deep neural networks (DNNs). Recently, \emph{robustness to adversarial examples} has emerged as a desirable property…

2021

Lower Bounds on Cross-Entropy Loss in the Presence of Test-time Adversaries

ICML 2021spotlight

Understanding the fundamental limits of robust supervised learning has emerged as a problem of immense interest, from both practical and theoretical standpoints. In particular, it is critical to determine classifier-agnostic bounds on the training loss to establish when learning is possible. In this…

2021

RobustBench: a standardized adversarial robustness benchmark

NeurIPS 2021poster

As a research community, we are still lacking a systematic understanding of the progress on adversarial robustness which often makes it hard to identify the most promising ideas in training robust models. A key challenge in benchmarking robustness is that its evaluation is often error-prone leading…

Cited by 843SourcecodeScholar
2021

SSD: A Unified Framework for Self-Supervised Outlier Detection

ICLR 2021poster

We ask the following question: what training information is required to design an effective outlier/out-of-distribution (OOD) detector, i.e., detecting samples that lie far away from training distribution? Since unlabeled data is easily accessible for many applications, the most compelling approach…

2020

HYDRA: Pruning Adversarially Robust Neural Networks

NeurIPS 2020poster

In safety-critical but computationally resource-constrained applications, deep learning faces two key challenges: lack of robustness against adversarial attacks and large neural network size (often millions of parameters). While the research community has extensively explored the use of robust train…