← Search

Babak Salimi

5 accepted papers

2025

Graph Machine Learning based Doubly Robust Estimator for Network Causal Effects

AISTATS 2025poster

Estimating causal effects in social network data presents unique challenges due to the presence of spillover effects and network-induced confounding. While much of the existing literature addresses causal inference in social networks, many methods rely on strong assumptions about the form of network…

Cited by 0SourceScholar
2025

Scalable Out-of-Distribution Robustness in the Presence of Unobserved Confounders

AISTATS 2025poster

We consider the task of out-of-distribution (OOD) generalization, where the distribution shift is due to an unobserved confounder ($Z$) affecting both the covariates ($X$) and the labels ($Y$). This confounding introduces heterogeneity in the predictor, i.e., $P(Y \mid X) = E_{P(Z \mid X)}[P(Y \mid…

Cited by 0SourceScholar
2025

TokenSwap: A Lightweight Method to Disrupt Memorized Sequences in LLMs

NeurIPS 2025spotlight

As language models scale, their performance improves dramatically across a wide range of tasks, but so does their tendency to memorize and regurgitate parts of their training data verbatim. This tradeoff poses serious legal, ethical, and safety concerns, especially in real-world deployments. Existin…

Cited by 0SourcecodeScholar
2024

Learning from Uncertain Data: From Possible Worlds to Possible Models

NeurIPS 2024poster

We introduce an efficient method for learning linear models from uncertain data, where uncertainty is represented as a set of possible variations in the data, leading to predictive multiplicity. Our approach leverages abstract interpretation and zonotopes, a type of convex polytope, to compactly rep…

Cited by 1SourcePDFScholar