4 papers
Risk-Controlled Post-Processing of Decision Policies
Sunay Joshi, Tao Wang, Hamed Hassani +1
Predictive models are often deployed through existing decision policies that stakeholders are reluctant to change unless a risk constraint requires intervention. We study risk-cont…
Foundations of Top- Decoding For Language Models
Georgy Noarov, Soham Mallick, Tao Wang +5
Top- decoding is a widely used method for sampling from LLMs: at each token, only the largest next-token-probabilities are kept, and the next token is sampled after re-norma…
MultiRisk: Multiple Risk Control via Iterative Score Thresholding
Sunay Joshi, Yan Sun, Hamed Hassani +1
As generative AI systems are increasingly deployed in real-world applications, regulating multiple dimensions of model behavior has become essential. We focus on test-time filterin…
Conformal Inference under High-Dimensional Covariate Shifts via Likelihood-Ratio Regularization
Sunay Joshi, Shayan Kiyani, George Pappas +2
We consider the problem of conformal prediction under covariate shift. Given labeled data from a source domain and unlabeled data from a covariate shifted target domain, we seek to…