Continual Training of Language Models for Few-Shot Learning. - Zixuan Ke; Haowei Lin; Yijia Shao; Hu Xu 001; Lei Shu 004; Bing Liu 001 (EMNLP, 2022)

Adapting a Language Model While Preserving its General Knowledge. - Zixuan Ke; Yijia Shao; Haowei Lin; Hu Xu 001; Lei Shu 004; Bing Liu 001 (EMNLP, 2022)

FormLM: Recommending Creation Ideas for Online Forms by Modelling Semantic and Structural Information. - Yijia Shao; Mengyu Zhou; Yifan Zhong; Tao Wu; Hongwei Han; Shi Han; Gideon Huang; Dongmei Zhang 001 (EMNLP, 2022)

CMG: A Class-Mixed Generation Approach to Out-of-Distribution Detection. - Mengyu Wang 002; Yijia Shao; Haowei Lin; Wenpeng Hu; Bing Liu 001 (ECML/PKDD (4), 2022)

Inferring Tabular Analysis Metadata by Infusing Distribution and Knowledge Information. - Xinyi He; Mengyu Zhou; Jialiang Xu; Xiao Lv; Tianle Li; Yijia Shao; Shi Han; Zejian Yuan; Dongmei Zhang 001 (CoRR, 2022)

LUNA: Language Understanding with Number Augmentations on Transformers via Number Plugins and Pre-training. - Hongwei Han; Jialiang Xu; Mengyu Zhou; Yijia Shao; Shi Han; Dongmei Zhang 001 (CoRR, 2022)

ACCENT: An Automatic Event Commonsense Evaluation Metric for Open-Domain Dialogue Systems. - Sarik Ghazarian; Yijia Shao; Rujun Han; Aram Galstyan; Nanyun Peng 001 (ACL (1), 2023)

AnaMeta: A Table Understanding Dataset of Field Metadata Knowledge Shared by Multi-dimensional Data Analysis Tasks. - Xinyi He; Mengyu Zhou; Mingjie Zhou; Jialiang Xu; Xiao Lv; Tianle Li; Yijia Shao; Shi Han; Zejian Yuan; Dongmei Zhang 001 (ACL (Findings), 2023)

Class-Incremental Learning based on Label Generation. - Yijia Shao; Yiduo Guo; Dongyan Zhao 001; Bing Liu 001 (ACL (2), 2023)

Continual Pre-training of Language Models. - Zixuan Ke; Yijia Shao; Haowei Lin; Tatsuya Konishi; Gyuhak Kim; Bing Liu 001 (ICLR, 2023)

Continual Learning of Language Models. - Zixuan Ke; Yijia Shao; Haowei Lin; Tatsuya Konishi; Gyuhak Kim; Bing Liu 001 (CoRR, 2023)

Into the Unknown Unknowns: Engaged Human Learning through Participation in Language Model Agent Conversations. - Yucheng Jiang; Yijia Shao; Dekun Ma; Sina J. Semnani; Monica S. Lam (EMNLP, 2024)

Class Incremental Learning via Likelihood Ratio Based Task Prediction. - Haowei Lin; Yijia Shao; Weinan Qian; Ningxin Pan; Yiduo Guo; Bing Liu 001 (ICLR, 2024)

Assisting in Writing Wikipedia-like Articles From Scratch with Large Language Models. - Yijia Shao; Yucheng Jiang; Theodore A. Kanell; Peter Xu 003; Omar Khattab; Monica S. Lam (NAACL-HLT, 2024)

PrivacyLens: Evaluating Privacy Norm Awareness of Language Models in Action. - Yijia Shao; Tianshi Li 006; Weiyan Shi; Yanchen Liu; Diyi Yang (NeurIPS, 2024)

Quiet-STaR: Language Models Can Teach Themselves to Think Before Speaking. - Eric Zelikman; Georges Harik; Yijia Shao; Varuna Jayasiri; Nick Haber; Noah D. Goodman (CoRR, 2024)

Show, Don't Tell: Aligning Language Models with Demonstrated Feedback. - Omar Shaikh; Michelle S. Lam; Joey Hejna; Yijia Shao; Michael S. Bernstein; Diyi Yang (CoRR, 2024)

Collaborative Gym: A Framework for Enabling and Evaluating Human-Agent Collaboration. - Yijia Shao; Vinay Samuel; Yucheng Jiang; John Yang 002; Diyi Yang (CoRR, 2024)

Aligning Language Models with Demonstrated Feedback. - Omar Shaikh; Michelle S. Lam; Joey Hejna; Yijia Shao; Hyundong Justin Cho; Michael S. Bernstein; Diyi Yang (ICLR, 2025)

Personalization of Large Language Models: A Survey. - Zhehao Zhang 001; Ryan A. Rossi; Branislav Kveton; Yijia Shao; Diyi Yang; Hamed Zamani; Franck Dernoncourt; Joe Barrow; Tong Yu 001; Sungchul Kim; Ruiyi Zhang 002; Jiuxiang Gu; et al. (Trans. Mach. Learn. Res., 2025)

Challenges and Paths Towards AI for Software Engineering. - Alex Gu; Naman Jain; Wen-Ding Li; Manish Shetty; Yijia Shao; Ziyang Li; Diyi Yang; Kevin Ellis; Koushik Sen; Armando Solar-Lezama (CoRR, 2025)

Future of Work with AI Agents: Auditing Automation and Augmentation Potential across the U.S. Workforce. - Yijia Shao; Humishka Zope; Yucheng Jiang; Jiaxin Pei; David Nguyen; Erik Brynjolfsson; Diyi Yang (CoRR, 2025)