| 2026 | AAAI | Optimizing Urban Service Allocation with Time-Constrained Restless Bandits. | Yi Mao, Andrew Perrault |
| 2024 | EMNLP | Enhancing Language Model Alignment: A Confidence-Based Approach to Label Smoothing. | Baihe Huang, Hiteshi Sharma, Yi Mao |
| 2023 | EMNLP | Exploring Distributional Shifts in Large Language Models for Code Analysis. | Shushan Arakelyan, Rocktim Jyoti Das, Yi Mao, Xiang Ren |
| 2023 | EMNLP | RepoCoder: Repository-Level Code Completion Through Iterative Retrieval and Generation. | Fengji Zhang, Bei Chen, Yue Zhang, Jacky Keung, Jin Liu, Daoguang Zan, Yi Mao, Jian-Guang Lou, Weizhu Chen |
| 2023 | GECCO | A Cluster-Based Multiobjective Optimization Algorithm for VRPSTW with Salary Balance. | Yi Mao, Jingjing Li, Na Tang |
| 2023 | ICML | HyperTuning: Toward Adapting Large Language Models without Back-propagation. | Jason Phang, Yi Mao, Pengcheng He, Weizhu Chen |
| 2022 | ACL | DialogVED: A Pre-trained Latent Variable Encoder-Decoder Model for Dialog Response Generation. | Wei Chen, Yeyun Gong, Song Wang, Bolun Yao, Weizhen Qi, Zhongyu Wei, Xiaowu Hu, Bartuer Zhou, Yi Mao, Weizhu Chen, Biao Cheng, Nan Duan |
| 2022 | ACL | A Token-level Reference-free Hallucination Detection Benchmark for Free-form Text Generation. | Tianyu Liu, Yizhe Zhang, Chris Brockett, Yi Mao, Zhifang Sui, Weizhu Chen, Bill Dolan |
| 2022 | DSAA | AMD-DBSCAN: An Adaptive Multi-density DBSCAN for datasets of extremely variable density. | Ziqing Wang, Zhirong Ye, Yuyang Du, Yi Mao, Yanying Liu, Ziling Wu, Jun Wang |
| 2022 | NAACL | An End-to-End Dialogue Summarization System for Sales Calls. | Abedelkadir Asi, Song Wang, Roy Eisenstadt, Dean Geckt, Yarin Kuper, Yi Mao, Royi Ronen |
| 2022 | NAACL | OmniTab: Pretraining with Natural and Synthetic Data for Few-shot Table-based Question Answering. | Zhengbao Jiang, Yi Mao, Pengcheng He, Graham Neubig, Weizhu Chen |
| 2022 | NAACL | Knowledge-Grounded Dialogue Generation with a Unified Knowledge Representation. | Yu Li, Baolin Peng, Yelong Shen, Yi Mao, Lars Liden, Zhou Yu, Jianfeng Gao |
| 2021 | EMNLP | Finetuning Pretrained Transformers into RNNs. | Jungo Kasai, Hao Peng, Yizhe Zhang, Dani Yogatama, Gabriel Ilharco, Nikolaos Pappas, Yi Mao, Weizhu Chen, Noah A. Smith |
| 2020 | EMNLP | Exploiting Structured Knowledge in Text via Graph-Guided Representation Learning. | Tao Shen, Yi Mao, Pengcheng He, Guodong Long, Adam Trischler, Weizhu Chen |
| 2018 | ICLR | Action-dependent Control Variates for Policy Optimization via Stein Identity. | Hao Liu, Yihao Feng, Yi Mao, Dengyong Zhou, Jian Peng, Qiang Liu |
| 2013 | ICDM | Kernel Density Metric Learning. | Yujie He, Wenlin Chen, Yixin Chen, Yi Mao |
| 2013 | KDD | Density-based logistic regression. | Wenlin Chen, Yixin Chen, Yi Mao, Baolong Guo |
| 2012 | KDD | An integrated data mining approach to real-time clinical monitoring and deterioration warning. | Yi Mao, Wenlin Chen, Yixin Chen, Chenyang Lu, Marin Kollef, Thomas C. Bailey |
| 2012 | PSB | The Structure and Function of Chromatin and Chromosomes. | William Stafford Noble, C. Anthony Blau, Job Dekker, Zhi-jun Duan, Yi Mao |
| 2011 | ICDM | Medical Data Mining for Early Deterioration Warning in General Hospital Wards. | Yi Mao, Yixin Chen, Gregory Hackmann, Minmin Chen, Chenyang Lu, Marin Kollef, Thomas C. Bailey |
| 2010 | COLING | Dimensionality Reduction for Text using Domain Knowledge. | Yi Mao, Krishnakumar Balasubramanian, Guy Lebanon |
| 2009 | UAI | Domain Knowledge Uncertainty and Probabilistic Parameter Constraints. | Yi Mao, Guy Lebanon |
| 2007 | UAI | Statistical Translation, Heat Kernels and Expected Distances. | Joshua V. Dillon, Yi Mao, Guy Lebanon, Jian Zhang |
| 2003 | ICASSP | Audio restoration by constrained audio texture synthesis. | Lie Lu, Yi Mao, Wenyin Liu, Hong-Jiang Zhang |
| 2002 | ICASSP | Audio textures. | Lie Lu, Stan Z. Li, Liu Wenyin, Hong-Jiang Zhang, Yi Mao |