Bao, Yujia

15 publications

ICLR 2026 DRAGON: Guard LLM Unlearning in Context via Negative Detection and Reasoning Yaxuan Wang, Chris Yuhao Liu, Quan Liu, Jinlong Pang, Wei Wei, Yujia Bao, Yang Liu

ICLR 2026 MCP-Bench: Benchmarking Tool-Using LLM Agents with Complex Real-World Tasks via MCP Servers Zhenting Wang, Qi Chang, Hemani Patel, Shashank Biju, Cheng-En Wu, Quan Liu, Aolin Ding, Alireza Rezazadeh, Ankit Shah, Yujia Bao, Eugene Siow

ICLR 2025 From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs Alireza Rezazadeh, Zichao Li, Wei Wei, Yujia Bao

ICLR 2025 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

NeurIPS 2025 KVLink: Accelerating Large Language Models via Efficient KV Cache Reuse Jingbo Yang, Bairu Hou, Wei Wei, Yujia Bao, Shiyu Chang

ICLR 2025 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

TMLR 2025 Sample, Estimate, Aggregate: A Recipe for Causal Discovery Foundation Models Menghua Wu, Yujia Bao, Regina Barzilay, Tommi Jaakkola

ICLR 2024 Channel Vision Transformers: An Image Is Worth 1 X 16 X 16 Words Yujia Bao, Srinivasan Sivanandan, Theofanis Karaletsos

NeurIPSW 2024 From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs Alireza Rezazadeh, Zichao Li, Wei Wei, Yujia Bao

ICLRW 2024 Sample, Estimate, Aggregate: A Recipe for Causal Discovery Foundation Models Menghua Wu, Yujia Bao, Regina Barzilay, Tommi Jaakkola

NeurIPSW 2023 Channel Vision Transformers: An Image Is Worth C X 16 X 16 Words Yujia Bao, Srinivasan Sivanandan, Theofanis Karaletsos

ICML 2022 Learning Stable Classifiers by Transferring Unstable Features Yujia Bao, Shiyu Chang, Dr.Regina Barzilay

ICML 2021 Predict Then Interpolate: A Simple Algorithm to Learn Stable Classifiers Yujia Bao, Shiyu Chang, Regina Barzilay

ICLR 2020 Few-Shot Text Classification with Distributional Signatures Yujia Bao, Menghua Wu, Shiyu Chang, Regina Barzilay

MLHC 2017 Hawkes Process Modeling of Adverse Drug Reactions with Longitudinal Observational Data Yujia Bao, Zhaobin Kuang, Peggy Peissig, David Page, Rebecca Willett