He Zhao

Seasoned ML/AI Researcher

Hello! This is my research page.

I'm a machine learning and AI researcher at CSIRO and an adjunct researcher at Monash University. Previously, I obtained my PhD from the Department of Data Science and AI of Monash University and worked as a postdoc there.

I started my research jouney on Bayesian machine learning in my PhD and then worked on representation learning via optimal transport in my postdoc. Now I have broad interests in causality, uncertainty, and robustness in machine learning on language, vision, and tabular data. For details, check out my papers and their topics below.

I regularly serve as an area chair for NeurIPS, ICML, and ICLR, and I am on the editorial board of Machine Learning Journal. I am an IEEE Senior Member from 2026.

My KDD 2023 paper won the Best Student Paper Award.
Asian Conference on Machine Learning (ACML) 2026 is calling for tutorial proposals.

Research output

Selected publications

48 papers shown

TopicSort

Year	Publication
2026	KDD Ordering-based Causal Discovery via Generalized Score Matching V. Vo, H. Zhao, T. Le, E. V. Bonilla, D. Phung arxiv Causality
2026	ICML Causal Preference Elicitation E. V. Bonilla, H. Zhao, D. Steinberg arxiv code CausalityBayesian Models
2026	ICML Multi-Scale Wavelet Transformers for Operator Learning of Dynamical Systems X. Wang, M. Groom, R. Oliveira, H. Zhao, T. O'kane, E. V. Bonilla arxiv AI for Science
2026	ACL Safeguarding LLM Fine-tuning via Push-Pull Distributional Alignment H. Wang, Z. Li, Y. Yang, H. Zhao, H. Zha, D. Guo arxiv LLMsUncertainty, Robustness, Safety
2026	CVPR FindingsCVPR Compute Transparency Champion FedNPC: Stochastic Noise-driven Post-hoc Classifier Calibration Method for Federated Long-tailed Learning J. Gao1, H. Zhao, Y. Yang, D. Guo award code Federated Learning
2026	TMLR HiBaNG: Hierarchical Bayesian Nonparametric Granger Causal Discovery in Low-Data Regimes H. Zhao, V. Kitsios, T. O'kane, E. V. Bonilla CausalityAI for ScienceBayesian Models
2026	ICLR Imitating the Truth: Attention-aware Truth-Guided Enhancement for Hallucination Mitigation in Large Vision-Language Models H. Ren, Z. Wang, Y. Yang, H. Zhao, F. Tang, D. Guo, Y. Chang LLMsVision
2026	ICLR LLM as an Algorithmist: Enhancing Anomaly Detectors via Programmatic Synthesis H. Ye, J. Li, H. Zhao, M. Zhuge, D. Guo, Y. Chang, H. Zha LLMsTabular Data
2026	IJCV Near OOD Detection for Vision-Language Prompt Learning with Contrastive Logit Score M.C. Jung, J. Dipnall, B. Gabbe, H. Zhao code VisionUncertainty, Robustness, Safety
2026	ICLR Unifying Stable Optimization and Reference Regularization in RLHF L. He, Q. Qu, H. Zhao, S. Wan, D. Wang, L. Yao, T. Liu LLMs
2025	CVPR Balancing Two Classifiers via A Simplex ETF Structure for Model Calibration J. Ni, H. Zhao, J. Gao, D. Guo, H. Zha VisionUncertainty, Robustness, Safety
2025	CVPR Beyond Words: Augmenting Discriminative Richness via Diffusions in Unsupervised Prompt learning H. Ren, F. Tang, H. Zhao, Z. Wang, D. Guo, Y. Chang VisionLLMs
2025	TPAMI Deep Tabular Representation Corrector H. Ye, P. Wang, W. Fan, X. Song, H. Zhao, D. Guo, Y. Chang Tabular Data
2025	ICLR DRL: Decomposed Representation Learning for Tabular Anomaly Detection H. Ye, H. Zhao, W. Fan, M. Zhou, D. Guo, Y. Chang Tabular Data
2025	CVPR FedAWA: Adaptive Optimization of Aggregation Weights in Federated Learning Using Client Vectors C. Shi, H. Zhao, B. Zhang, M. Zhou, D. Guo, Y. Chang code Federated LearningVision
2025	ICLR FedLWS: Federated Learning with Adaptive Layer-wise Weight Shrinking C. Shi, J. Li, H. Zhao, D. Guo, Y. Chang Federated Learning
2025	NeurIPSSpotlight, Top 3% LLM Meeting Decision Trees on Tabular Data H. Ye, J. Li, H. Zhao, D. Guo, Y. Chang code LLMsTabular Data
2025	TACLPresented at ACL 2025 LLM Reading Tea Leaves: Automatically Evaluating Topic Models with Large Language Models X. Yang, H. Zhao, D. Phung, W. Buntine, L. Du code LLMsText Understanding
2025	ACLMain Conference Neural Topic Modeling with Large Language Models in the Loop X. Yang, H. Zhao, W. Xu, Y. Qi, J. Lu, D. Phung, L. Du code LLMsText Understanding
2025	ICMLOral, Top 1% Renyi Neural Processes X. Wang, H. Zhao, E. V. Bonilla code Uncertainty, Robustness, SafetyBayesian Models
2024	ICML Distribution Alignment Optimization through Neural Collapse for Long-tailed Classification J. Gao, H. Zhao, D. Guo, H. Zha code VisionOptimal Transport
2024	ICML Parameter Estimation in DAGs from Incomplete Data via Optimal Transport V. Vo, T. Le, L. T. Vuong, H. Zhao, E. V. Bonilla, D. Phung code CausalityOptimal Transport
2024	ICLRSpotlight, Top 5% PTaRL: Prototype-based Tabular Representation Learning via Space Calibration H. Ye, W. Fan, X. Song, S. Zheng, H. Zhao, D. Guo, Y. Chang code Tabular Data
2023	NeurIPS Beyond Unimodal: Generalising Neural Processes for Multimodal Uncertainty Estimation M.C. Jung, H. Zhao, J. Dipnall, L. Du code Uncertainty, Robustness, Safety
2023	NeurIPS Enhancing Minority Classes by Mixing: An Adaptative Optimal Transport Approach for Long-tailed Classification J. Gao, H. Zhao, Z. Li, D. Guo code Optimal TransportVision
2023	KDDBest Student Paper Feature-based Learning for Diverse and Privacy-Preserving Counterfactual Explanations V. Vo, T. Le, V. Nguyen, H. Zhao, E. Bonilla, G. Haffari, D. Phung award code Causality
2023	TMLR Generating Adversarial Examples with Task Oriented Multi-Objective Optimization A. Bui, T. Le, H. Zhao, Q.H. Tran, P. Montague, D. Phung code Uncertainty, Robustness, Safety
2023	NeurIPS NPCL: Neural Processes for Uncertainty-Aware Continual Learning S. Jha, D. Gong, H. Zhao, L. Yao code Uncertainty, Robustness, Safety
2023	ICML Transformed Distribution Matching for Missing Value Imputation H. Zhao, K. Sun, A. Dezfouli, E. Bonilla code Optimal TransportTabular Data
2023	ICML Vector Quantized Wasserstein Auto-Encoder L.T. Vuong, T. Le, H. Zhao, C. Zheng, M. Harandi, J. Cai, D. Phung code Optimal TransportVision
2022	AISTATS A Global Defense Approach via Adversarial Attack and Defense Risk Guaranteed Bounds T. Le, A. Bui, M.T.T. Le, H. Zhao, P. Montague, Q. Tran, D. Phung Uncertainty, Robustness, Safety
2022	ICLR A Unified Wasserstein Distributional Robustness Framework for Adversarial Training A. Bui, T. Le, Q. Tran, H. Zhao, D. Phung code Optimal TransportUncertainty, Robustness, Safety
2022	NeurIPS Adaptive Distribution Calibration for Few-Shot Learning with Hierarchical Optimal Transport D. Guo, L. Tian, H. Zhao, M. Zhou, H. Zha code Optimal TransportVision
2022	NeurIPS Learning to Re-weight Examples with Optimal Transport for Imbalanced Classification D. Guo, Z. Li, M. Zheng, H. Zhao, M. Zhou, H. Zha code Optimal TransportVision
2022	AISTATS Particle-based Adversarial Local Distribution Regularization T. Nguyen, T. Le, H. Zhao, J. Cai, D. Phung Uncertainty, Robustness, Safety
2022	ICLR Representing Mixtures of Word Embeddings with Mixtures of Topic Embeddings D. Wang, D. Guo, H. Zhao, H. Zheng, K. Tanwisuth, B. Chen, M. Zhou code Optimal TransportText Understanding
2022	NeurIPS Uncertainty Estimation for Multi-view Data: The Power of Seeing the Whole Picture M.C. Jung, H. Zhao, J. Dipnall, B. Gabbe, L. Du code Uncertainty, Robustness, Safety
2021	ICLRSpotlight, Top 5% Neural Topic Model via Optimal Transport H. Zhao, D. Phung, V. Huynh, T. Le, W. Buntine code Text UnderstandingOptimal Transport
2021	IJCAISurvey Track Topic Modelling Meets Deep Neural Networks: A Survey H. Zhao, D. Phung, V. Huynh, Y. Jin, L. Du, W. Buntine arxiv Text Understanding
2020	ECCV Improving Adversarial Robustness by Enforcing Local and Global Compactness A. Bui, T. Le, H. Zhao, P. Montague, O. de Vel, T. Abraham, D. Phung arxiv code Uncertainty, Robustness, Safety
2020	NeurIPS OTLDA: A Geometry-aware Optimal Transport Approach for Topic Modeling V. Huynh, H. Zhao, D. Phung Text UnderstandingOptimal Transport
2020	AISTATS Variational Autoencoders for Sparse and Overdispersed Discrete Data H. Zhao, P. Rai, L. Du, W. Buntine, D. Phung, M. Zhou code Text UnderstandingBayesian Models
2018	AISTATS Bayesian multi-label learning with sparse features and labels, and label co-occurrences H. Zhao, P. Rai, L. Du, W. Buntine code Tabular DataBayesian Models
2018	NeurIPS Dirichlet Belief Networks for Topic Structure Learning H. Zhao, L. Du, W. Buntine, M. Zhou code Text UnderstandingBayesian Models
2018	ICML Inter and Intra Topic Structure Learning with Word Embeddings H. Zhao, L. Du, W. Buntine, M. Zhou code Text UnderstandingBayesian Models
2017	ACML A Word Embeddings Informed Focused Topic Model H. Zhao, L. Du, W. Buntine code Text UnderstandingBayesian Models
2017	ICML Leveraging Node Attributes for Incomplete Relational Data H. Zhao, L. Du, W. Buntine arxiv code Tabular DataBayesian Models
2017	ICDMLong Paper MetaLDA: A Topic Model that Efficiently Incorporates Meta information H. Zhao, L. Du, W. Buntine, G. Liu arxiv code Text UnderstandingBayesian Models

Community

Professional services

Reviewing

NeurIPS Area Chair (2025-); top reviewer in 2018
ICLR Area Chair (2025-)
ICML Area Chair (2026-)
Machine Learning Journal editorial board member (2023-)

Organising

Tutorial Chair Asian Conference on Machine Learning, 2026
Tutorial: Optimal Transport for Unsupervised Learning Asian Conference on Machine Learning, 2021
OT-SDMG: Workshop on Optimal Transport for Structured Data Modeling and Generation The Web Conference, 2025

Speaking

Recent talks

Beyond Predictions: Making AI Reliable with Uncertainty and Causal Understanding

UNSW AI Symposium , 2024

link

Enhancing Generalisation and Robustness in Deep Learning with Optimal Transport

School of Computer Science and Engineering, UNSW Sydney , July 2024

slides

Learning Deep Representations with Optimal Transport

Statistics Seminar, The University of Sydney