Deep Crossing: Web-Scale Modeling without Manually Crafted Combinatorial Features. - Ying Shan; T. Ryan Hoens; Jian Jiao 007; Haijing Wang; Dong Yu; J. C. Mao (KDD, 2016)

Recurrent Binary Embedding for GPU-Enabled Exhaustive Retrieval from Billion-Scale Semantic Vectors. - Ying Shan; Jian Jiao 007; Jie Zhu 007; J. C. Mao (KDD, 2018)

TwinBERT: Distilling Knowledge to Twin-Structured Compressed BERT Models for Large-Scale Retrieval. - Wenhao Lu; Jian Jiao 007; Ruofei Zhang (CIKM, 2020)

An Enhanced Knowledge Injection Model for Commonsense Generation. - Zhihao Fan; Yeyun Gong; Zhongyu Wei; Siyuan Wang; Yameng Huang; Jian Jiao 007; Xuanjing Huang 001; Nan Duan; Ruofei Zhang (COLING, 2020)

ProphetNet-Ads: A Looking Ahead Strategy for Generative Retrieval Models in Sponsored Search Engine. - Weizhen Qi; Yeyun Gong; Yu Yan; Jian Jiao 007; Bo Shao; Ruofei Zhang; Houqiang Li; Nan Duan; Ming Zhou 001 (NLPCC (2), 2020)

TwinBERT: Distilling Knowledge to Twin-Structured BERT Models for Efficient Retrieval. - Wenhao Lu; Jian Jiao 007; Ruofei Zhang (CoRR, 2020)

GLGE: A New General Language Generation Evaluation Benchmark. - Dayiheng Liu; Yu Yan; Yeyun Gong; Weizhen Qi; Hang Zhang 029; Jian Jiao 007; Weizhu Chen; Jie Fu 001; Linjun Shou; Ming Gong 001; Pengcheng Wang; Jiusheng Chen; et al. (ACL/IJCNLP (Findings), 2021)

KFCNet: Knowledge Filtering and Contrastive Learning for Generative Commonsense Reasoning. - Haonan Li 002; Yeyun Gong; Jian Jiao 007; Ruofei Zhang; Timothy Baldwin; Nan Duan (EMNLP (Findings), 2021)

HittER: Hierarchical Transformers for Knowledge Graph Embeddings. - Sanxing Chen; Xiaodong Liu 003; Jianfeng Gao 001; Jian Jiao 007; Ruofei Zhang; Yangfeng Ji (EMNLP (1), 2021)

SiameseXML: Siamese Networks meet Extreme Classifiers with 100M Labels. - Kunal Dahiya; Ananye Agarwal; Deepak Saini; Gururaj K; Jian Jiao 007; Amit Singh 003; Sumeet Agarwal; Purushottam Kar; Manik Varma (ICML, 2021)

BANG: Bridging Autoregressive and Non-autoregressive Generation with Large Scale Pretraining. - Weizhen Qi; Yeyun Gong; Jian Jiao 007; Yu Yan; Weizhu Chen; Dayiheng Liu; Kewen Tang; Houqiang Li; Jiusheng Chen; Ruofei Zhang; Ming Zhou 001; Nan Duan (ICML, 2021)

Mask Attention Networks: Rethinking and Strengthen Transformer. - Zhihao Fan; Yeyun Gong; Dayiheng Liu; Zhongyu Wei; Siyuan Wang; Jian Jiao 007; Nan Duan; Ruofei Zhang; Xuanjing Huang 001 (NAACL-HLT, 2021)

GalaXC: Graph Neural Networks with Labelwise Attention for Extreme Classification. - Deepak Saini; Arnav Kumar Jain; Kushal Dave 001; Jian Jiao 007; Amit Singh 003; Ruofei Zhang; Manik Varma (WWW, 2021)

KFCNet: Knowledge Filtering and Contrastive Learning Network for Generative Commonsense Reasoning. - Haonan Li 002; Yeyun Gong; Jian Jiao 007; Ruofei Zhang; Timothy Baldwin; Nan Duan (CoRR, 2021)

Metric-guided Distillation: Distilling Knowledge from the Metric to Ranker and Retriever for Generative Commonsense Reasoning. - Xingwei He 003; Yeyun Gong; A-Long Jin; Weizhen Qi; Hang Zhang 029; Jian Jiao 007; Bartuer Zhou; Biao Cheng; Siu-Ming Yiu; Nan Duan (EMNLP, 2022)

PromptBERT: Improving BERT Sentence Embeddings with Prompts. - Ting Jiang; Jian Jiao 007; Shaohan Huang; Zihan Zhang; Deqing Wang; Fuzhen Zhuang; Furu Wei; Haizhen Huang; Denvy Deng; Qi Zhang 066 (EMNLP, 2022)

Taming Sparsely Activated Transformer with Stochastic Experts. - Simiao Zuo; Xiaodong Liu 003; Jian Jiao 007; Young Jin Kim 001; Hany Hassan; Ruofei Zhang; Jianfeng Gao 001; Tuo Zhao (ICLR, 2022)

CULG: Commercial Universal Language Generation. - Haonan Li 002; Yameng Huang; Yeyun Gong; Jian Jiao 007; Ruofei Zhang; Timothy Baldwin; Nan Duan (NAACL-HLT (Industry Papers), 2022)

A Self-Paced Mixed Distillation Method for Non-Autoregressive Generation. - Weizhen Qi; Yeyun Gong; Yelong Shen; Jian Jiao 007; Yu Yan; Houqiang Li; Ruofei Zhang; Weizhu Chen; Nan Duan (CoRR, 2022)

LEAD: Liberal Feature-based Distillation for Dense Retrieval. - Hao Sun 015; Xiao Liu 029; Yeyun Gong; Anlei Dong; Jian Jiao 007; Jingwen Lu; Yan Zhang 117; Daxin Jiang; Linjun Yang; Rangan Majumder; Nan Duan (CoRR, 2022)

Efficient Long Sequence Modeling via State Space Augmented Transformer. - Simiao Zuo; Xiaodong Liu 003; Jian Jiao 007; Denis Charles; Eren Manavoglu; Tuo Zhao; Jianfeng Gao 001 (CoRR, 2022)

Curriculum Sampling for Dense Retrieval with Document Expansion. - Xingwei He 003; Yeyun Gong; A-Long Jin; Hang Zhang 029; Anlei Dong; Jian Jiao 007; Siu-Ming Yiu; Nan Duan (CoRR, 2022)

Dual-Alignment Pre-training for Cross-lingual Sentence Embedding. - Ziheng Li; Shaohan Huang; Zihan Zhang; Zhi-Hong Deng 001; Qiang Lou; Haizhen Huang; Jian Jiao 007; Furu Wei; Weiwei Deng; Qi Zhang 066 (ACL (1), 2023)

DeepTagger: Knowledge Enhanced Named Entity Recognition for Web-Based Ads Queries. - Simiao Zuo; Pengfei Tang; Xinyu Hu; Qiang Lou; Jian Jiao 007; Denis Charles (CIKM, 2023)

CAPSTONE: Curriculum Sampling for Dense Retrieval with Document Expansion. - Xingwei He 003; Yeyun Gong; A-Long Jin; Hang Zhang 029; Anlei Dong; Jian Jiao 007; Siu-Ming Yiu; Nan Duan (EMNLP, 2023)

Deep Encoders with Auxiliary Parameters for Extreme Classification. - Kunal Dahiya; Sachin Yadav 002; Sushant Sondhi; Deepak Saini; Sonu Mehta; Jian Jiao 007; Sumeet Agarwal; Purushottam Kar; Manik Varma (KDD, 2023)

Renee: End-To-End Training of Extreme Classification Models. - Vidit Jain; Jatin Prakash; Deepak Saini; Jian Jiao 007; Ramachandran Ramjee; Manik Varma (MLSys, 2023)

AR-Diffusion: Auto-Regressive Diffusion Model for Text Generation. - Tong Wu; Zhihao Fan; Xiao Liu 029; Hai-Tao Zheng 002; Yeyun Gong; Yelong Shen; Jian Jiao 007; Juntao Li; Zhongyu Wei; Jian Guo; Nan Duan; Weizhu Chen (NeurIPS, 2023)

NGAME: Negative Mining-aware Mini-batching for Extreme Classification. - Kunal Dahiya; Nilesh Gupta; Deepak Saini; Akshay Soni; Yajun Wang; Kushal Dave 001; Jian Jiao 007; Gururaj K; Prasenjit Dey; Amit Singh 003; Deepesh Hada; Vidit Jain; et al. (WSDM, 2023)

PROD: Progressive Distillation for Dense Retrieval. - Zhenghao Lin; Yeyun Gong; Xiao Liu 029; Hang Zhang 029; Chen Lin 001; Anlei Dong; Jian Jiao 007; Jingwen Lu; Daxin Jiang; Rangan Majumder; Nan Duan (WWW, 2023)

Pre-training Transformers for Knowledge Graph Completion. - Sanxing Chen; Hao Cheng 002; Xiaodong Liu 003; Jian Jiao 007; Yangfeng Ji; Jianfeng Gao 001 (CoRR, 2023)

AutoHint: Automatic Prompt Optimization with Hint Generation. - Hong Sun; Xue Li 005; Yinchuan Xu; Youkow Homma; Qi Cao; Min Wu; Jian Jiao 007; Denis Charles (CoRR, 2023)

Evoke: Evoking Critical Thinking Abilities in LLMs via Reviewer-Author Prompt Editing. - Xinyu Hu; Pengfei Tang; Simiao Zuo; Zihan Wang; Bowen Song; Qiang Lou; Jian Jiao 007; Denis Charles (ICLR, 2024)

OAK: Enriching Document Representations using Auxiliary Knowledge for Extreme Classification. - Shikhar Mohan; Deepak Saini; Anshul Mittal; Sayak Ray Chowdhury; Bhawna Paliwal; Jian Jiao 007; Manish Gupta; Manik Varma (ICML, 2024)

Optimizing Novelty of Top-k Recommendations using Large Language Models and Reinforcement Learning. - Amit Sharma 007; Hua Li; Xue Li 005; Jian Jiao 007 (KDD, 2024)

Extreme Meta-Classification for Large-Scale Zero-Shot Retrieval. - Sachin Yadav 002; Deepak Saini; Anirudh Buvanesh; Bhawna Paliwal; Kunal Dahiya; Siddarth Asokan; Yashoteja Prabhu; Jian Jiao 007; Manik Varma (KDD, 2024)

AnnoLLM: Making Large Language Models to Be Better Crowdsourced Annotators. - Xingwei He 003; Zhenghao Lin; Yeyun Gong; A-Long Jin; Hang Zhang 029; Chen Lin 001; Jian Jiao 007; Siu Ming Yiu; Nan Duan; Weizhu Chen (NAACL (Industry Track), 2024)

Not All Tokens Are What You Need for Pretraining. - Zhenghao Lin; Zhibin Gou; Yeyun Gong; Xiao Liu 029; Yelong Shen; Ruochen Xu; Chen Lin 001; Yujiu Yang; Jian Jiao 007; Nan Duan; Weizhu Chen (NeurIPS, 2024)

Graph Regularized Encoder Training for Extreme Classification. - Anshul Mittal; Shikhar Mohan; Deepak Saini; Suchith C. Prabhu; Jian Jiao 007; Sumeet Agarwal; Soumen Chakrabarti; Purushottam Kar; Manik Varma (CoRR, 2024)

Rho-1: Not All Tokens Are What You Need. - Zhenghao Lin; Zhibin Gou; Yeyun Gong; Xiao Liu 029; Yelong Shen; Ruochen Xu; Chen Lin 001; Yujiu Yang; Jian Jiao 007; Nan Duan; Weizhu Chen (CoRR, 2024)

Scaling the Vocabulary of Non-autoregressive Models for Efficient Generative Retrieval. - Ravisri Valluri; Akash Kumar Mohankumar; Kushal Dave 001; Amit Singh 003; Jian Jiao 007; Manik Varma; Gaurav Sinha 001 (CoRR, 2024)

Task Oriented In-Domain Data Augmentation. - Xiao Liang; Xinyu Hu; Simiao Zuo; Yeyun Gong; Qiang Lou; Yi Liu; Shao-Lun Huang; Jian Jiao 007 (CoRR, 2024)

CROSS-JEM: Accurate and Efficient Cross-encoders for Short-text Ranking Tasks. - Bhawna Paliwal; Deepak Saini; Mudit Dhawan; Siddarth Asokan; Nagarajan Natarajan; Surbhi Aggarwal; Pankaj Malhotra; Jian Jiao 007; Manik Varma (CoRR, 2024)

Integrative Decoding: Improve Factuality via Implicit Self-consistency. - Yi Cheng; Xiao Liang; Yeyun Gong; Wen Xiao; Song Wang; Yuji Zhang 002; Wenjun Hou; Kaishuai Xu; Wenge Liu; Wenjie Li 002; Jian Jiao 007; Qi Chen 009; et al. (CoRR, 2024)

Integrative Decoding: Improving Factuality via Implicit Self-consistency. - Yi Cheng; Xiao Liang; Yeyun Gong; Wen Xiao; Song Wang; Yuji Zhang 002; Wenjun Hou; Kaishuai Xu; Wenge Liu; Wenjie Li 002; Jian Jiao 007; Qi Chen 009; et al. (ICLR, 2025)

On the Necessity of World Knowledge for Mitigating Missing Labels in Extreme Classification. - Jatin Prakash; Anirudh Buvanesh; Bishal Santra; Deepak Saini; Sachin Yadav 002; Jian Jiao 007; Yashoteja Prabhu; Amit Sharma 007; Manik Varma (KDD (1), 2025)

Scaling the Vocabulary of Non-autoregressive Models for Fast Generative Retrieval. - Ravisri Valluri; Akash Kumar Mohankumar; Kushal Dave 001; Amit Singh 003; Jian Jiao 007; Manik Varma; Gaurav Sinha 001 (KDD (1), 2025)

Graph Regularized Encoder Training for Extreme Classification. - Anshul Mittal; Shikhar Mohan; Deepak Saini; Siddarth Asokan; Suchith Chidananda Prabhu; Lakshya Kumar; Pankaj Malhotra; Jian Jiao 007; Amit Singh 003; Sumeet Agarwal; Soumen Chakrabarti; Purushottam Kar; et al. (WWW (Companion Volume), 2025)

Sigma: Differential Rescaling of Query, Key and Value for Efficient Language Models. - Zhenghao Lin; Zihao Tang; Xiao Liu 029; Yeyun Gong; Yi Cheng; Qi Chen 009; Hang Li; Ying Xin; Ziyue Yang; Kailai Yang; Yu Yan; Xiao Liang; et al. (CoRR, 2025)