ML Anthology
Authors
Search
About
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