← Search

Niv Cohen

13 accepted papers

2025

Deep Unfolding of Full Waveform Inversion for Quantitative Ultrasound Imaging

ICASSP 2025accepted

This paper introduces a deep unfolding-based approach for Full Waveform Inversion (FWI) in quantitative ultrasound imaging. Our technique leverages trained deep neural networks to perform an optimized gradient step that achieves superior results and significantly reduces the number of iterations req…

Cited by 0SourceScholar
2025

Hidden in the Noise: Two-Stage Robust Watermarking for Images

ICLR 2025poster

As the quality of image generators continues to improve, deepfakes become a topic of considerable societal debate. Image watermarking allows responsible model owners to detect and label their AI-generated content, which can mitigate the harm. Yet, current state-of-the-art methods in image watermarki…

2025

When Are Concepts Erased From Diffusion Models?

NeurIPS 2025poster

In concept erasure, a model is modified to selectively prevent it from generating a target concept. Despite the rapid development of new methods, it remains unclear how thoroughly these approaches remove the target concept from the model. We begin by proposing two conceptual models for the erasure m…

Cited by 0SourcecodeScholar
2024

Circumventing Concept Erasure Methods For Text-To-Image Generative Models

ICLR 2024poster

Text-to-image generative models can produce photo-realistic images for an extremely broad range of concepts, and their usage has proliferated widely among the general public. On the flip side, these models have numerous drawbacks, including their potential to generate images featuring sexually expli…

2024

Dataset and Lessons Learned from the 2024 SaTML LLM Capture-the-Flag Competition

NeurIPS 2024spotlight

Large language model systems face significant security risks from maliciously crafted messages that aim to overwrite the system's original instructions or leak private data. To study this problem, we organized a capture-the-flag competition at IEEE SaTML 2024, where the flag is a secret string in th…

2024

SELECT: A Large-Scale Benchmark of Data Curation Strategies for Image Classification

NeurIPS 2024poster

Data curation is the problem of how to collect and organize samples into a dataset that supports efficient learning. Despite the centrality of the task, little work has been devoted towards a large-scale, systematic comparison of various curation methods. In this work, we take steps towards a formal…

2024

TuneTables: Context Optimization for Scalable Prior-Data Fitted Networks

NeurIPS 2024poster

While tabular classification has traditionally relied on from-scratch training, a recent breakthrough called prior-data fitted networks (PFNs) challenges this approach. Similar to large language models, PFNs make use of pretraining and in-context learning to achieve strong performance on new tasks i…

Cited by 26SourcePDFScholar
2023

Red PANDA: Disambiguating Image Anomaly Detection by Removing Nuisance Factors

ICLR 2023poster

Anomaly detection methods strive to discover patterns that differ from the norm in a meaningful way. This goal is ambiguous as different human operators may find different attributes meaningful. An image differing from the norm by an attribute such as pose may be considered anomalous by some operato…

Cited by 4SourcePDFScholar
2022

"“This Is My Unicorn, Fluffy”: Personalizing Frozen Vision-Language Representations"

ECCV 2022poster

"Large Vision & Language models pretrained on web-scale data provide representations that are invaluable for numerous V&L problems. However, it is unclear how they can be extended to reason about user-specific visual concepts in unstructured language. This problem arises in multiple domains, from pe…

Cited by 94SourcePDFScholar
2021

An Image is Worth More Than a Thousand Words: Towards Disentanglement in The Wild

NeurIPS 2021poster

Unsupervised disentanglement has been shown to be theoretically impossible without inductive biases on the models and the data. As an alternative approach, recent methods rely on limited supervision to disentangle the factors of variation and allow their identifiability. While annotating the true ge…

2021

PANDA: Adapting Pretrained Features for Anomaly Detection and Segmentation

CVPR 2021poster

Anomaly detection methods require high-quality features. In recent years, the anomaly detection community has attempted to obtain better features using advances in deep self-supervised feature learning. Surprisingly, a very promising direction, using pre-trained deep features, has been mostly overlo…

Cited by 336PDFcodeScholar