Hybrid Attention-Based Prototypical Networks for Noisy Few-Shot Relation Classification. - Tianyu Gao 001; Xu Han 007; Zhiyuan Liu 001; Maosong Sun 001 (AAAI, 2019)

FewRel 2.0: Towards More Challenging Few-Shot Relation Classification. - Tianyu Gao 001; Xu Han 007; Hao Zhu 006; Zhiyuan Liu 001; Peng Li 030; Maosong Sun 001; Jie Zhou 016 (EMNLP/IJCNLP (1), 2019)

OpenNRE: An Open and Extensible Toolkit for Neural Relation Extraction. - Xu Han 007; Tianyu Gao 001; Yuan Yao 013; Deming Ye; Zhiyuan Liu 001; Maosong Sun 001 (EMNLP/IJCNLP (3), 2019)

KEPLER: A Unified Model for Knowledge Embedding and Pre-trained Language Representation. - Xiaozhi Wang; Tianyu Gao 001; Zhaocheng Zhu; Zhiyuan Liu 001; Juanzi Li; Jian Tang 005 (CoRR, 2019)

Neural Snowball for Few-Shot Relation Learning. - Tianyu Gao 001; Xu Han 007; Ruobing Xie; Zhiyuan Liu 001; Fen Lin; Leyu Lin; Maosong Sun 001 (AAAI, 2020)

Continual Relation Learning via Episodic Memory Activation and Reconsolidation. - Xu Han 007; Yi Dai; Tianyu Gao 001; Yankai Lin; Zhiyuan Liu 001; Peng Li 030; Maosong Sun 001; Jie Zhou 016 (ACL, 2020)

Meta-Information Guided Meta-Learning for Few-Shot Relation Classification. - Bowen Dong; Yuan Yao 013; Ruobing Xie; Tianyu Gao 001; Xu Han 007; Zhiyuan Liu 001; Fen Lin; Leyu Lin; Maosong Sun 001 (COLING, 2020)

Learning from Context or Names? An Empirical Study on Neural Relation Extraction. - Hao Peng 015; Tianyu Gao 001; Xu Han 007; Yankai Lin; Peng Li 030; Zhiyuan Liu 001; Maosong Sun 001; Jie Zhou 016 (EMNLP (1), 2020)

Few-shot Relation Extraction via Bayesian Meta-learning on Relation Graphs. - Meng Qu; Tianyu Gao 001; Louis-Pascal A. C. Xhonneux; Jian Tang 005 (ICML, 2020)

More Data, More Relations, More Context and More Openness: A Review and Outlook for Relation Extraction. - Xu Han 007; Tianyu Gao 001; Yankai Lin; Hao Peng 015; Yaoliang Yang; Chaojun Xiao; Zhiyuan Liu 001; Peng Li 030; Jie Zhou 016; Maosong Sun 001 (AACL/IJCNLP, 2020)

Making Pre-trained Language Models Better Few-shot Learners. - Tianyu Gao 001; Adam Fisch; Danqi Chen 001 (ACL/IJCNLP (1), 2021)

Manual Evaluation Matters: Reviewing Test Protocols of Distantly Supervised Relation Extraction. - Tianyu Gao 001; Xu Han 007; Yuzhuo Bai; Keyue Qiu; Zhiyu Xie 001; Yankai Lin; Zhiyuan Liu 001; Peng Li 030; Maosong Sun 001; Jie Zhou 016 (ACL/IJCNLP (Findings), 2021)

SimCSE: Simple Contrastive Learning of Sentence Embeddings. - Tianyu Gao 001; Xingcheng Yao; Danqi Chen 001 (EMNLP (1), 2021)

KEPLER: A Unified Model for Knowledge Embedding and Pre-trained Language Representation. - Xiaozhi Wang; Tianyu Gao 001; Zhaocheng Zhu; Zhengyan Zhang; Zhiyuan Liu 001; Juanzi Li; Jian Tang 005 (Trans. Assoc. Comput. Linguistics, 2021)

Ditch the Gold Standard: Re-evaluating Conversational Question Answering. - Huihan Li 001; Tianyu Gao 001; Manan Goenka; Danqi Chen 001 (ACL (1), 2022)

Automatic Label Sequence Generation for Prompting Sequence-to-sequence Models. - Zichun Yu; Tianyu Gao 001; Zhengyan Zhang; Yankai Lin; Zhiyuan Liu 001; Maosong Sun 001; Jie Zhou 016 (COLING, 2022)

Recovering Private Text in Federated Learning of Language Models. - Samyak Gupta; Yangsibo Huang; Zexuan Zhong; Tianyu Gao 001; Kai Li 001; Danqi Chen 001 (NeurIPS, 2022)

What In-Context Learning "Learns" In-Context: Disentangling Task Recognition and Task Learning. - Jane Pan; Tianyu Gao 001; Howard Chen 003; Danqi Chen 001 (ACL (Findings), 2023)

Should You Mask 15% in Masked Language Modeling? - Alexander Wettig; Tianyu Gao 001; Zexuan Zhong; Danqi Chen 001 (EACL, 2023)

Enabling Large Language Models to Generate Text with Citations. - Tianyu Gao 001; Howard Yen; Jiatong Yu; Danqi Chen 001 (EMNLP, 2023)

Fine-Tuning Language Models with Just Forward Passes. - Sadhika Malladi; Tianyu Gao 001; Eshaan Nichani; Alex Damian; Jason D. Lee; Danqi Chen 001; Sanjeev Arora (NeurIPS, 2023)

Long-Context Language Modeling with Parallel Context Encoding. - Howard Yen; Tianyu Gao 001; Danqi Chen 001 (ACL (1), 2024)

LitSearch: A Retrieval Benchmark for Scientific Literature Search. - Anirudh Ajith; Mengzhou Xia; Alexis Chevalier; Tanya Goyal; Danqi Chen 001; Tianyu Gao 001 (EMNLP, 2024)

Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning. - Mengzhou Xia; Tianyu Gao 001; Zhiyuan Zeng; Danqi Chen 001 (ICLR, 2024)

Evaluating Large Language Models at Evaluating Instruction Following. - Zhiyuan Zeng; Jiatong Yu; Tianyu Gao 001; Yu Meng 001; Tanya Goyal; Danqi Chen 001 (ICLR, 2024)

Improving Language Understanding from Screenshots. - Tianyu Gao 001; Zirui Wang; Adithya Bhaskar; Danqi Chen 001 (CoRR, 2024)

HELMET: How to Evaluate Long-Context Language Models Effectively and Thoroughly. - Howard Yen; Tianyu Gao 001; Minmin Hou; Ke Ding; Daniel Fleischer; Peter Izsak; Moshe Wasserblat; Danqi Chen 001 (CoRR, 2024)

How to Train Long-Context Language Models (Effectively). - Tianyu Gao 001; Alexander Wettig; Howard Yen; Danqi Chen 001 (ACL (1), 2025)

HELMET: How to Evaluate Long-context Models Effectively and Thoroughly. - Howard Yen; Tianyu Gao 001; Minmin Hou; Ke Ding; Daniel Fleischer; Peter Izsak; Moshe Wasserblat; Danqi Chen 001 (ICLR, 2025)

Metadata Conditioning Accelerates Language Model Pre-training. - Tianyu Gao 001; Alexander Wettig; Luxi He; Yihe Dong; Sadhika Malladi; Danqi Chen 001 (CoRR, 2025)

LongProc: Benchmarking Long-Context Language Models on Long Procedural Generation. - Xi Ye; Fangcong Yin; Yinghui He; Joie Zhang; Howard Yen; Tianyu Gao 001; Greg Durrett; Danqi Chen 001 (CoRR, 2025)

Cache Me If You Can: How Many KVs Do You Need for Effective Long-Context LMs? - Adithya Bhaskar; Alexander Wettig; Tianyu Gao 001; Yihe Dong; Danqi Chen 001 (CoRR, 2025)