Dwaipayan Saha

Hey! I primarily work on efficient policy optimization via reinforcement learning, foundational models for alternative data modalities (tabular, time-series, etc), and diffusion models. Prior to this, I did some work at the intersection of high-dimensional statistics and econometrics.

I am a PhD student at Columbia University advised by Anish Agarwal.

Before that, I graduated Summa Cum Laude from Princeton University with a B.S.E. in Computer Science and Math. Here, I was fortunate to be advised by Matt Weinberg. I received the department’s Outstanding Student Teaching Award for my TA contributions across several undergraduate and graduate courses.

Two of my favorite courses at Princeton were Stochastic Calculus and Advanced Algorithm Design. A sizeable chunk of my time at here was spent thinking about theoretical computer science and probability theory all of which I still love chatting about!

Feel free to reach out at ds4386 (at) columbia (dot) edu. Here is my CV.

news

Dec 06, 2025	Presented TabImpute: Zero-Shot Missing-Data Imputation with a Pre-Trained Transformer at AI for Tabular Data Workshop @ EurIPS 2025 in Copenhagen, Denmark!
Oct 03, 2025	Poster presentation on recent work TabImpute: Zero-Shot Missing-Data Imputation with a Pre-Trained Transformer at CAIRFI 2025.
Sep 29, 2025	Started a deep learning alpha research internship at Headlands Technologies!
Jun 13, 2025	Gave a flash talk on Synthetic Blip Effects at the Causal Inference Workshop, held in conjunction with ACM SIGMETRICS 2025.
Apr 29, 2025	My work Synthetic Blip Effects was accepted to and presented at the Foundations of Data Science Workshop (Spring 2025).

latest posts

Aug 10, 2024	Variational Autoencoders and the EM Algorithm

selected publications

TabImpute: Accurate and Fast Zero-Shot Missing-Data Imputation with a Pre-Trained Transformer

Jacob Feitelberg, Dwaipayan Saha, Kyuseong Choi , and 3 more authors

AI for Tabular Data Workshop @ EurIPS, 2025

arXiv Bib

@article{feitelberg2025tabimputeaccuratefastzeroshot,
  title = {TabImpute: Accurate and Fast Zero-Shot Missing-Data Imputation with a Pre-Trained Transformer},
  author = {Feitelberg, Jacob and Saha, Dwaipayan and Choi, Kyuseong and Ahmad, Zaid and Agarwal, Anish and Dwivedi, Raaz},
  year = {2025},
  archiveprefix = {arXiv},
  primaryclass = {cs.LG},
  journal = {AI for Tabular Data Workshop @ EurIPS},
}

Stat & Econ

Synthetic Blips: Generalizing Synthetic Controls for Dynamic Treatment Effects

Anish Agarwal+, Sukjin Han+, Dwaipayan Saha+ , and 2 more authors

2025

arXiv Bib

@article{agarwal2025syntheticblipsgeneralizingsynthetic,
  title = {Synthetic Blips: Generalizing Synthetic Controls for Dynamic Treatment Effects},
  author = {Agarwal+, Anish and Han+, Sukjin and Saha+, Dwaipayan and Syrgkanis+, Vasilis and Yoon+, Haeyeon},
  year = {2025},
  archiveprefix = {arXiv},
  primaryclass = {econ.EM},
}