1 citations · 4 across the 11 of their papers we have counts for
12 papers
Temporal Self-Distillation: Faster Inference in Discrete Diffusion Language Models
Shijian Xu, Andrea Miele, Metod Jazbec +3
Diffusion language models (dLLMs) promise fast inference by generating multiple tokens in parallel, but suffer severe performance degradation when parallel decoding is pushed too a…
Online Safety Monitoring for LLMs
Mona Schirmer, Metod Jazbec, Alexander Timans +3
Despite alignment training, LLMs remain prone to generating unsafe outputs at deployment time. Monitoring outputs online and raising an alarm when safety can no longer be assumed i…
Conformal Thinking: Risk Control for Reasoning on a Compute Budget
Xi Wang, Anushri Suresh, Alvin Zhang +6
Reasoning Large Language Models (LLMs) enable test-time scaling, with dataset-level accuracy improving as the token budget increases, motivating adaptive reasoning -- spending toke…
Controlling the Risk of Corrupted Contexts for Language Models via Early-Exiting
Andrea Wynn, Metod Jazbec, Charith Peris +4
Large language models (LLMs) can be influenced by harmful or irrelevant context, which can significantly harm model performance on downstream tasks. This motivates principled desig…
Monitoring Risks in Test-Time Adaptation
Mona Schirmer, Metod Jazbec, Christian A. Naesseth +1
Encountering shifted data at test time is a ubiquitous challenge when deploying predictive models. Test-time adaptation (TTA) methods address this issue by continuously adapting a…
Generative Uncertainty in Diffusion Models
Metod Jazbec, Eliot Wong-Toi, Guoxuan Xia +3
Diffusion models have recently driven significant breakthroughs in generative modeling. While state-of-the-art models produce high-quality samples on average, individual samples ca…
Lightning UQ Box: A Comprehensive Framework for Uncertainty Quantification in Deep Learning
Nils Lehmann, Jakob Gawlikowski, Adam J. Stewart +4
Uncertainty quantification (UQ) is an essential tool for applying deep neural networks (DNNs) to real world tasks, as it attaches a degree of confidence to DNN outputs. However, de…
Temporal Test-Time Adaptation with State-Space Models
Mona Schirmer, Dan Zhang, Eric Nalisnick
Distribution shifts between training and test data are inevitable over the lifecycle of a deployed model, leading to performance decay. Adapting a model on test samples can help mi…
Fast yet Safe: Early-Exiting with Risk Control
Metod Jazbec, Alexander Timans, Tin Hadži Veljković +4
Scaling machine learning models significantly improves their performance. However, such gains come at the cost of inference being slow and resource-intensive. Early-exit neural net…
Uncertainty Aware Tropical Cyclone Wind Speed Estimation from Satellite Data
Nils Lehmann, Nina Maria Gottschling, Stefan Depeweg +1
Deep neural networks (DNNs) have been successfully applied to earth observation (EO) data and opened new research avenues. Despite the theoretical and practical advances of these t…
Adaptive Bounding Box Uncertainties via Two-Step Conformal Prediction
Alexander Timans, Christoph-Nikolas Straehle, Kaspar Sakmann +1
Quantifying a model's predictive uncertainty is essential for safety-critical applications such as autonomous driving. We consider quantifying such uncertainty for multi-object det…
Max-Rank: Efficient Multiple Testing for Conformal Prediction
Alexander Timans, Christoph-Nikolas Straehle, Kaspar Sakmann +2
Multiple hypothesis testing (MHT) frequently arises in scientific inquiries, and concurrent testing of multiple hypotheses inflates the risk of Type-I errors or false positives, re…