Profile image
Zhitong Gao 郜之桐
Final-Year CS Master Student

About Me

Hi there! I am a final-year master’s student in Computer Science at ShanghaiTech University, under the guidance of Prof. Xuming He. Currently, I am also a visiting student at EPFL CVLab, working with Dr. Mathieu Salzmann.

I am deeply enthusiastic about the reliability and safety aspects of machine learning. My research interests include

(1) uncovering the underlying reasons behind model failures,

(2) devising strategies to enhance model robustness against identified causes, and

(3) developing methods for enabling models to signal 'I do not know' in the face of inevitable failures.

My work has encompassed research on uncertainty estimation, robust learning with noisy labels/distribution shift, and test-time adaptation. I am interested in tackling these challenges within real-world applications. My previous studies have focused on medical image analysis and autonomous driving scenarios. Recently, my interests have expanded to exploring these issues for large language models.

I am always eager to collaborate and engage in discussions. Feel free to reach out if you share similar interests or wish to discuss my work. 😊

News

  • 12/2023: I attended the NeurIPS 2023 in New Orleans and gave a poster presentation.
  • 11/2023: I received the China National Scholarship (rate 0.2%)!
  • 11/2023: One paper was accepted by ML4H 2023. Congrats to Bingnan!
  • 10/2023: I gave both an oral and a poster presentation at the ICCV 2023 UNCV Workshop in Paris.
  • 09/2023: One paper was accepted by NeurIPS 2023.
  • 09/2023: I have joined the EPFL CV lab as a visiting scholar, working with Mathieu Salzmann.
  • 08/2023: I served as a reviewer for NeurIPS 2023, reviewed 2 papers.
  • 08/2023: I attended the IJCAI conference onsite in Macau and gave both an oral and a poster presentation.
  • 05/2023: I attended the ICLR conference onsite in Rwanda and gave a poster presentation.
  • 04/2023: One paper was accepted by IJCAI 2023. Congrats to Chuanyang!
  • 01/2023: I served as a reviewer for CVPR 2023, reviewed 2 papers.
  • 01/2023: One paper was accepted by ICLR 2023.
  • 10/2021: I attended the MICCAI conference online and gave a poster presentation.
  • 07/2021: I finished my undergraduate studies and received an outstanding undergraduate thesis award!
  • 05/2021: One paper was accepted by MICCAI 2021.

Publications

ATTA: Anomaly-aware Test-Time Adaptation for Out-of-Distribution Detection in Segmentation
[PDF] [Code] [Poster] [Slides]
Zhitong Gao, Shipeng Yan, Xuming He
Conference on Neural Information Processing Systems (NeurIPS), 2023
Out-of-Distribution Detection Anomaly Segmentation Test-Time Adaptation
Gradient-Map-Guided Adaptive Domain Generalization for Cross Modality MRI Segmentation
[PDF] [Code]
Bingnan Li, Zhitong Gao, Xuming He
Machine Learning for Health (ML4H), 2023
Domain Generalization Test-Time Adaptation Medical Image Segmentation
Modeling Multimodal Aleatoric Uncertainty in Segmentation with Mixture of Stochastic Experts
[PDF] [Code] [Poster] [Video]
Zhitong Gao, Yucong Chen, Chuyu Zhang, Xuming He
International Conference on Learning Representations (ICLR), 2023
Aleatoric Uncertainty Estimation Stochastic Segmentation Generative Model
MILD: Modeling the Instance Learning Dynamics for Learning with Noisy Labels
[PDF] [Code]
Chuanyang Hu, Shipeng Yan, Zhitong Gao, Xuming He
International Joint Conference on Artificial Intelligence (IJCAI), 2023
Robust learning Learning Dynamics
Superpixel-guided Iterative Learning from Noisy Labels for Medical Image Segmentation
[PDF] [Code] [Poster] [Video]
Shuailin Li*, Zhitong Gao*, Xuming He
International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI), 2021
Robust learning Medical Image Segmentation
Note: * indicates equal contribution.

Challenges and Workshops

MUAD Uncertainty Estimation for Semantic Segmentation Challenge at ICCV2023
[Challenge Website] [Report] [Slides]
Zhitong Gao, Xuming He
Top 7; Invited for oral and poster presentation at ICCV UNCV Workshop.
Out-of-distribution Detection Uncertainty Estimation Test-Time Adaptation
Quantification of Uncertainties in Biomedical Image Quantification Challenge at MICCAI2021
[Challenge Website] [Report] [Poster]
Yucong Chen, Guanqi He, Zhitong Gao, Xuming He
Invited for oral and poster presentation at MICCAI 2021 Workshop.
Uncertainty Estimation Medical Image Segmentation
-->

Course Projects

Understanding and Exploration for "Adaptive Online Learning in Dynamic Environments"
[Report]
Zhitong Gao*, Haotian Tian*, Tianyue Zhou*
Final Project of CS245 Online Optimization and Learning, 2023 Spring, ShanghaiTech University
Online Learning Optimization Non-stationary Environment
Generalized DUQ: Generalized Deterministic Uncertainty Quantification
[Report]
Zhitong Gao*, Ziyao Zeng*
Final Project of CS282 Machine Learning, 2021 Spring, ShanghaiTech University
Uncertainty Quantification Aleatoric Uncertainty Epistemic Uncertainty
Seek Common while Shelving Differences: A New Way for dealing with Noisy Labels
[Report]
Zhitong Gao*, Ziyao Zeng*
Final Project of CS280 Deep Learning, 2020 Fall, ShanghaiTech University
Robust Learning Noisy label

Education

M.Sc. in School of Information Science and Technology, ShanghaiTech University, Shanghai, China
Sep. 2021 - present
B.E. in School of Information Science and Technology, ShanghaiTech University, Shanghai, China
Sep. 2017 - Jun. 2021

Services

  • Reviewer of CVPR 2023, NeurIPS 2023, CVPR 2024, ECCV 2024
  • Teaching Assistant of CS280 Deep Learning, 2022 Fall, ShanghaiTech University
Copyright © Zhitong Gao 2023 (Thanks to Shuaiyi Huang)

Last Updated: