← Search

Yue He

22 accepted papers

2026

Error Slice Discovery via Manifold Compactness

AAAI 2026technical

Despite the great performance of deep learning models in many areas, they still make mistakes and underperform on certain subsets of data, i.e. error slices. Given a trained model, it is important to identify its semantically coherent error slices that are easy to interpret, which is referred to as

Cited by 0SourcePDFScholar
2026

Generating Risky Samples with Conformity Constraints via Diffusion Models

AAAI 2026technical

Although neural networks achieve promising performance in many tasks, they may still fail when encountering some examples and bring about risks to applications. To discover risky samples, previous literature attempts to search for patterns of risky samples within existing datasets or inject perturba

Cited by 0SourcePDFScholar
2026

LogicSAGE: Neuro-Symbolic Reasoning with Socratic-Guided Enhancement

ICML 2026poster

Large Language Models (LLMs) often struggle with complex logical reasoning. Existing approaches typically rely on either purely neural reasoning in natural language or offloading to formal solvers via symbolic representations. However, both paradigms face significant limitations: while LLMs exhibit …

Cited by 0SourceScholar
2025

COUNTS: Benchmarking Object Detectors and Multimodal Large Language Models under Distribution Shifts

CVPR 2025highlight

Current object detectors often suffer significant performance degradation in real-world applications when encountering distributional shifts, posing serious risks in high-stakes domains such as autonomous driving and medical diagnosis. Consequently, the out-of-distribution (OOD) generalization capab…

Cited by 0SourcePDFScholar
2025

Environment Inference for Learning Generalizable Dynamical System

NeurIPS 2025spotlight

Data-driven methods offer efficient and robust solutions for analyzing complex dynamical systems but rely on the assumption of I.I.D. data, driving the development of generalization techniques for handling environmental differences. These techniques, however, are limited by their dependence on envir…

Cited by 0SourceScholar
2025

ODP-Bench: Benchmarking Out-of-Distribution Performance Prediction

ICCV 2025poster

Recently, there has been gradually more attention paid to Out-of-Distribution (OOD) performance prediction, whose goal is to predict the performance of trained models on unlabeled OOD test datasets, so that we could better leverage and deploy off-the-shelf trained models in risk-sensitive scenarios.…

2024

Domain-wise Data Acquisition to Improve Performance under Distribution Shift

ICML 2024poster

Despite notable progress in enhancing the capability of machine learning against distribution shifts, training data quality remains a bottleneck for cross-distribution generalization. Recently, from a data-centric perspective, there have been considerable efforts to improve model performance through…

2024

Full Bayesian Significance Testing for Neural Networks in Traffic Forecasting

IJCAI 2024poster

Due to the complex and dynamic traffic contexts, the interpretability and uncertainty of traffic forecasting have gained increasing attention. Significance testing is a powerful tool in statistics used to determine whether a hypothesis is valid, facilitating the identification of pivotal features th…

Cited by 6SourcePDFScholar
2024

Rethinking the Evaluation Protocol of Domain Generalization

CVPR 2024poster

Domain generalization aims to solve the challenge of Out-of-Distribution (OOD) generalization by leveraging common knowledge learned from multiple training domains to generalize to unseen test domains. To accurately evaluate the OOD generalization ability it is required that test data information is…

2023

CFCG: Semi-Supervised Semantic Segmentation via Cross-Fusion and Contour Guidance Supervision

ICCV 2023poster

Current state-of-the-art semi-supervised semantic segmentation (SSSS) methods typically adopt pseudo labeling and consistency regularization between multiple learners with different perturbations. Although the performance is desirable, many issues remain: (1) supervisions from a single learner tend…

Cited by 16PDFScholar
2023

Covariate-Shift Generalization via Random Sample Weighting

AAAI 2023technical

Shifts in the marginal distribution of covariates from training to the test phase, named covariate-shifts, often lead to unstable prediction performance across agnostic testing data, especially under model misspecification. Recent literature on invariant learning attempts to learn an invariant predi…

Cited by 6SourcePDFScholar
2023

MAP: Towards Balanced Generalization of IID and OOD through Model-Agnostic Adapters

ICCV 2023oral

Deep learning has achieved tremendous success in recent years, but most of these successes are built on an independent and identically distributed (IID) assumption. This somewhat hinders the application of deep learning to the more challenging out-of-distribution (OOD) scenarios. Although many OOD m…

Cited by 22PDFcodeScholar
2023

NICO++: Towards Better Benchmarking for Domain Generalization

CVPR 2023poster

Despite the remarkable performance that modern deep neural networks have achieved on independent and identically distributed (I.I.D.) data, they can crash under distribution shifts. Most current evaluation methods for domain generalization (DG) adopt the leave-one-out strategy as a compromise on the…

2023

Stable Learning via Sparse Variable Independence

AAAI 2023technical

The problem of covariate-shift generalization has attracted intensive research attention. Previous stable learning algorithms employ sample reweighting schemes to decorrelate the covariates when there is no explicit domain information about training data. However, with finite samples, it is difficul…

Cited by 17SourcePDFScholar
2023

TransVCL: Attention-Enhanced Video Copy Localization Network with Flexible Supervision

AAAI 2023technical

Video copy localization aims to precisely localize all the copied segments within a pair of untrimmed videos in video retrieval applications. Previous methods typically start from frame-to-frame similarity matrix generated by cosine similarity between frame-level features of the input video pair, an…

2022

StylizedNeRF: Consistent 3D Scene Stylization As Stylized NeRF via 2D-3D Mutual Learning

CVPR 2022poster

3D scene stylization aims at generating stylized images of the scene from arbitrary novel views following a given set of style examples, while ensuring consistency when rendered from different views. Directly applying methods for image or video stylization to 3D scenes cannot achieve such consistenc…

Cited by 172PDFcodeScholar
2022

Visual Semantics Allow for Textual Reasoning Better in Scene Text Recognition

AAAI 2022technical

Existing Scene Text Recognition (STR) methods typically use a language model to optimize the joint probability of the 1D character sequence predicted by a visual recognition (VR) model, which ignore the 2D spatial context of visual semantics within and between character instances, making them not ge…

2021

Deep Stable Learning for Out-of-Distribution Generalization

CVPR 2021poster

Approaches based on deep neural networks have achieved striking performance when testing data and training data share similar distribution, but can significantly fail otherwise. Therefore, eliminating the impact of distribution shifts between training and testing data is crucial for building perform…

Cited by 347PDFcodeScholar
2020

Counterfactual Prediction for Bundle Treatment

NeurIPS 2020poster

Estimating counterfactual outcome of different treatments from observational data is an important problem to assist decision making in a variety of fields. Among the various forms of treatment specification, bundle treatment has been widely adopted in many scenarios, such as recommendation systems…