| 2026 | ACL | Observations and Remedies for Large Language Model Bias in Self-Consuming Performative Loop. | Yaxuan Wang, Zhongteng Cai, Yujia Bao, Xueru Zhang, Yang Liu |
| 2026 | ACL | WebDART: Dynamic Decomposition and Re-planning for Complex Web Tasks. | Jingbo Yang, Bairu Hou, Wei Wei, Shiyu Chang, Yujia Bao |
| 2025 | ACL | Enhancing Retrieval Systems with Inference-Time Logical Reasoning. | Felix Faltings, Wei Wei, Yujia Bao |
| 2025 | ICLR | Improving Data Efficiency via Curating LLM-Driven Rating Systems. | Jinlong Pang, Jiaheng Wei, Ankit Shah, Zhaowei Zhu, Yaxuan Wang, Chen Qian, Yang Liu, Yujia Bao, Wei Wei |
| 2025 | ICLR | From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs. | Alireza Rezazadeh, Zichao Li, Wei Wei, Yujia Bao |
| 2025 | ICLR | LLM Unlearning via Loss Adjustment with Only Forget Data. | Yaxuan Wang, Jiaheng Wei, Chris Yuhao Liu, Jinlong Pang, Quan Liu, Ankit Shah, Yujia Bao, Yang Liu, Wei Wei |
| 2024 | ICLR | Channel Vision Transformers: An Image Is Worth 1 x 16 x 16 Words. | Yujia Bao, Srinivasan Sivanandan, Theofanis Karaletsos |
| 2022 | ICML | Learning Stable Classifiers by Transferring Unstable Features. | Yujia Bao, Shiyu Chang, Regina Barzilay |
| 2021 | ICML | Predict then Interpolate: A Simple Algorithm to Learn Stable Classifiers. | Yujia Bao, Shiyu Chang, Regina Barzilay |
| 2020 | ICLR | Few-shot Text Classification with Distributional Signatures. | Yujia Bao, Menghua Wu, Shiyu Chang, Regina Barzilay |
| 2018 | EMNLP | Deriving Machine Attention from Human Rationales. | Yujia Bao, Shiyu Chang, Mo Yu, Regina Barzilay |