← Search

Weiyan Xie

4 accepted papers

2024

Consistency Regularization for Domain Generalization with Logit Attribution Matching

UAI 2024poster

Domain generalization (DG) is about training models that generalize well under domain shift. Previous research on DG has been conducted mostly in single-source or multi-source settings. In this paper, we consider a third lesser-known setting where a training domain is endowed with a collection of pa…

2024

Dual Risk Minimization: Towards Next-Level Robustness in Fine-tuning Zero-Shot Models

NeurIPS 2024poster

Fine-tuning foundation models often compromises their robustness to distribution shifts. To remedy this, most robust fine-tuning methods aim to preserve the pre-trained features. However, not all pre-trained features are robust and those methods are largely indifferent to which ones to preserve. We…

2023

Two-stage holistic and contrastive explanation of image classification

UAI 2023poster

The need to explain the output of a deep neural network classifier is now widely recognized. While previous methods typically explain a single class in the output, we advocate explaining the whole output, which is a probability distribution over multiple classes. A whole-output explanation can help…

2023

ViT-CX: Causal Explanation of Vision Transformers

IJCAI 2023poster

Despite the popularity of Vision Transformers (ViTs) and eXplainable AI (XAI), only a few explanation methods have been designed specially for ViTs thus far. They mostly use attention weights of the [CLS] token on patch embeddings and often produce unsatisfactory saliency maps. This paper proposes a…