← Search

Mohsen Bayati

12 accepted papers

2025

Conformal Arbitrage: Risk-Controlled Balancing of Competing Objectives in Language Models

NeurIPS 2025poster

Modern language‑model deployments must often balance competing objectives—for example, helpfulness versus harmlessness, cost versus accuracy, and reward versus safety. We introduce Conformal Arbitrage, a post‑hoc framework that learns a data‑driven threshold to mediate between a Primary model optimi…

Cited by 0SourceScholar
2025

Geometry-Aware Approaches for Balancing Performance and Theoretical Guarantees in Linear Bandits

ICLR 2025poster

This paper is motivated by recent research in the $d$-dimensional stochastic linear bandit literature, which has revealed an unsettling discrepancy: algorithms like Thompson sampling and Greedy demonstrate promising empirical performance, yet this contrasts with their pessimistic theoretical regret…

Cited by 1SourcePDFScholar
2025

Quantile Regression with Large Language Models for Price Prediction

ACL 2025finding

Large Language Models (LLMs) have shown promise in structured prediction tasks, including regression, but existing approaches primarily focus on point estimates and lack systematic comparison across different methods.We investigate probabilistic regression using LLMs for unstructured inputs, address…

2024

Higher-Order Causal Message Passing for Experimentation with Complex Interference

NeurIPS 2024poster

Accurate estimation of treatment effects is essential for decision-making across various scientific fields. This task, however, becomes challenging in areas like social sciences and online marketplaces, where treating one experimental unit can influence outcomes for others through direct or indirect…

Cited by 1SourcePDFScholar
2022

Thompson Sampling Efficiently Learns to Control Diffusion Processes

NeurIPS 2022accept

Diffusion processes that evolve according to linear stochastic differential equations are an important family of continuous-time dynamic decision-making models. Optimal policies are well-studied for them, under full certainty about the drift matrices. However, little is known about data-driven contr…

Cited by 11SourcePDFScholar
2020

Recommendation on a Budget: Column Space Recovery from Partially Observed Entries with Random or Active Sampling

AISTATS 2020poster

We analyze alternating minimization for column space recovery of a partially observed, approximately low rank matrix with a growing number of columns and a fixed budget of observations per column. We prove that if the budget is greater than the rank of the matrix, column space recovery succeeds – as…

Cited by 1SourcePDFScholar
2020

Unreasonable Effectiveness of Greedy Algorithms in Multi-Armed Bandit with Many Arms

NeurIPS 2020spotlight

We study the structure of regret-minimizing policies in the {\em many-armed} Bayesian multi-armed bandit problem: in particular, with $k$ the number of arms and $T$ the time horizon, we consider the case where $k \geq \sqrt{T}$. We first show that {\em subsampling} is a critical step for designing o…