← Search

Wanqian Yang

3 accepted papers

2025

Learning Is Not A Race: Improving Retrieval in Language Models via Equal Learning

EMNLP 2025

Many applications that modern large language models (LLMs) are deployed on are retrieval tasks: the answer can be recovered from context and success is a matter of learning generalizable features from data. However, this is easier said than done. Overparametrized models trained on cross-entropy loss

Cited by 0SourcePDFScholar
2022

Chroma-VAE: Mitigating Shortcut Learning with Generative Classifiers

NeurIPS 2022accept

Deep neural networks are susceptible to shortcut learning, using simple features to achieve low training loss without discovering essential semantic structure. Contrary to prior belief, we show that generative models alone are not sufficient to prevent shortcut learning, despite an incentive to reco…

Cited by 15SourcePDFScholar
2020

Incorporating Interpretable Output Constraints in Bayesian Neural Networks

NeurIPS 2020spotlight

Domains where supervised models are deployed often come with task-specific constraints, such as prior expert knowledge on the ground-truth function, or desiderata like safety and fairness. We introduce a novel probabilistic framework for reasoning with such constraints and formulate a prior that ena…