Publications

2026

1. Bayesian Invariant Representation Learning for Robust Prediction of Multi-Environment Data

   Matthew Shen, Luhuan Wu, and David M. Blei

   2026

   In submission to NeurIPS 2026
2. Environment-Robust Representation Learning with Empirical Bayes

   Yuli Slavutsky, Matthew Shen, Bohan Wu, and 1 more author

   2026

   In submission to NeurIPS 2026

   arXiv HTML
3. Test-Time Scaling of Diffusion Models at the Pareto Frontier

   Matthew Shen, Luhuan Wu, Jonathan Wenger, and 1 more author

   2026

   In preparation
4. Group-Aware Matrix Estimation and Latent Subspace Recovery

   Hamza Golubovic^*, Matthew Shen^*, Genevera Allen, and 1 more author

   2026

   In submission to NeurIPS 2026

   arXiv HTML

2025

1. Building Trust @ ICLR

   Adaptive Test-Time Intervention for Concept Bottleneck Models

   Matthew Shen^*, Aliyah Hsu^*, Abhineet Agarwal, and 1 more author

   In Building Trust Workshop - ICLR, 2025

   arXiv HTML Website

2024

1. Gastro.

   Validation of an Electronic Health Record Based Machine Learning Model Compared With Clinical Risk Scores for Gastrointestinal Bleeding

   Dennis L. Shung, Colleen E. Chan, Kisung You, and 11 more authors

   Gastroenterology, 2024

   HTML