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

Deriving Machine Attention from Human Rationales. - Yujia Bao; Shiyu Chang; Mo Yu; Regina Barzilay (EMNLP, 2018)

Using Machine Learning and Natural Language Processing to Review and Classify the Medical Literature on Cancer Susceptibility Genes. - Yujia Bao; Zhengyi Deng; Yan Wang; Heeyoon Kim; Victor Diego Armengol; Francisco Acevedo; Nofal Ouardaoui; Cathy Wang; Giovanni Parmigiani; Regina Barzilay; Danielle Braun; Kevin S. Hughes (CoRR, 2019)

Few-shot Text Classification with Distributional Signatures. - Yujia Bao; Menghua Wu; Shiyu Chang; Regina Barzilay (ICLR, 2020)

Predict then Interpolate: A Simple Algorithm to Learn Stable Classifiers. - Yujia Bao; Shiyu Chang; Regina Barzilay (ICML, 2021)

Learning Stable Classifiers by Transferring Unstable Features. - Yujia Bao; Shiyu Chang; Regina Barzilay (ICML, 2022)

Learning to Split for Automatic Bias Detection. - Yujia Bao; Regina Barzilay (CoRR, 2022)

Efficient and Robust Algorithms for Practical Machine Learning - Yujia Bao (2022)

Contextual Vision Transformers for Robust Representation Learning. - Yujia Bao; Theofanis Karaletsos (CoRR, 2023)

Channel Vision Transformers: An Image Is Worth C x 16 x 16 Words. - Yujia Bao; Srinivasan Sivanandan; Theofanis Karaletsos (CoRR, 2023)

Channel Vision Transformers: An Image Is Worth 1 x 16 x 16 Words. - Yujia Bao; Srinivasan Sivanandan; Theofanis Karaletsos (ICLR, 2024)

Harnessing Business and Media Insights with Large Language Models. - Yujia Bao; Ankit Parag Shah; Neeru Narang; Jonathan Rivers; Rajeev Maksey; Lan Guan; Louise N. Barrere; Shelley Evenson; Rahul Basole; Connie Miao; Ankit Mehta; Fabien Boulay; et al. (CoRR, 2024)

Improving Data Efficiency via Curating LLM-Driven Rating Systems. - Jinlong Pang; Jiaheng Wei; Ankit Parag Shah; Zhaowei Zhu; Yaxuan Wang; Chen Qian 001; Yang Liu 018; Yujia Bao; Wei Wei 019 (CoRR, 2024)

LLM Unlearning via Loss Adjustment with Only Forget Data. - Yaxuan Wang; Jiaheng Wei; Chris Yuhao Liu; Jinlong Pang; Quan Liu; Ankit Parag Shah; Yujia Bao; Yang Liu 018; Wei Wei 019 (CoRR, 2024)

Improving Data Efficiency via Curating LLM-Driven Rating Systems. - Jinlong Pang; Jiaheng Wei; Ankit Shah 001; Zhaowei Zhu; Yaxuan Wang; Chen Qian 001; Yang Liu 018; Yujia Bao; Wei Wei 019 (ICLR, 2025)

From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs. - Alireza Rezazadeh; Zichao Li; Wei Wei 019; Yujia Bao (ICLR, 2025)

LLM Unlearning via Loss Adjustment with Only Forget Data. - Yaxuan Wang; Jiaheng Wei; Chris Yuhao Liu; Jinlong Pang; Quan Liu; Ankit Shah 001; Yujia Bao; Yang Liu 018; Wei Wei 019 (ICLR, 2025)

Sample, estimate, aggregate: A recipe for causal discovery foundation models. - Menghua Wu; Yujia Bao; Regina Barzilay; Tommi S. Jaakkola (Trans. Mach. Learn. Res., 2025)

H-CoT: Hijacking the Chain-of-Thought Safety Reasoning Mechanism to Jailbreak Large Reasoning Models, Including OpenAI o1/o3, DeepSeek-R1, and Gemini 2.0 Flash Thinking. - Martin Kuo; Jianyi Zhang; Aolin Ding; Qinsi Wang; Louis DiValentin; Yujia Bao; Wei Wei 019; Hai Li 001; Yiran Chen 001 (CoRR, 2025)

KVLink: Accelerating Large Language Models via Efficient KV Cache Reuse. - Jingbo Yang; Bairu Hou; Wei Wei 019; Yujia Bao; Shiyu Chang (CoRR, 2025)

Enhancing Retrieval for ESGLLM via ESG-CID - A Disclosure Content Index Finetuning Dataset for Mapping GRI and ESRS. - Shafiuddin Rehan Ahmed; Ankit Parag Shah; Quan Hung Tran; Vivek Khetan; Sukryool Kang; Ankit Mehta; Yujia Bao; Wei Wei 019 (CoRR, 2025)

Enhancing Retrieval Systems with Inference-Time Logical Reasoning. - Felix Faltings; Wei Wei 019; Yujia Bao (CoRR, 2025)

Collaborative Memory: Multi-User Memory Sharing in LLM Agents with Dynamic Access Control. - Alireza Rezazadeh; Zichao Li; Ange Lou; Yuying Zhao; Wei Wei 019; Yujia Bao (CoRR, 2025)

Advertising in AI systems: Society must be vigilant. - Menghua Wu; Yujia Bao (CoRR, 2025)

SFT-GO: Supervised Fine-Tuning with Group Optimization for Large Language Models. - Gyuhak Kim; Sumiran Thakur; Su Min Park; Wei Wei; Yujia Bao (CoRR, 2025)