← Search

Nevin Zhang

6 accepted papers

2026

Causal Fine-Tuning under Latent Confounded Shift

ICML 2026poster

Adapting to latent confounded shift remains a core challenge in modern AI. This setting is driven by hidden variables that induce spurious correlations between inputs and outputs during training, leading models to rely on non-causal shortcuts. For example, a model may learn to treat metadata (e.g., …

Cited by 0SourceScholar
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…

2022

Adaptive Label Smoothing with Self-Knowledge in Natural Language Generation

EMNLP 2022main

Overconfidence has been shown to impair generalization and calibration of a neural network. Previous studies remedy this issue by adding a regularization term to a loss function, preventing a model from making a peaked distribution. Label smoothing smoothes target labels with a pre-defined prior lab…

2022

Hard Gate Knowledge Distillation - Leverage Calibration for Robust and Reliable Language Model

EMNLP 2022main

In knowledge distillation, a student model is trained with supervisions from both knowledge from a teacher and observations drawn from a training data distribution. Knowledge of a teacher is considered a subject that holds inter-class relations which send a meaningful supervision to a student; hence…

Cited by 3SourcePDFScholar
2022

Improving Meta-learning for Low-resource Text Classification and Generation via Memory Imitation

ACL 2022long

Building models of natural language processing (NLP) is challenging in low-resource scenarios where limited data are available. Optimization-based meta-learning algorithms achieve promising results in low-resource scenarios by adapting a well-generalized model initialization to handle new tasks. Non…

Cited by 29SourcePDFScholar
2022

SeqPATE: Differentially Private Text Generation via Knowledge Distillation

NeurIPS 2022accept

Protecting the privacy of user data is crucial for text generation models, which can leak sensitive information during generation. Differentially private (DP) learning methods provide guarantees against identifying the existence of a training sample from model outputs. PATE is a recent DP learning a…

Cited by 24SourcePDFScholar