← Search

Soumya Basu

13 accepted papers

2024

A Statistical Framework for Data-dependent Retrieval-Augmented Models

ICML 2024poster

Modern ML systems increasingly augment input instances with additional relevant information to enhance final prediction. Despite growing interest in such retrieval-augmented models, their fundamental properties and training are not well understood. We propose a statistical framework to study such mo…

Cited by 0SourcePDFScholar
2022

Recoverability Landscape of Tree Structured Markov Random Fields under Symmetric Noise

AISTATS 2022poster

We study the problem of learning tree-structured Markov random fields (MRF) on discrete random variables with common support when the observations are corrupted by a k-ary symmetric noise channel with unknown probability of error. For Ising models (support size = 2), past work has shown that graph s…

2021

Beyond $log^2(T)$ regret for decentralized bandits in matching markets

ICML 2021spotlight

We design decentralized algorithms for regret minimization in the two sided matching market with one-sided bandit feedback that significantly improves upon the prior works (Liu et al.\,2020a, Sankararaman et al.\,2020, Liu et al.\,2020b). First, for general markets, for any $\varepsilon > 0$, we des…

Cited by 52SourcePDFScholar
2021

Combinatorial Blocking Bandits with Stochastic Delays

ICML 2021spotlight

Recent work has considered natural variations of the {\em multi-armed bandit} problem, where the reward distribution of each arm is a special function of the time passed since its last pulling. In this direction, a simple (yet widely applicable) model is that of {\em blocking bandits}, where an arm…

Cited by 16SourcePDFScholar
2021

Contextual Blocking Bandits

AISTATS 2021poster

We study a novel variant of the multi-armed bandit problem, where at each time step, the player observes an independently sampled context that determines the arms’ mean rewards. However, playing an arm blocks it (across all contexts) for a fixed number of future time steps. The above contextual sett…

Cited by 28SourcePDFScholar
2021

Dominate or Delete: Decentralized Competing Bandits in Serial Dictatorship

AISTATS 2021poster

Online learning in a two-sided matching market, with demand side agents continuously competing to be matched with supply side (arms), abstracts the complex interactions under partial information on matching platforms (e.g. UpWork, TaskRabbit). We study the decentralized serial dictatorship setting,…

Cited by 47SourcePDFScholar
2020

Learning Mixtures of Graphs from Epidemic Cascades

ICML 2020poster

We consider the problem of learning the weighted edges of a balanced mixture of two undirected graphs from epidemic cascades. While mixture models are popular modeling tools, algorithmic development with rigorous guarantees has lagged. Graph mixtures are apparently no exception: until now, very litt…

Cited by 10SourcePDFScholar