Bao, Yujia

13 publications

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