← Search

Kilian Q. Weinberger

69 accepted papers

2026

Adaptive Moments are Surprisingly Effective for Plug-and-Play Diffusion Sampling

ICLR 2026poster

Guided diffusion sampling relies on approximating intractable likelihood scores, which introduces significant noise into the sampling dynamics. We propose using adaptive moment estimation to stabilize these noisy likelihood scores during sampling. Despite its simplicity, our approach achieves state-…

Cited by 0SourcecodeScholar
2026

Learning from Synthetic Data Improves Multi-hop Reasoning

ICLR 2026poster

Reinforcement Learning (RL) has been shown to significantly boost reasoning capabilities of large language models (LLMs) in math, coding, and multi-hop reasoning tasks. However, RL fine-tuning requires abundant high-quality verifiable data, often obtained through human-annotated datasets and LLM-as-…

Cited by 0SourcecodeScholar
2026

Pre-training Limited Memory Language Models with Internal and External Knowledge

ICLR 2026poster

Neural language models are black-boxes--both linguistic patterns and factual knowledge are distributed across billions of opaque parameters. This entangled encoding makes it difficult to reliably inspect, verify, or update specific facts. We introduce Limited Memory Language Models (LMLM), a new cla…

Cited by 0SourcecodeScholar
2026

SpeechOp: Inference-Time Task Composition for Generative Speech Processing

ICLR 2026poster

While generative Text-to-Speech (TTS) systems leverage vast "in-the-wild" data to achieve remarkable success, speech-to-speech processing tasks like enhancement face data limitations, which lead data-hungry generative approaches to distort speech content and speaker identity. To bridge this gap, we…

Cited by 0SourceScholar
2025

$Q\sharp$: Provably Optimal Distributional RL for LLM Post-Training

NeurIPS 2025poster

Reinforcement learning (RL) post-training is crucial for LLM alignment and reasoning, but existing policy-based methods, such as PPO and DPO, can fall short of fixing shortcuts inherited from pre-training. In this work, we introduce $Q\sharp$, a value-based algorithm for KL-regularized RL that guide…

Cited by 0SourcecodeScholar
2025

Graders Should Cheat: Privileged Information Enables Expert-Level Automated Evaluations

EMNLP 2025

Auto-evaluating language models (LMs), *i.e*., using a grader LM to evaluate the candidate LM, is an appealing way to accelerate the evaluation process and the cost associated with it. But this presents a paradox: how can we trust the grader LM, which is presumably weaker than the candidate LM, to a

Cited by 0SourcePDFScholar
2025

Learning 3D Perception from Others' Predictions

ICLR 2025poster

Accurate 3D object detection in real-world environments requires a huge amount of annotated data with high quality. Acquiring such data is tedious and expensive, and often needs repeated effort when a new sensor is adopted or when the detector is deployed in a new environment. We investigate a new s…

Cited by 1SourcePDFScholar
2025

Mixed Signals: A Diverse Point Cloud Dataset for Heterogeneous LiDAR V2X Collaboration

ICCV 2025poster

Vehicle-to-everything (V2X) collaborative perception has emerged as a promising solution to address the limitations of single-vehicle perception systems. However, existing V2X datasets are limited in scope, diversity, and quality. To address these gaps, we present Mixed Signals, a comprehensive V2X…

Cited by 0SourcePDFScholar
2025

On Speeding Up Language Model Evaluation

ICLR 2025poster

Developing prompt-based methods with Large Language Models (LLMs) requires making numerous decisions, which give rise to a combinatorial search problem over hyper-parameters. This exhaustive evaluation can be time-consuming and costly. In this paper, we propose an \textit{adaptive} approach to explo…

2025

PhantomWiki: On-Demand Datasets for Reasoning and Retrieval Evaluation

ICML 2025poster

High-quality benchmarks are essential for evaluating reasoning and retrieval capabilities of large language models (LLMs). However, curating datasets for this purpose is not a permanent solution as they are prone to data leakage and inflated performance results. To address these challenges, we prop…

2025

Rethinking LLM Unlearning Objectives: A Gradient Perspective and Go Beyond

ICLR 2025poster

Large language models (LLMs) should undergo rigorous audits to identify potential risks, such as copyright and privacy infringements. Once these risks emerge, timely updates are crucial to remove undesirable responses, ensuring legal and safe model usage. It has spurred recent research into LLM unle…

Cited by 2SourcePDFScholar
2025

Transfer Your Perspective: Controllable 3D Generation from Any Viewpoint in a Driving Scene

CVPR 2025poster

Self-driving cars relying solely on ego-centric perception face limitations in sensing, often failing to detect occluded, faraway objects. Collaborative autonomous driving (CAV) seems like a promising direction, but collecting data for development is non-trivial. It requires placing multiple sensor-…

Cited by 0SourcePDFScholar
2024

Augmenting Lane Perception and Topology Understanding with Standard Definition Navigation Maps

ICRA 2024poster

Autonomous driving has traditionally relied heavily on costly and labor-intensive High Definition (HD) maps, hindering scalability. In contrast, Standard Definition (SD) maps are more affordable and have worldwide coverage, offering a scalable alternative. In this work, we systematically explore the…

Cited by 34SourcecodeScholar
2024

Better Monocular 3D Detectors with LiDAR from the Past

ICRA 2024poster

Accurate 3D object detection is crucial to autonomous driving. Though LiDAR-based detectors have achieved impressive performance, the high cost of LiDAR sensors precludes their widespread adoption in affordable vehicles. Camera-based detectors are cheaper alternatives but often suffer inferior perfo…

Cited by 1SourcecodeScholar
2024

DiffuBox: Refining 3D Object Detection with Point Diffusion

NeurIPS 2024poster

Ensuring robust 3D object detection and localization is crucial for many applications in robotics and autonomous driving. Recent models, however, face difficulties in maintaining high performance when applied to domains with differing sensor setups or geographic locations, often resulting in poor lo…

2024

Don't Trust: Verify -- Grounding LLM Quantitative Reasoning with Autoformalization

ICLR 2024poster

Large language models (LLM), such as Google's Minerva and OpenAI's GPT families, are becoming increasingly capable of solving mathematical quantitative reasoning problems. However, they still make unjustified logical and computational errors in their reasoning steps and answers. In this paper, we le…

2024

Online Feature Updates Improve Online (Generalized) Label Shift Adaptation

NeurIPS 2024poster

This paper addresses the prevalent issue of label shift in an online setting with missing labels, where data distributions change over time and obtaining timely labels is challenging. While existing methods primarily focus on adjusting or updating the final layer of a pre-trained classifier, we expl…

Cited by 2SourcePDFScholar
2024

Pre-training LiDAR-based 3D Object Detectors through Colorization

ICLR 2024poster

Accurate 3D object detection and understanding for self-driving cars heavily relies on LiDAR point clouds, necessitating large amounts of labeled data to train. In this work, we introduce an innovative pre-training approach, Grounded Point Colorization (GPC), to bridge the gap between data and label…

2023

Does Label Differential Privacy Prevent Label Inference Attacks?

AISTATS 2023poster

Label differential privacy (label-DP) is a popular framework for training private ML models on datasets with public features and sensitive private labels. Despite its rigorous privacy guarantee, it has been observed that in practice label-DP does not preclude label inference attacks (LIAs): Models t…

2023

Image-to-Image Translation for Autonomous Driving from Coarsely-Aligned Image Pairs

ICRA 2023poster

A self-driving car must be able to reliably handle adverse weather conditions (e.g., snowy) to operate safely. In this paper, we investigate the idea of turning sensor inputs (i.e., images) captured in an adverse condition into a benign one (i.e., sunny), upon which the downstream tasks (e.g., seman…

Cited by 18SourceScholar
2023

IncDSI: Incrementally Updatable Document Retrieval

ICML 2023poster

Differentiable Search Index is a recently proposed paradigm for document retrieval, that encodes information about a corpus of documents within the parameters of a neural network and directly maps queries to corresponding documents. These models have achieved state-of-the-art performances for docume…

2023

Latent Diffusion for Language Generation

NeurIPS 2023poster

Diffusion models have achieved great success in modeling continuous data modalities such as images, audio, and video, but have seen limited use in discrete domains such as language. Recent attempts to adapt diffusion to language have presented diffusion as an alternative to existing pretrained langu…

2023

Learning Iterative Neural Optimizers for Image Steganography

ICLR 2023poster

Image steganography is the process of concealing secret information in images through imperceptible changes. Recent work has formulated this task as a classic constrained optimization problem. In this paper, we argue that image steganography is inherently performed on the (elusive) manifold of natu…

2023

Learning To Invert: Simple Adaptive Attacks for Gradient Inversion in Federated Learning

UAI 2023poster

Gradient inversion attack enables recovery of training samples from model gradients in federated learning (FL), and constitutes a serious threat to data privacy. To mitigate this vulnerability, prior work proposed both principled defenses based on differential privacy, as well as heuristic defenses…

2023

On the Effectiveness of Offline RL for Dialogue Response Generation

ICML 2023poster

A common training technique for language models is teacher forcing (TF). TF attempts to match human language exactly, even though identical meanings can be expressed in different ways. This motivates use of sequence-level objectives for dialogue response generation. In this paper, we study the effic…

2023

Reward Finetuning for Faster and More Accurate Unsupervised Object Discovery

NeurIPS 2023poster

Recent advances in machine learning have shown that Reinforcement Learning from Human Feedback (RLHF) can improve machine learning models and align them with human preferences. Although very successful for Large Language Models (LLMs), these advancements have not had a comparable impact in research…

2023

Unsupervised Out-of-Distribution Detection with Diffusion Inpainting

ICML 2023poster

Unsupervised out-of-distribution detection (OOD) seeks to identify out-of-domain data by learning only from unlabeled in-domain data. We present a novel approach for this task -- Lift, Map, Detect (LMD) -- that leverages recent advancement in diffusion models. Diffusion models are one type of genera…

2023

Wav2Seq: Pre-Training Speech-to-Text Encoder-Decoder Models Using Pseudo Languages

ICASSP 2023accepted

We introduce Wav2Seq, the first self-supervised approach to pre-train both parts of encoder-decoder models for speech data. We induce a pseudo language as a compact discrete representation, and formulate a self-supervised pseudo speech recognition task — transcribing audio inputs into pseudo subword…

Cited by 0SourceScholar
2022

Exploiting Playbacks in Unsupervised Domain Adaptation for 3D Object Detection in Self-Driving Cars

ICRA 2022poster

Self-driving cars must detect other traffic participants like vehicles and pedestrians in 3D in order to plan safe routes and avoid collisions. State-of-the-art 3D object detectors, based on deep learning, have shown promising accuracy but are prone to over-fit domain idiosyncrasies, making them fai…

Cited by 25SourceScholar
2022

Fixed Neural Network Steganography: Train the images, not the network

ICLR 2022poster

Recent attempts at image steganography make use of advances in deep learning to train an encoder-decoder network pair to hide and retrieve secret messages in images. These methods are able to hide large amounts of data, but they also incur high decoding error rates (around 20%). In this paper, we pr…

2022

Hindsight is 20/20: Leveraging Past Traversals to Aid 3D Perception

ICLR 2022poster

Self-driving cars must detect vehicles, pedestrians, and other traffic participants accurately to operate safely. Small, far-away, or highly occluded objects are particularly challenging because there is limited information in the LiDAR point clouds for detecting them. To address this challenge, we l…

2022

Is High Variance Unavoidable in RL? A Case Study in Continuous Control

ICLR 2022poster

Reinforcement learning (RL) experiments have notoriously high variance, and minor details can have disproportionately large effects on measured outcomes. This is problematic for creating reproducible research and also serves as an obstacle when applying RL to sensitive real-world applications. In th…

Cited by 27SourcePDFScholar
2022

Ithaca365: Dataset and Driving Perception Under Repeated and Challenging Weather Conditions

CVPR 2022poster

Advances in perception for self-driving cars have accelerated in recent years due to the availability of large-scale datasets, typically collected at specific locations and under nice weather conditions. Yet, to achieve the high safety requirement, these perceptual systems must operate robustly unde…

Cited by 53PDFScholar
2022

Language-driven Semantic Segmentation

ICLR 2022poster

We present LSeg, a novel model for language-driven semantic image segmentation. LSeg uses a text encoder to compute embeddings of descriptive input labels (e.g., ``grass'' or ``building'') together with a transformer-based image encoder that computes dense per-pixel embeddings of the input image. Th…

2022

Learning To Detect Mobile Objects From LiDAR Scans Without Labels

CVPR 2022poster

Current 3D object detectors for autonomous driving are almost entirely trained on human-annotated data. Although of high quality, the generation of such data is laborious and costly, restricting them to a few specific locations and object types. This paper proposes an alternative approach entirely b…

Cited by 51PDFcodeScholar
2022

Performance-Efficiency Trade-Offs in Unsupervised Pre-Training for Speech Recognition

ICASSP 2022accepted

This paper is a study of performance-efficiency trade-offs in pre-trained models for automatic speech recognition (ASR). We focus on wav2vec 2.0, and formalize several architecture designs that influence both the model performance and its efficiency. Putting together all our observations, we introdu…

Cited by 0SourceScholar
2022

Unsupervised Adaptation from Repeated Traversals for Autonomous Driving

NeurIPS 2022accept

For a self-driving car to operate reliably, its perceptual system must generalize to the end-user's environment --- ideally without additional annotation efforts. One potential solution is to leverage unlabeled data (e.g., unlabeled LiDAR point clouds) collected from the end-users' environments (i.e…

2021

Characterizing the Loss Landscape in Non-Negative Matrix Factorization

AAAI 2021technical

Non-negative matrix factorization (NMF) is a highly celebrated algorithm for matrix decomposition that guarantees non-negative factors. The underlying optimization problem is computationally intractable, yet in practice, gradient-descent-based methods often find good solutions. In this paper, we rev…

Cited by 8SourcePDFScholar
2021

Deep Co-Training With Task Decomposition for Semi-Supervised Domain Adaptation

ICCV 2021poster

Semi-supervised domain adaptation (SSDA) aims to adapt models trained from a labeled source domain to a different but related target domain, from which unlabeled data and a small set of labeled data are provided. Current methods that treat source and target supervision without distinction overlook t…

Cited by 118PDFcodeScholar
2021

On Feature Normalization and Data Augmentation

CVPR 2021poster

The moments (a.k.a., mean and standard deviation) of latent features are often removed as noise when training image recognition models, to increase stability and reduce training time. However, in the field of image generation, the moments play a much more central role. Studies have shown that the mo…

Cited by 201PDFcodeScholar
2021

Revisiting Few-sample BERT Fine-tuning

ICLR 2021poster

This paper is a study of fine-tuning of BERT contextual representations, with focus on commonly observed instabilities in few-sample scenarios. We identify several factors that cause this instability: the common use of a non-standard optimization method with biased gradient estimation; the limited a…

2021

Towards Deeper Deep Reinforcement Learning with Spectral Normalization

NeurIPS 2021poster

In computer vision and natural language processing, innovations in model architecture that increase model capacity have reliably translated into gains in performance. In stark contrast with this trend, state-of-the-art reinforcement learning (RL) algorithms often use small MLPs, and gains in perform…

Cited by 30SourcePDFScholar
2020

BERTScore: Evaluating Text Generation with BERT

ICLR 2020poster

We propose BERTScore, an automatic evaluation metric for text generation. Analogously to common metrics, BERTScore computes a similarity score for each token in the candidate sentence with each token in the reference sentence. However, instead of exact matches, we compute token similarity using cont…

Cited by 6747SourcecodeScholar
2020

End-to-End Pseudo-LiDAR for Image-Based 3D Object Detection

CVPR 2020poster

Reliable and accurate 3D object detection is a necessity for safe autonomous driving. Although LiDAR sensors can provide accurate 3D point cloud estimates of the environment, they are also prohibitively expensive for many settings. Recently, the introduction of pseudo-LiDAR (PL) has led to a drastic…

Cited by 262PDFcodeScholar
2020

Identifying Mislabeled Data using the Area Under the Margin Ranking

NeurIPS 2020poster

Not all data in a typical training set help with generalization; some samples can be overly ambiguous or outrightly mislabeled. This paper introduces a new method to identify such samples and mitigate their impact when training neural networks. At the heart of our algorithm is the Area Under the Mar…

2020

Pseudo-LiDAR++: Accurate Depth for 3D Object Detection in Autonomous Driving

ICLR 2020poster

Detecting objects such as cars and pedestrians in 3D plays an indispensable role in autonomous driving. Existing approaches largely rely on expensive LiDAR sensors for accurate depth information. While recently pseudo-LiDAR has been introduced as a promising alternative, at a much lower cost based s…

Cited by 515SourcecodeScholar
2020

Train in Germany, Test in the USA: Making 3D Object Detectors Generalize

CVPR 2020poster

In the domain of autonomous driving, deep learning has substantially improved the 3D object detection accuracy for LiDAR and stereo camera data alike. While deep networks are great at generalization, they are also notorious to overfit to all kinds of spurious artifacts, such as brightness, car sizes…

Cited by 215PDFcodeScholar
2020

Wasserstein Distances for Stereo Disparity Estimation

NeurIPS 2020spotlight

Existing approaches to depth or disparity estimation output a distribution over a set of pre-defined discrete values. This leads to inaccurate results when the true depth or disparity does not match any of these values. The fact that this distribution is usually learned indirectly through a regressi…

2019

A New Defense Against Adversarial Images: Turning a Weakness into a Strength

NeurIPS 2019poster

Natural images are virtually surrounded by low-density misclassified regions that can be efficiently discovered by gradient-guided search --- enabling the generation of adversarial images. While many techniques for detecting these attacks have been proposed, they are easily bypassed when the adversa…

2019

Anytime Stereo Image Depth Estimation on Mobile Devices

ICRA 2019poster

Many applications of stereo depth estimation in robotics require the generation of accurate disparity maps in real time under significant computational constraints. Current state-of-the-art algorithms force a choice between either generating accurate mappings at a slow pace, or quickly generating in…

Cited by 265SourcecodeScholar
2019

Exact Gaussian Processes on a Million Data Points

NeurIPS 2019poster

Gaussian processes (GPs) are flexible non-parametric models, with a capacity that grows with the available data. However, computational constraints with standard inference procedures have limited exact GPs to problems with fewer than about ten thousand training points, necessitating approximations f…

2019

LDLS: 3-D Object Segmentation Through Label Diffusion From 2-D Images

RA-L 2019

Object segmentation in three-dimensional (3-D) point clouds is a critical task for robots capable of 3-D perception. Despite the impressive performance of deep learning-based approaches on object segmentation in 2-D images, deep learning has not been applied nearly as successfully for 3-D point clou

Cited by 39SourcecodeScholar
2019

Pseudo-LiDAR From Visual Depth Estimation: Bridging the Gap in 3D Object Detection for Autonomous Driving

CVPR 2019poster

3D object detection is an essential task in autonomous driving. Recent techniques excel with highly accurate detection rates, provided the 3D input data is obtained from precise but expensive LiDAR technology. Approaches based on cheaper monocular or stereo imagery data have, until now, resulted in…

Cited by 1347PDFcodeScholar
2018

CondenseNet: An Efficient DenseNet Using Learned Group Convolutions

CVPR 2018poster

Deep neural networks are increasingly used on mobile devices, where computational resources are limited. In this paper we develop CondenseNet, a novel network architecture with unprecedented efficiency. It combines dense connectivity with a novel module called learned group convolution. The dense co…

2018

GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration

NeurIPS 2018spotlight

Despite advances in scalable models, the inference tools used for Gaussian processes (GPs) have yet to fully capitalize on developments in computing hardware. We present an efficient and general approach to GP inference based on Blackbox Matrix-Matrix multiplication (BBMM). BBMM inference uses a mod…

2018

Resource Aware Person Re-Identification Across Multiple Resolutions

CVPR 2018poster

Not all people are equally easy to identify: color statistics might be enough for some cases while others might require careful reasoning about high- and low-level details. However, prevailing person re-identification(re-ID) methods use one-size-fits-all high-level embeddings from deep convolutional…

2017

On Fairness and Calibration

NeurIPS 2017poster

The machine learning community has become increasingly concerned with the potential for bias and discrimination in predictive models. This has motivated a growing line of work on what it means for a classification procedure to be "fair." In this paper, we investigate the tension between minimizing e…

2017

Snapshot Ensembles: Train 1, Get M for Free

ICLR 2017poster

Ensembles of neural networks are known to be much more robust and accurate than individual networks. However, training multiple deep networks for model averaging is computationally expensive. In this paper, we propose a method to obtain the seemingly contradictory goal of ensembling multiple neural…

Cited by 1227SourcecodeScholar
2016

Supervised Word Mover's Distance

NeurIPS 2016oral

Accurately measuring the similarity between text documents lies at the core of many real world applications of machine learning. These include web-search ranking, document recommendation, multi-lingual document matching, and article categorization. Recently, a new document metric, the word mover's d…

2015

Bayesian Active Model Selection with an Application to Automated Audiometry

NeurIPS 2015poster

We introduce a novel information-theoretic approach for active model selection and demonstrate its effectiveness in a real-world application. Although our method can work with arbitrary models, we focus on actively learning the appropriate structure for Gaussian process (GP) models with arbitrary ob…

Cited by 58SourcePDFScholar
2015

Fast Distributed k-Center Clustering with Outliers on Massive Data

NeurIPS 2015poster

Clustering large data is a fundamental problem with a vast number of applications. Due to the increasing size of data, practitioners interested in clustering have turned to distributed computation methods. In this work, we consider the widely used k-center clustering problem and its variant used t…

Cited by 108SourcePDFScholar