← Search

Jiangchao Yao

62 accepted papers

2026

Bilateral Information-aware Test-time Adaptation for Vision-Language Models

ICLR 2026poster

Test-time adaptation (TTA) fine-tunes models using new data encountered during inference, which enables the vision-language models to handle test data with covariant shifts. Unlike training-time adaptation, TTA does not require a test-distributed validation set or consider the worst-case distributio…

Cited by 0SourcecodeScholar
2026

Co-rewarding: Stable Self-supervised RL for Eliciting Reasoning in Large Language Models

ICLR 2026poster

Although reinforcement learning with verifiable rewards (RLVR) shows promise in improving the reasoning ability of large language models (LLMs), the scaling up dilemma remains due to the reliance on human-annotated labels especially for complex tasks. Recent self-rewarding methods provide a label-fr…

Cited by 0SourcecodeScholar
2026

DePO: Demonstration-guided Policy Optimization for Molecular Optimization

ICLR 2026poster

Large language models (LLMs) exhibit remarkable mathematical reasoning abilities through supervised fine-tuning (SFT) or reinforcement learning with verifiable rewards (RLVR). However, adapting them to scientific domains like molecular optimization is challenging: its datasets provide only reference…

Cited by 0SourceScholar
2026

Decoupling the Class Label and the Target Concept in Machine Unlearning

ICLR 2026poster

Machine unlearning as an emerging research topic for data regulations, aims to adjust a trained model to approximate a retrained one that excludes a portion of training data. Previous studies showed that class-wise unlearning is effective in forgetting the knowledge of a training class, either throu…

Cited by 0SourceScholar
2026

GenMask: Adapting DiT for Segmentation via Direct Mask Generation

CVPR 2026

Recent approaches for segmentation have leveraged pretrained generative models as feature extractors, treating segmentation as a downstream adaptation task via indirect feature retrieval. This implicit use suffers from a fundamental misalignment in representation.It also depends heavily on indirect

Cited by 0SourceScholar
2026

Improving Diffusion Models for Class-imbalanced Training Data via Capacity Manipulation

ICLR 2026oral

While diffusion models have achieved remarkable performance in image generation, they often struggle with the imbalanced datasets frequently encountered in real-world applications, resulting in significant performance degradation on minority classes. In this paper, we identify model capacity allocat…

Cited by 0SourceScholar
2026

One-Step Diffusion Transformer for Controllable Real-World Image Super-Resolution

CVPR 2026

Recent advances in diffusion-based real-world image super-resolution (Real-ISR) have demonstrated remarkable perceptual quality, yet the balance between fidelity and controllability remains a problem: multi-step diffusion-based methods suffer from generative diversity and randomness, resulting in lo

Cited by 0SourcecodeScholar
2026

Reason, Then Re-reason: Cross-view Revisiting Improves Spatial Reasoning

ICML 2026poster

Spatial reasoning from egocentric videos is inherently challenging because the observable evidence is constrained by the camera trajectory. Existing methods perform spatial reasoning in a single inference pass, forcing models to resolve geometric ambiguity through semantic priors rather than verifia…

Cited by 0SourceScholar
2026

Rejection Mixing: Fast Semantic Propagation of Mask Tokens for Efficient DLLM Inference

CVPR 2026

Diffusion Large Language Models (DLLMs) promise fast non-autoregressive inference but suffer a severe quality and speed tradeoff in parallel decoding. This stems from the "combinatorial contradiction" phenomenon, where parallel tokens form semantically inconsistent combinations. We address this by i

Cited by 0SourcecodeScholar
2026

Towards Understanding Valuable Preference Data for Large Language Model Alignment

ICLR 2026poster

Large language model (LLM) alignment is typically achieved through learning from human preference comparisons, making the quality of preference data critical to its success. Existing studies often pre-process raw training datasets to identify valuable preference pairs using external reward models or…

Cited by 0SourceScholar
2026

Wide-In, Narrow-Out: Revokable Decoding for Efficient and Effective DLLMs

ICLR 2026poster

Diffusion Large Language Models (DLLMs) have emerged as a compelling alternative to Autoregressive models, designed for fast parallel generation. However, existing DLLMs are plagued by a severe quality-speed trade-off, where faster parallel decoding leads to significant performance degradation. We a…

Cited by 0SourcecodeScholar
2025

4DGCPro: Efficient Hierarchical 4D Gaussian Compression for Progressive Volumetric Video Streaming

NeurIPS 2025poster

Achieving seamless viewing of high-fidelity volumetric video, comparable to 2D video experiences, remains an open challenge. Existing volumetric video compression methods either lack the flexibility to adjust quality and bitrate within a single model for efficient streaming across diverse networks a…

Cited by 0SourceScholar
2025

CateKV: On Sequential Consistency for Long-Context LLM Inference Acceleration

ICML 2025poster

Large language models (LLMs) have demonstrated strong capabilities in handling long-context tasks, but processing such long contexts remains challenging due to the substantial memory requirements and inference latency. In this work, we discover that certain attention heads exhibit sequential consist…

Cited by 0SourcePDFScholar
2025

Differential-informed Sample Selection Accelerates Multimodal Contrastive Learning

ICCV 2025poster

The remarkable success of contrastive-learning-based multimodal models has been greatly driven by training on ever-larger datasets with expensive compute consumption. Sample selection as an alternative efficient paradigm plays an important direction to accelerate the training process. However, recen…

2025

Fast and Accurate Blind Flexible Docking

ICLR 2025poster

Molecular docking that predicts the bound structures of small molecules (ligands) to their protein targets, plays a vital role in drug discovery. However, existing docking methods often face limitations: they either overlook crucial structural changes by assuming protein rigidity or suffer from low…

2025

From Passive to Active Reasoning: Can Large Language Models Ask the Right Questions under Incomplete Information?

ICML 2025poster

While existing benchmarks probe the reasoning abilities of large language models (LLMs) across diverse domains, they predominantly assess passive reasoning, providing models with all the information needed to reach a solution. By contrast, active reasoning—where an LLM must interact with external sy…

2025

LamRA: Large Multimodal Model as Your Advanced Retrieval Assistant

CVPR 2025poster

With the rapid advancement of multimodal information retrieval, increasingly complex retrieval tasks have emerged. Existing methods predominately rely on task-specific fine-tuning of vision-language models, often those trained with image-text contrastive learning. In this paper, we explore the possi…

Cited by 8SourcePDFScholar
2025

Large Language Models Enhanced Personalized Graph Neural Architecture Search in Federated Learning

AAAI 2025technical

Personalized federated learning (PFL) on graphs is an emerging field focusing on the collaborative development of architectures across multiple clients, each with distinct graph data distributions while adhering to strict privacy standards. This area often requires extensive expert intervention in m…

2025

Learning to Instruct for Visual Instruction Tuning

NeurIPS 2025poster

We propose L2T, an advancement of visual instruction tuning (VIT). While VIT equips Multimodal LLMs (MLLMs) with promising multimodal capabilities, the current design choices for VIT often result in overfitting and shortcut learning, potentially degrading performance. This gap arises from an overemp…

Cited by 7SourcecodeScholar
2025

MetaNeRV: Meta Neural Representations for Videos with Spatial-Temporal Guidance

AAAI 2025technical

Neural Representations for Videos (NeRV) has emerged as a promising implicit neural representation (INR) approach for video analysis, which represents videos as neural networks with frame indexes as inputs. However, NeRV-based methods are time-consuming when adapting to a large number of diverse v…

2025

MoMa: Modulating Mamba for Adapting Image Foundation Models to Video Recognition

ICML 2025poster

Video understanding is a complex challenge that requires effective modeling of spatial-temporal dynamics. With the success of image foundation models (IFMs) in image understanding, recent approaches have explored parameter-efficient fine-tuning (PEFT) to adapt IFMs for video. However, most of the…

Cited by 0SourcePDFScholar
2025

Multi-modal Medical Diagnosis via Large-small Model Collaboration

CVPR 2025poster

Recent advances in medical AI have shown a clear trend towards large models in healthcare. However, developing large models for multi-modal medical diagnosis remains challenging due to a lack of sufficient modal-complete medical data. Most existing multi-modal diagnostic models are relatively small…

Cited by 0SourcePDFScholar
2025

RAD: Towards Trustworthy Retrieval-Augmented Multi-modal Clinical Diagnosis

NeurIPS 2025poster

Clinical diagnosis is a highly specialized discipline requiring both domain expertise and strict adherence to rigorous guidelines. While current AI-driven medical research predominantly focuses on knowledge graphs or natural text pretraining paradigms to incorporate medical knowledge, these approac…

Cited by 0SourcecodeScholar
2025

SaFiRe: Saccade-Fixation Reiteration with Mamba for Referring Image Segmentation

NeurIPS 2025poster

Referring Image Segmentation (RIS) aims to segment the target object in an image given a natural language expression. While recent methods leverage pre-trained vision backbones and more training corpus to achieve impressive results, they predominantly focus on simple expressions—short, clear noun ph…

Cited by 0SourceScholar
2025

Towards Regularized Mixture of Predictions for Class-Imbalanced Semi-Supervised Facial Expression Recognition

IJCAI 2025

Semi-supervised facial expression recognition (SSFER) effectively assigns pseudo-labels to confident unlabeled samples when only limited emotional annotations are available. Existing SSFER methods are typically built upon an assumption of the class-balanced distribution. However, they are far from r

2024

Diversified Batch Selection for Training Acceleration

ICML 2024poster

The remarkable success of modern machine learning models on large datasets often demands extensive training time and resource consumption. To save cost, a prevalent research line, known as online batch selection, explores selecting informative subsets during the training process. Although recent eff…

2024

Domain-Inspired Sharpness-Aware Minimization Under Domain Shifts

ICLR 2024poster

This paper presents a Domain-Inspired Sharpness-Aware Minimization (DISAM) algorithm for optimization under domain shifts. It is motivated by the inconsistent convergence degree of SAM across different domains, which induces optimization bias towards certain domains and thus impairs the overall conv…

2024

Exploring Training on Heterogeneous Data with Mixture of Low-rank Adapters

ICML 2024poster

Training a unified model to take multiple targets into account is a trend towards artificial general intelligence. However, how to efficiently mitigate the training conflicts among heterogeneous data collected from different domains or tasks remains under-explored. In this study, we explore to lever…

2024

Less is More: One-shot Subgraph Reasoning on Large-scale Knowledge Graphs

ICLR 2024poster

To deduce new facts on a knowledge graph (KG), a link predictor learns from the graph structure and collects local evidence to find the answer to a given query. However, existing methods suffer from a severe scalability problem due to the utilization of the whole KG for prediction, which hinders the…

2024

Locally Estimated Global Perturbations are Better than Local Perturbations for Federated Sharpness-aware Minimization

ICML 2024spotlight

In federated learning (FL), the multi-step update and data heterogeneity among clients often lead to a loss landscape with sharper minima, degenerating the performance of the resulted global model. Prevalent federated approaches incorporate sharpness-aware minimization (SAM) into local training to m…

2024

Long-tailed Diffusion Models with Oriented Calibration

ICLR 2024poster

Diffusion models are acclaimed for generating high-quality and diverse images. However, their performance notably degrades when trained on data with a long-tailed distribution. For long tail diffusion model generation, current works focus on the calibration and enhancement of the tail generation wit…

2024

Low-Rank Knowledge Decomposition for Medical Foundation Models

CVPR 2024poster

The popularity of large-scale pre-training has promoted the development of medical foundation models. However some studies have shown that although foundation models exhibit strong general feature extraction capabilities their performance on specific tasks is still inferior to task-specific methods.…

2024

Mitigating Label Noise on Graphs via Topological Sample Selection

ICML 2024poster

Despite the success of the carefully-annotated benchmarks, the effectiveness of existing graph neural networks (GNNs) can be considerably impaired in practice when the real-world graph data is noisily labeled. Previous explorations in sample selection have been demonstrated as an effective way for r…

Cited by 8SourcePDFScholar
2024

Mitigating Noisy Correspondence by Geometrical Structure Consistency Learning

CVPR 2024poster

Noisy correspondence that refers to mismatches in cross-modal data pairs is prevalent on human-annotated or web-crawled datasets. Prior approaches to leverage such data mainly consider the application of uni-modal noisy label learning without amending the impact on both cross-modal and intra-modal g…

2024

Neural Atoms: Propagating Long-range Interaction in Molecular Graphs through Efficient Communication Channel

ICLR 2024poster

Graph Neural Networks (GNNs) have been widely adopted for drug discovery with molecular graphs. Nevertheless, current GNNs mainly excel in leveraging short-range interactions (SRI) but struggle to capture long-range interactions (LRI), both of which are crucial for determining molecular properties.…

2024

On Harmonizing Implicit Subpopulations

ICLR 2024poster

Machine learning algorithms learned from data with skewed distributions usually suffer from poor generalization, especially when minority classes matter as much as, or even more than majority ones. This is more challenging on class-balanced data that has some hidden imbalanced subpopulations, since…

Cited by 8SourcePDFScholar
2024

Pre-Post Interaction Learning for Brain Tumor Segmentation with Missing MRI Modalities

ICASSP 2024accepted

Complete multimodal Magnetic Resonance Imaging (MRI) plays an indispensable role in the task of brain tumor segmentation. However, the issue of missing-modality often arises in clinical practice, leading to a significant decline in the accuracy of segmentation. Current methods exhibit suboptimal per…

Cited by 0SourceScholar
2024

Probabilistic Conformal Distillation for Enhancing Missing Modality Robustness

NeurIPS 2024poster

Multimodal models trained on modality-complete data are plagued with severe performance degradation when encountering modality-missing data. Prevalent cross-modal knowledge distillation-based methods precisely align the representation of modality-missing data and that of its modality-complete counte…

2024

ReMamber: Referring Image Segmentation with Mamba Twister

ECCV 2024poster

"Referring Image Segmentation (RIS) leveraging transformers has achieved great success on the interpretation of complex visual-language tasks. However, the quadratic computation cost makes it resource-consuming in capturing long-range visual-language dependencies. Fortunately, Mamba addresses this w…

2024

Revive Re-weighting in Imbalanced Learning by Density Ratio Estimation

NeurIPS 2024poster

In deep learning, model performance often deteriorates when trained on highly imbalanced datasets, especially when evaluation metrics require robust generalization across underrepresented classes. To address the challenges posed by imbalanced data distributions, this study introduces a novel method…

Cited by 1SourcePDFScholar
2024

Self-Calibrated Tuning of Vision-Language Models for Out-of-Distribution Detection

NeurIPS 2024poster

Out-of-distribution (OOD) detection is crucial for deploying reliable machine learning models in open-world applications. Recent advances in CLIP-based OOD detection have shown promising results via regularizing prompt tuning with OOD features extracted from ID data. However, the irrelevant context…

2024

Unraveling the Impact of Heterophilic Structures on Graph Positive-Unlabeled Learning

ICML 2024poster

While Positive-Unlabeled (PU) learning is vital in many real-world scenarios, its application to graph data still remains under-explored. We unveil that a critical challenge for PU learning on graph lies on the edge heterophily, which directly violates the $\textit{irreducibility assumption}$ for $\…

Cited by 5SourcePDFScholar
2023

Combating Bilateral Edge Noise for Robust Link Prediction

NeurIPS 2023poster

Although link prediction on graphs has achieved great success with the development of graph neural networks (GNNs), the potential robustness under the edge noise is still less investigated. To close this gap, we first conduct an empirical study to disclose that the edge noise bilaterally perturbs bo…

2023

Combating Exacerbated Heterogeneity for Robust Models in Federated Learning

ICLR 2023poster

Privacy and security concerns in real-world applications have led to the development of adversarially robust federated models. However, the straightforward combination between adversarial training and federated learning in one framework can lead to the undesired robustness deterioration. We discover…

2023

Combating Representation Learning Disparity with Geometric Harmonization

NeurIPS 2023spotlight

Self-supervised learning (SSL) as an effective paradigm of representation learning has achieved tremendous success on various curated datasets in diverse scenarios. Nevertheless, when facing the long-tailed distribution in real-world applications, it is still hard for existing methods to capture tra…

2023

Diversified Outlier Exposure for Out-of-Distribution Detection via Informative Extrapolation

NeurIPS 2023poster

Out-of-distribution (OOD) detection is important for deploying reliable machine learning models on real-world applications. Recent advances in outlier exposure have shown promising results on OOD detection via fine-tuning model with informatively sampled auxiliary outliers. However, previous methods…

2023

Exploring Model Dynamics for Accumulative Poisoning Discovery

ICML 2023poster

Adversarial poisoning attacks pose huge threats to various machine learning applications. Especially, the recent accumulative poisoning attacks show that it is possible to achieve irreparable harm on models via a sequence of imperceptible attacks followed by a trigger batch. Due to the limited data-…

2023

Federated Domain Generalization With Generalization Adjustment

CVPR 2023poster

Federated Domain Generalization (FedDG) attempts to learn a global model in a privacy-preserving manner that generalizes well to new clients possibly with domain shift. Recent exploration mainly focuses on designing an unbiased training strategy within each individual domain. However, without the su…

2023

Federated Learning with Bilateral Curation for Partially Class-Disjoint Data

NeurIPS 2023poster

Partially class-disjoint data (PCDD), a common yet under-explored data formation where each client contributes a part of classes (instead of all classes) of samples, severely challenges the performance of federated algorithms. Without full classes, the local objective will contradict the global obje…

2023

Long-Tailed Partial Label Learning via Dynamic Rebalancing

ICLR 2023poster

Real-world data usually couples the label ambiguity and heavy imbalance, challenging the algorithmic robustness of partial label learning (PLL) and long-tailed learning (LT). The straightforward combination of LT and PLL, i.e., LT-PLL, suffers from a fundamental dilemma: LT methods build upon a give…

2023

NAS-LID: Efficient Neural Architecture Search with Local Intrinsic Dimension

AAAI 2023technical

One-shot neural architecture search (NAS) substantially improves the search efficiency by training one supernet to estimate the performance of every possible child architecture (i.e., subnet). However, the inconsistency of characteristics among subnets incurs serious interference in the optimization…

2023

On Strengthening and Defending Graph Reconstruction Attack with Markov Chain Approximation

ICML 2023poster

Although powerful graph neural networks (GNNs) have boosted numerous real-world applications, the potential privacy risk is still underexplored. To close this gap, we perform the first comprehensive study of graph reconstruction attack that aims to reconstruct the adjacency of nodes. We show that a…

2023

Uncovering Prototypical Knowledge for Weakly Open-Vocabulary Semantic Segmentation

NeurIPS 2023poster

This paper studies the problem of weakly open-vocabulary semantic segmentation (WOVSS), which learns to segment objects of arbitrary classes using mere image-text pairs. Existing works turn to enhance the vanilla vision transformer by introducing explicit grouping recognition, i.e., employing severa…

Cited by 29SourcePDFScholar
2023

Unleashing Mask: Explore the Intrinsic Out-of-Distribution Detection Capability

ICML 2023poster

Out-of-distribution (OOD) detection is an indispensable aspect of secure AI when deploying machine learning models in real-world applications. Previous paradigms either explore better scoring functions or utilize the knowledge of outliers to equip the models with the ability of OOD detection. Howeve…

2022

Contrastive Learning with Boosted Memorization

ICML 2022spotlight

Self-supervised learning has achieved a great success in the representation learning of visual and textual data. However, the current methods are mainly validated on the well-curated datasets, which do not exhibit the real-world long-tailed distribution. Recent attempts to consider self-supervised l…

2022

Reliable Adversarial Distillation with Unreliable Teachers

ICLR 2022poster

In ordinary distillation, student networks are trained with soft labels (SLs) given by pretrained teacher networks, and students are expected to improve upon teachers since SLs are stronger supervision than the original hard labels. However, when considering adversarial robustness, teachers may beco…

2021

Learning with Group Noise

AAAI 2021technical

Machine learning in the context of noise is a challenging but practical setting to plenty of real-world applications. Most of the previous approaches in this area focus on the pairwise relation (casual or correlational relationship) with noise, such as learning with noisy labels. However, the group…

Cited by 11SourcePDFScholar
2019

How does Disagreement Help Generalization against Label Corruption?

ICML 2019oral

Learning with noisy labels is one of the hottest problems in weakly-supervised learning. Based on memorization effects of deep neural networks, training on small-loss instances becomes very promising for handling noisy labels. This fosters the state-of-the-art approach "Co-teaching" that cross-train…

Cited by 975SourcePDFScholar
2018

Masking: A New Perspective of Noisy Supervision

NeurIPS 2018poster

It is important to learn various types of classifiers given training data with noisy labels. Noisy labels, in the most popular noise model hitherto, are corrupted from ground-truth labels by an unknown noise transition matrix. Thus, by estimating this matrix, classifiers can escape from overfitting…