← Search

Zhitong Gao

7 accepted papers

2026

(1D) Ordered Tokens Enable Efficient Test-Time Search

ICML 2026poster

Tokenization is a key component of autoregressive generative models, converting raw data into more manageable units for modeling. Commonly, tokens describe local information, such as regions of pixels in images or word pieces in text, and autoregressive generation commonly predicts these tokens in a…

Cited by 0SourceScholar
2026

MODUS: Decoder-only Any-to-Any Modeling of Diverse Modalities

ICML 2026poster

Any-to-any modeling aims to flexibly relate arbitrary modalities within a single system, a requirement that arises across multimodal learning and scientific domains such as ecology and astronomy. However, existing any-to-any approaches are typically trained from scratch using encoder–decoder or diff…

Cited by 0SourceScholar
2025

Large (Vision) Language Models are Unsupervised In-Context Learners

ICLR 2025poster

Recent advances in large language and vision-language models have enabled zero-shot inference, allowing models to solve new tasks without task-specific training. Various adaptation techniques such as prompt engineering, In-Context Learning (ICL), and supervised fine-tuning can further enhance the mo…

2024

Generalize or Detect? Towards Robust Semantic Segmentation Under Multiple Distribution Shifts

NeurIPS 2024poster

In open-world scenarios, where both novel classes and domains may exist, an ideal segmentation model should detect anomaly classes for safety and generalize to new domains. However, existing methods often struggle to distinguish between domain-level and semantic-level distribution shifts, leading to…

2023

ATTA: Anomaly-aware Test-Time Adaptation for Out-of-Distribution Detection in Segmentation

NeurIPS 2023poster

Recent advancements in dense out-of-distribution (OOD) detection have primarily focused on scenarios where the training and testing datasets share a similar domain, with the assumption that no domain shift exists between them. However, in real-world situations, domain shift often exits and significa…

2023

MILD: Modeling the Instance Learning Dynamics for Learning with Noisy Labels

IJCAI 2023poster

Despite deep learning has achieved great success, it often relies on a large amount of training data with accurate labels, which are expensive and time-consuming to collect. A prominent direction to reduce the cost is to learn with noisy labels, which are ubiquitous in the real-world applications. A…

2023

Modeling Multimodal Aleatoric Uncertainty in Segmentation with Mixture of Stochastic Experts

ICLR 2023poster

Equipping predicted segmentation with calibrated uncertainty is essential for safety-critical applications. In this work, we focus on capturing the data-inherent uncertainty (aka aleatoric uncertainty) in segmentation, typically when ambiguities exist in input images. Due to the high-dimensional out…