| 2025 | COLING | Embedding-Informed Adaptive Retrieval-Augmented Generation of Large Language Models. | Chengkai Huang, Yu Xia, Rui Wang, Kaige Xie, Tong Yu, Julian J. McAuley, Lina Yao |
| 2025 | COLING | Making Large Language Models into World Models with Precondition and Effect Knowledge. | Kaige Xie, Ian Yang, John Gunerli, Mark O. Riedl |
| 2025 | NAACL | Do RAG Systems Cover What Matters? Evaluating and Optimizing Responses with Sub-Question Coverage. | Kaige Xie, Philippe Laban, Prafulla Kumar Choubey, Caiming Xiong, Chien-Sheng Wu |
| 2024 | EACL | Creating Suspenseful Stories: Iterative Planning with Large Language Models. | Kaige Xie, Mark O. Riedl |
| 2024 | EACL | Few-Shot Dialogue Summarization via Skeleton-Assisted Prompt Transfer in Prompt Tuning. | Kaige Xie, Tong Yu, Haoliang Wang, Junda Wu, Handong Zhao, Ruiyi Zhang, Kanak Mahadik, Ani Nenkova, Mark O. Riedl |
| 2024 | ICASSP | Exploring Soft Prompt Initialization Strategy for Few-Shot Continual Text Classification. | Zhehao Zhang, Tong Yu, Handong Zhao, Kaige Xie, Lina Yao, Shuai Li |
| 2024 | WSDM | Interact with the Explanations: Causal Debiased Explainable Recommendation System. | Xu Liu, Tong Yu, Kaige Xie, Junda Wu, Shuai Li |
| 2022 | EMNLP | Guiding Neural Story Generation with Reader Models. | Xiangyu Peng, Kaige Xie, Amal Alabdulkarim, Harshith Kayam, Samihan Dani, Mark O. Riedl |
| 2022 | EMNLP | Calibrating Trust of Multi-Hop Question Answering Systems with Decompositional Probes. | Kaige Xie, Sarah Wiegreffe, Mark O. Riedl |
| 2019 | NAACL | Rethinking Action Spaces for Reinforcement Learning in End-to-end Dialog Agents with Latent Variable Models. | Tiancheng Zhao, Kaige Xie, Maxine Esknazi |
| 2018 | EMNLP | Towards Universal Dialogue State Tracking. | Liliang Ren, Kaige Xie, Lu Chen, Kai Yu |
| 2018 | SIGdial | Cost-Sensitive Active Learning for Dialogue State Tracking. | Kaige Xie, Cheng Chang, Liliang Ren, Lu Chen, Kai Yu |