← Search

Stephan Rabanser

9 accepted papers

2026

Cascadia: An Efficient Cascade Serving System for Large Language Models

ICLR 2026poster

Recent advances in large language models (LLMs) have intensified the need to deliver both rapid responses and high-quality outputs. More powerful models yield better results but incur higher inference latency, whereas smaller models are faster yet less capable. Recent work proposes balancing this la…

Cited by 6SourceScholar
2026

Towards a Science of AI Agent Reliability

ICML 2026poster

AI agents are increasingly deployed for consequential tasks. Yet existing benchmarks evaluate only task success rates, ignoring whether agents behave consistently, remain robust to perturbations, fail predictably, or bound error severity. We propose a framework for measuring agent reliability ground…

Cited by 0SourceScholar
2025

Confidential Guardian: Cryptographically Prohibiting the Abuse of Model Abstention

ICML 2025poster

Cautious predictions—where a machine learning model abstains when uncertain—are crucial for limiting harmful errors in safety-critical applications. In this work, we identify a novel threat: a dishonest institution can exploit these mechanisms to discriminate or unjustly deny services under the guis…

2025

Gatekeeper: Improving Model Cascades Through Confidence Tuning

NeurIPS 2025poster

Large-scale machine learning models deliver strong performance across a wide range of tasks but come with significant computational and resource constraints. To mitigate these challenges, local smaller models are often deployed alongside larger models, relying on routing and deferral mechanisms to o…

Cited by 0SourceScholar
2025

Suitability Filter: A Statistical Framework for Classifier Evaluation in Real-World Deployment Settings

ICML 2025oral

Deploying machine learning models in safety-critical domains poses a key challenge: ensuring reliable model performance on downstream user data without access to ground truth labels for direct validation. We propose the _suitability filter_, a novel framework designed to detect performance deteriora…

Cited by 0SourcePDFScholar
2023

Robust and Actively Secure Serverless Collaborative Learning

NeurIPS 2023poster

Collaborative machine learning (ML) is widely used to enable institutions to learn better models from distributed data. While collaborative approaches to learning intuitively protect user data, they remain vulnerable to either the server, the clients, or both, deviating from the protocol. Indeed, be…

Cited by 4SourcePDFScholar
2023

Training Private Models That Know What They Don’t Know

NeurIPS 2023poster

Training reliable deep learning models which avoid making overconfident but incorrect predictions is a longstanding challenge. This challenge is further exacerbated when learning has to be differentially private: protection provided to sensitive data comes at the price of injecting additional random…

Cited by 5SourcePDFScholar
2019

Failing Loudly: An Empirical Study of Methods for Detecting Dataset Shift

NeurIPS 2019poster

We might hope that when faced with unexpected inputs, well-designed software systems would fire off warnings. Machine learning (ML) systems, however, which depend strongly on properties of their inputs (e.g. the i.i.d. assumption), tend to fail silently. This paper explores the problem of building M…