Bruce (Zhi) Wen

Senior applied research scientist at Mila - Quebec AI Institute

portrait.jpg

I am part of the applied research team at Mila, where we solve real-world problems with AI. Example projects include working with Dialogue on automatic symptom detection, with D-BOX on movie event detection, and with AlayaCare on agent evaluation.

These days I work a lot on building and evaluating agents for various use cases. Fittingly, DDXPlus, which we released in 2022 from the Dialogue project, has become a popular benchmark and training dataset for medical LLMs and agents. Personally, I am most interested in a few topics in NLP, such as generalization of self-improving agents, alternatives to current LLM recipes, controllable generation, and decoding algorithms. I also keep an eye on healthcare/bio applications, such as protein design.

I earned my master’s degree at McGill University, and I did my undergrad at Wuhan University, China. My research focused on machine learning and NLP for healthcare. For instance, I worked on COVID-19 media news surveillance for public health measures, constructing a large medical NLP pre-training dataset, among others.

In my free time, I’ve recently gotten into rock climbing. I used to play a lot of football (the one where you play with your foot). I have an eclectic taste in music, but what I listen to mostly clusters around: progressive, post-rock, alternative, and ’60s–’80s Japanese music.

selected publications

  1. NeurIPS
    Towards Trustworthy Automatic Diagnosis Systems by Emulating Doctors’ Reasoning with Deep Reinforcement Learning
    Arsene Fansi Tchango, Rishab Goel, Julien Martel, and 3 more authors
    Thirty-sixth Conference on Neural Information Processing Systems, 2022
  2. NeurIPS Dataset
    DDXPlus: A new Dataset for Medical Automatic Diagnosis
    Arsene Fansi Tchango, Rishab Goel, Zhi Wen, and 2 more authors
    Thirty-sixth Conference on Neural Information Processing Systems Datasets and Benchmarks Track, 2022
    Standard benchmark for medical LLM agents, used in Agentic Context Engineering (ICLR 2026), MediQ (NeurIPS 2024), StreamBench (NeurIPS 2024), and Chain-of-Diagnosis (ACL 2024).
  3. Patterns
    Inferring global-scale temporal latent topics from news reports to predict public health interventions for COVID-19
    Zhi Wen, Guido Powell, Imane Chafi, and 2 more authors
    Patterns, 2022
  4. EMNLP Workshop
    MeDAL: Medical Abbreviation Disambiguation Dataset for Natural Language Understanding Pretraining
    Zhi Wen, Xing Han Lu, and Siva Reddy
    In Proceedings of the 3rd Clinical Natural Language Processing Workshop, 2020
  5. Nat. Commun.
    Inferring multimodal latent topics from electronic health records
    Yue Li, Pratheeksha Nair, Xing Han Lu, and 8 more authors
    Nature communications, 2020