Selected Research

Selected research in reversed chronological order. For the full list of my research, please refer to my Google Scholar.
Co-first authors ⁺; Coresponding authors ^*.

My previous research contributes to enhancing the reliability and robustness of AI for healthcare, centering on developing weakly supervised learning frameworks to perform robust data mining from imperfect datasets and designing uncertainty-aware machine learning strategies to enhance model reliability, optimize decision making under uncertainty, and improve overall prediction performance.

Machine Learning Model Development/Optimization

• Efficient Uncertainty Quantification and Reduction for Over-Parameterized Neural Networks
  Ziyi Huang, Henry Lam, Haofeng Zhang
  Advances in Neural Information Processing Systems (NeurIPS), 2023

  Previous manuscript: Quantifying Epistemic Uncertainty in Deep Learning

• Conditional Coverage Estimation for High-Quality Prediction Intervals
  Ziyi Huang, Henry Lam, Haofeng Zhang
  Journal of Systems Science and Systems Engineering

• Learning Prediction Intervals for Regression: Generalization and Calibration
  Haoxian Chen, Ziyi Huang, Henry Lam, Huajie Qian, Haofeng Zhang
  International Conference on Artificial Intelligence and Statistics (AISTATS), 2021

• Evaluating Aleatoric Uncertainty via Conditional Generative Models
  Ziyi Huang, Henry Lam, Haofeng Zhang
  Manuscript
  

Data-Driven Decision Making

• Bayesian Bandit Algorithms with Approximate Inference in Stochastic Linear Bandits
  Ziyi Huang, Henry Lam, Haofeng Zhang
  Under Review
  

• Optimal Regret Is Achievable with Bounded Approximate Inference Error: An Enhanced Bayesian Upper Confidence Bound Framework
  Ziyi Huang, Henry Lam, Amirhossein Meisami, Haofeng Zhang
  Advances in Neural Information Processing Systems (NeurIPS), 2023

Learning from Data Imperfections

• Cardiac Adipose Tissue Segmentation via Image-Level Annotations
  Ziyi Huang, Yu Gan, Theresa Lye, Yanchen Liu, Haofeng Zhang, Andrew Laine, Elsa Angelini, Christine Hendon
  IEEE Journal of Biomedical and Health Informatics (JBHI), 2023

• Co-Seg: An Image Segmentation Framework against Label Corruption
  Ziyi Huang, Haofeng Zhang, Andrew Laine, Elsa Angelini, Christine Hendon, Yu Gan
  IEEE International Symposium on Biomedical Imaging (ISBI), 2021

• Heterogeneity Measurement of Cardiac Tissues Leveraging Uncertainty Information from Image Segmentation
  Ziyi Huang, Yu Gan, Theresa Lye, Haofeng Zhang, Andrew Laine, Elsa Angelini, Christine Hendon
  International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), 2020

• Push the Boundary of SAM: A Pseudo-label Correction Framework for Medical Segmentation
  Ziyi Huang⁺, Hongshan Liu⁺, Haofeng Zhang, Xueshen Li, Haozhe Liu, Fuyong Xing, Andrew Laine, Elsa Angelini, Christine Hendon, Yu Gan
  Manuscript
  

General AI for Health

Reinforcement Learning Based Proactive Medical Dialogue System for Health Status and Medical Image Collection
Xueshen Li, Xinlong Hou, Ziyi Huang*, Yu Gan*
To be presented at SPIE Medical Imaging (MI), 2025

• Method, System and Computer-Accessible Medium Extracting and Analyzing Electrogram Features Using Artificial Intelligence to Predict Successful Treatment Site(s)
  Christine Hendon, Deepak Saluja, Ziyi Huang
  US Patent App. 18/665,782 2024
  

• Automated Prediction of Isthmus Areas in Scar‐Related Atrial Tachycardias Using Artificial Intelligence
  Deepak Saluja, Ziyi Huang, Jonah Majumder, Lawrence Zeldin, Hirad Yarmohammadi, Angelo Biviano, Elaine Y Wan, Edward J Ciaccio, Christine P Hendon, Hasan Garan
  Journal of Cardiovascular Electrophysiology, 2024

• Automated Analysis Framework for In Vivo Cardiac Ablation Therapy Monitoring with Optical Coherence Tomography
  Ziyi Huang, Xiaowei Zhao, Ohad Ziv, Kenneth R. Laurita, Andrew M. Rollins, and Christine Hendon
  Biomedical Optics Express , 2023

• Segmentation and Uncertainty Measures of Cardiac Substrates within Optical Coherence Tomography Images via Convolutional Neural Networks
  Ziyi Huang, Yu Gan, Theresa Lye, Darnel Theagene, Spandana Chintapalli, Simeran Virdi, Andrew Laine, Elsa Angelini, Christine P Hendon
  IEEE International Symposium on Biomedical Imaging (ISBI) , 2020