Publications

Below is a selected list of publications. See my Google Scholar for a complete list. Authors marked with * contributed equally.

2026

  1. Isotonic Survival Regression: Calibrated Survival Distributions from Deep Cox Models
    Anchit Jain*, Kevin Zhang*, and Stephen Bates
    2026
    arXiv
  2. Meta-Probabilistic Modeling
    Kevin Zhang and Yixin Wang
    In Conference on Artificial Intelligence and Statistics, 2026
    arXiv PDF Code

2025

  1. Path-Specific Effects for Pulse-Oximetry Guided Decisions in Critical Care
    Kevin Zhang, Yonghan Jung, Divyat Mahajan, and 2 more authors
    In Advances in Neural Information Processing Systems, 2025
    arXiv PDF Code
  2. Deep Autoregressive Models as Causal Inference Engines
    Daniel Jiwoong Im, Kevin Zhang, Nakul Verma, and 1 more author
    Transactions on Machine Learning Research, 2025
    arXiv PDF Code

2024

  1. G-MATT: Single-Step Retrosynthesis Prediction using Molecular Grammar Tree Transformer
    Kevin Zhang*, Vipul Mann*, and Venkat Venkatasubramanian
    AIChE Journal, 2024
    arXiv PDF