Adaptive Inference

Welcome to the Adaptive Inference Lab at the University of Wisconsin-Madison. We build statistical AI systems for context-aware healthcare.

Quick links:

Madison, Wisconsin

news

Apr 23, 2026 Award: Zhiyuan has been award a Hilldale Fellowship. Congratulations to Zhiyuan!
Mar 09, 2026 New Paper: Memory-Keyed Attention is published in ACM Computing Frontiers. Congrats to first-author Dong!
Nov 22, 2025 New paper: The hidden risk of round numbers and sharp thresholds in clinical practice is published in Nature’s Digital Medicine.
Jul 17, 2025 New paper: PiKV is published in ICML ES-FOMO. Congrats to first-author Dong!
Jun 19, 2025 New paper: Memory-Keyed Attention is published in ICML Long Context. Congrats to first-author Dong!
May 24, 2025 Media: Prof. Lengerich appeared on Tomorrow’s Medicine podcast.
May 23, 2025 New paper: Contextualized Gene Networks is published in PNAS. Congrats to first-author Caleb!
May 15, 2025 Award: Jingyun has been awarded an Outstanding TA award. Congrats Jingyun!
Apr 17, 2025 New paper: FastCache is published in CVPR BASE. Congrats to first-author Dong!
Mar 18, 2025 Media: Prof. Ben Lengerich discusses AI’s role in personalized healthcare in a faculty Q&A. Read more.
Jan 26, 2025 Award: Sid has been named a finalist in the Regeneron Science Talent Search. Congratulations Sid!
Dec 18, 2024 Award: NVIDIA has awarded us an Academic Grant. Thank you for the support!
Nov 04, 2024 New papers: Two findings papers accepted at ML4H 2024: From One to Zero: RAG-IM Adapts Language Models for Interpretable Zero-Shot Clinical Predictions; and Sample-Specific Models of Treatment Effects Explain Heterogeneity in Tuberculosis. Congratulations to Sazan, Ethan, and the team!
Aug 29, 2024 We are writing an open, collaborative review paper on context-adaptive inference. Contribute here.
Aug 19, 2024 Adaptive Inference lab is open at the University of Wisconsin-Madison!

selected publications

  1. The hidden risk of round numbers and sharp thresholds in clinical practice
    Benjamin J LengerichRich Caruana, Mark E Nunnally, and 1 more author
    npj Digital Medicine, 2025
  2. Learning to estimate sample-specific transcriptional networks for 7,000 tumors
    Caleb N EllingtonBenjamin J Lengerich, Thomas BK Watkins, and 6 more authors
    Proceedings of the National Academy of Sciences (PNAS), 2025
  3. Purifying Interaction Effects with the Functional ANOVA: An Efficient Algorithm for Recovering Identifiable Additive Models
    Ben Lengerich, Sarah Tan, Chun-Hao Chang, and 2 more authors
    In Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics (AISTATS) , 26–28 aug 2020