BIOGRAPHY

I am currently a postdoctoral scholar at Stanford advised by Prof. Emily B. Fox. Subsequently, I will join MBZUAI Department of Statistics and Data Science and Department of Machine Learning as a tenure-track assistant professor in the fall of 2026.

I received my Ph.D. degree in Computer Science at UC San Diego in 2025, advised by Prof. Rose Yu and Prof. Yian Ma. I received my Bachelor degree of science in Applied Math, Physics, and Computer Sciences at University of Wisconsin-Madison in 2020. My research primarily lies in Bayesian Deep Learning, Sequential Decision Making, Scientific Machine Learning, Spatiotemporal Modeling, and AI for Public Health. The proposed approaches have been widely used in public health, traffic modeling, climate science, and drug design.

My lab is seeking several highly motivated PhD students/Postdocs/RAs/Visiting PhD students. All positions begin in Fall 2026. I recruit from the MBZUAI Statistics and Data Science Department and the Machine Learning Department. Specifically, I am seeking students in the following research areas:

  • Probabilistic Machine Learning: uncertainty quantification, Bayesian optimization
  • Spatiotemporal Modeling, Control, and Optimization: time series model (behavioral/biomedical), climate modeling, smart cities, supply chain, and industrial planning.
  • AI for Healthcare: disease control, drug design, and cell modeling.
  • Multimodal and Foundation Models: LLM trustworthiness, foundation models in science and engineering.

If you are interested in working with me, please email your CV, academic transcript, and a brief outline of your research plan to dowu@stanford.edu with the subject line: [Your Name] [Position You Are Applying For] - MBZUAI Research Opportunity Application.

Publications

  • MF-LAL: Drug Compound Generation Using Multi-Fidelity Latent Space Active Learning
  • Peter Eckmann, Dongxia Wu, Germano Heinzelmann, Michael K Gilson, Rose Yu.
    International Conference on Machine Learning (ICML) 2025
  • Diffusion Models as Constrained Samplers for Optimization with Unknown Constraints
  • Lingkai Kong, Yuanqi Du, Wenhao Mu, Kirill Neklyudov, Valentin De Bortoli, Dongxia Wu, Haorui Wang, Aaron Ferber, Yi-An Ma, Carla P. Gomes, Chao Zhang.
    International Conference on Artificial Intelligence and Statistics (AISTATS) 2025
  • Learning Granger Causality from Instance-wise Self-attentive Hawkes Processes
  • Dongxia Wu, Tsuyoshi Idé, Aurélie Lozano, Georgios Kollias, Jiří Navrátil, Naoki Abe, Yi-An Ma, Rose Yu.
    International Conference on Artificial Intelligence and Statistics (AISTATS) 2024
  • Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modeling
  • Ruijia Niu, Dongxia Wu, Kai Kim, Yi-An Ma, Duncan Watson-Parris, Rose Yu.
    International Conference on Machine Learning (ICML) 2024
  • Disentangled Multi-Fidelity Deep Bayesian Active Learning
  • Dongxia Wu, Ruijia Niu, Matteo Chinazzi, Yian Ma, Rose Yu.
    International Conference on Machine Learning (ICML) 2023
  • Deep Bayesian Active Learning for Accelerating Stochastic Simulation
  • Dongxia Wu, Ruijia Niu, Matteo Chinazzi, Alessandro Vespignani, Yian Ma, Rose Yu.
    ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) 2023
  • Multi-fidelity Hierarchical Neural Processes
  • Dongxia Wu, Matteo Chinazzi, Alessandro Vespignani, Yi-An Ma, Rose Yu.
    ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) 2022
  • DeepViFi: detecting oncoviral infections in cancer genomes using transformers
  • Utkrisht Rajkumar, Sara Javadzadeh, Mihir Bafna, Dongxia Wu, Rose Yu, Jingbo Shang, Vineet Bafna.
    ACM International Conference on Bioinformatics, Computational Biology and Health Informatics (BCB) 2022
  • Quantifying Uncertainty in Deep Spatiotemporal Forecasting
  • Dongxia Wu, Liyao Gao, Xinyue Xiong, Matteo Chinazzi, Alessandro Vespignani, Yian Ma, Rose Yu.
    ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) 2021
  • Evaluation of individual and ensemble probabilistic forecasts of COVID-19 mortality in the United States
  • Estee Cramer, [et al, including Dongxia Wu]
    Proceedings of the National Academy of Sciences, 2022, 119.15: e2113561119.
  • DeepGLEAM: A hybrid mechanistic and deep learning model for COVID-19 forecasting
  • Dongxia Wu, Liyao Gao, Xinyue Xiong, Matteo Chinazzi, Alessandro Vespignani, Yian Ma, Rose Yu.
    https://arxiv.org/abs/2102.06684
  • A deep learning based automatic defect analysis framework for In-situ TEM ion irradiations
  • Mingren Shen, Guanzhao Li, Dongxia Wu, et al.
    Computational Materials Science, 2021, 197: 110560.
  • Multi defect detection and analysis of electron microscopy images with deep learning
  • Mingren Shen, Guanzhao Li, Dongxia Wu, et al.
    Computational Materials Science, 2021, 197: 110560.
  • Diff-BBO: Diffusion-Based Inverse Modeling for Black-Box Optimization
  • Dongxia Wu, Nikki Lijing Kuang, Ruijia Niu, Yi-An Ma, Rose Yu.
    NeurIPS Workshop on Bayesian Decision-making and Uncertainty 2024
  • Functional-level Uncertainty Quantification for Calibrated Fine-tuning on LLMs
  • Ruijia Niu, Dongxia Wu, Rose Yu, Yi-An Ma.
    NeurIPS Workshop on Statistical Foundations of LLMs and Foundation Models 2024
  • GLEAM-AI: Neural Surrogate for Accelerated Epidemic Analytics and Forecasting
  • Mohammadmehdi Zahedi, Dongxia Wu, Jessica T. Davis, Yian Ma, Alessandro Vespignani, Rose Yu, Matteo Chinazzi.
    NeurIPS Workshop on Bayesian Decision-making and Uncertainty 2024

    Experiences

    Research Assistant

    2020 - 2025
    UC San Diego, La Jolla, CA
    • Spearheaded the uncertainty quantification study for spatiotemporal forecasting, compared Bayesian and Frequentist methods on traffic and COVID-19 predictions.

    Research Intern

    June 2024 - September 2024
    Samsung Research America, Mountain View, CA
    • Designed Time Series Foundation Model for Anomaly Detection.

    Applied Scientist Intern

    June 2023 - September 2023
    Amazon, Seattle, WA
    • Designed the contextual offer estimation method to understand and model customer preferences.

    AI Intern

    June 2022 - September 2022
    IBM Thomas J. Watson Research Center, Yorktown Heights, NY
    • Spearheaded the causal inference study on multivariate point processes.

    Awards/Honors

    HDSI Ph.D. Fellowship (2021-2024), top 10 in admitted HDSI Ph.D. students - UC San Diego (2021)
    College of Letters & Science Dean’s List - UW–Madison (2018, 2019)
    Undergraduate Summer Scholarship - UW-Madison (2018)
    The First Prize in China Region and the Fourth Prize in the final - The Third Hong Kong International Chamber Music Competition (2016)