| 2025 | ICML | P(all-atom) Is Unlocking New Path For Protein Design. | Wei Qu, Jiawei Guan, Rui Ma, Ke Zhai, Weikun Wu, Haobo Wang |
| 2018 | ICLR | Adaptive Dropout with Rademacher Complexity Regularization. | Ke Zhai, Huan Wang |
| 2016 | ACL | Recognizing Salient Entities in Shopping Queries. | Zornitsa Kozareva, Qi Li, Ke Zhai, Weiwei Guo |
| 2016 | SIGIR | Query to Knowledge: Unsupervised Entity Extraction from Shopping Queries using Adaptor Grammars. | Ke Zhai, Zornitsa Kozareva, Yuening Hu, Qi Li, Weiwei Guo |
| 2014 | ACL | Polylingual Tree-Based Topic Models for Translation Domain Adaptation. | Yuening Hu, Ke Zhai, Vladimir Eidelman, Jordan L. Boyd-Graber |
| 2014 | ACL | Discovering Latent Structure in Task-Oriented Dialogues. | Ke Zhai, Jason D. Williams |
| 2013 | ICML | Online Latent Dirichlet Allocation with Infinite Vocabulary. | Ke Zhai, Jordan L. Boyd-Graber |
| 2013 | PPoPP | TeamWork: synchronizing threads globally to detect real deadlocks for multithreaded programs. | Yan Cai, Ke Zhai, Shangru Wu, Wing Kwong Chan |
| 2012 | COMPSAC | Preemptive Regression Test Scheduling Strategies: A New Testing Approach to Thriving on the Volatile Service Environments. | Lijun Mei, Ke Zhai, Bo Jiang, W. K. Chan, T. H. Tse |
| 2012 | ICML | Modeling Images using Transformed Indian Buffet Processes. | Ke Zhai, Yuening Hu, Jordan L. Boyd-Graber, Sinead Williamson |
| 2012 | ISSTA | CARISMA: a context-sensitive approach to race-condition sample-instance selection for multithreaded applications. | Ke Zhai, Boni Xu, W. K. Chan, T. H. Tse |
| 2012 | WWW | Mr. LDA: a flexible large scale topic modeling package using variational inference in MapReduce. | Ke Zhai, Jordan L. Boyd-Graber, Sebastian Bruch, Mohamad L. Alkhouja |
| 2010 | ICWS | Taking Advantage of Service Selection: A Study on the Testing of Location-Based Web Services Through Test Case Prioritization. | Ke Zhai, Bo Jiang, W. K. Chan, T. H. Tse |