| 2026 | Networking | Graph-Attention Multi-Agent Reinforcement Learning with LLM-Guided Grouping for UAV-based Factory Fault Inspection. | Tianyu Gao, Bo Yang, Qiufeng Hu |
| 2025 | ACL | How to Train Long-Context Language Models (Effectively). | Tianyu Gao, Alexander Wettig, Howard Yen, Danqi Chen |
| 2025 | ICASSP | DTR: Dynamic Tree-Ring Watermarking Framework for Diffusion-Based Video Generation. | Shunyang Zeng, Linlin Yang, Jin Yang, Yezhen Wang, Tianyu Gao |
| 2025 | ICLR | HELMET: How to Evaluate Long-context Models Effectively and Thoroughly. | Howard Yen, Tianyu Gao, Minmin Hou, Ke Ding, Daniel Fleischer, Peter Izsak, Moshe Wasserblat, Danqi Chen |
| 2025 | ICML | Metadata Conditioning Accelerates Language Model Pre-training. | Tianyu Gao, Alexander Wettig, Luxi He, Yihe Dong, Sadhika Malladi, Danqi Chen |
| 2025 | INDIN | A corrosion detection method based on ultrasonic echo for gearbox surface. | Tianyu Gao, Yunlu Li, Jimin Zhang, Jingli Yang, Xiaopeng Fan |
| 2025 | INDIN | A multimodal cross-fusion network for rotating machinery fault diagnosis. | Jingli Yang, Zihan Wang, Tianyu Gao |
| 2025 | MICCAI | High-Res Brain Source Imaging of MEG Using a Vector Bayesian Beamformer with Noise Learning. | Tianyu Gao, Kunye Liu, Weikai Ma, Yang Gao, Xiaolin Ning |
| 2024 | ACL | Long-Context Language Modeling with Parallel Context Encoding. | Howard Yen, Tianyu Gao, Danqi Chen |
| 2024 | EMNLP | LitSearch: A Retrieval Benchmark for Scientific Literature Search. | Anirudh Ajith, Mengzhou Xia, Alexis Chevalier, Tanya Goyal, Danqi Chen, Tianyu Gao |
| 2024 | ICLR | Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning. | Mengzhou Xia, Tianyu Gao, Zhiyuan Zeng, Danqi Chen |
| 2024 | ICLR | Evaluating Large Language Models at Evaluating Instruction Following. | Zhiyuan Zeng, Jiatong Yu, Tianyu Gao, Yu Meng, Tanya Goyal, Danqi Chen |
| 2024 | INDIN | A Multi-Source Domain Generalization Network for Rotating Machinery Fault Diagnosis under Unseen Operating Conditions. | Tianyu Gao, Jingli Yang, Weiwei Hao, Xiaopeng Fan |
| 2023 | ACL | The CRINGE Loss: Learning what language not to model. | Leonard Adolphs, Tianyu Gao, Jing Xu, Kurt Shuster, Sainbayar Sukhbaatar, Jason Weston |
| 2023 | ACL | What In-Context Learning "Learns" In-Context: Disentangling Task Recognition and Task Learning. | Jane Pan, Tianyu Gao, Howard Chen, Danqi Chen |
| 2023 | EACL | Should You Mask 15% in Masked Language Modeling? | Alexander Wettig, Tianyu Gao, Zexuan Zhong, Danqi Chen |
| 2023 | EMNLP | Enabling Large Language Models to Generate Text with Citations. | Tianyu Gao, Howard Yen, Jiatong Yu, Danqi Chen |
| 2023 | GLOBECOM | STM-UNet: An Efficient U-shaped Architecture Based on Swin Transformer and Multiscale MLP for Medical Image Segmentation. | Lei Shi, Tianyu Gao, Zheng Zhang, Junxing Zhang |
| 2022 | ACL | Ditch the Gold Standard: Re-evaluating Conversational Question Answering. | Huihan Li, Tianyu Gao, Manan Goenka, Danqi Chen |
| 2022 | COLING | Automatic Label Sequence Generation for Prompting Sequence-to-sequence Models. | Zichun Yu, Tianyu Gao, Zhengyan Zhang, Yankai Lin, Zhiyuan Liu, Maosong Sun, Jie Zhou |
| 2021 | ACL | Making Pre-trained Language Models Better Few-shot Learners. | Tianyu Gao, Adam Fisch, Danqi Chen |
| 2021 | ACL | Manual Evaluation Matters: Reviewing Test Protocols of Distantly Supervised Relation Extraction. | Tianyu Gao, Xu Han, Yuzhuo Bai, Keyue Qiu, Zhiyu Xie, Yankai Lin, Zhiyuan Liu, Peng Li, Maosong Sun, Jie Zhou |
| 2021 | EMNLP | SimCSE: Simple Contrastive Learning of Sentence Embeddings. | Tianyu Gao, Xingcheng Yao, Danqi Chen |
| 2021 | KSEM | Representation Learning of Knowledge Graph with Semantic Vectors. | Tianyu Gao, Yuanming Zhang, Mengni Li, Jiawei Lu, Zhenbo Cheng, Gang Xiao |
| 2021 | KSEM | Adaptive Entity Alignment for Cross-Lingual Knowledge Graph. | Yuanming Zhang, Tianyu Gao, Jiawei Lu, Zhenbo Cheng, Gang Xiao |
| 2020 | AAAI | Neural Snowball for Few-Shot Relation Learning. | Tianyu Gao, Xu Han, Ruobing Xie, Zhiyuan Liu, Fen Lin, Leyu Lin, Maosong Sun |
| 2020 | ACL | Continual Relation Learning via Episodic Memory Activation and Reconsolidation. | Xu Han, Yi Dai, Tianyu Gao, Yankai Lin, Zhiyuan Liu, Peng Li, Maosong Sun, Jie Zhou |
| 2020 | COLING | Meta-Information Guided Meta-Learning for Few-Shot Relation Classification. | Bowen Dong, Yuan Yao, Ruobing Xie, Tianyu Gao, Xu Han, Zhiyuan Liu, Fen Lin, Leyu Lin, Maosong Sun |
| 2020 | EMNLP | Learning from Context or Names? An Empirical Study on Neural Relation Extraction. | Hao Peng, Tianyu Gao, Xu Han, Yankai Lin, Peng Li, Zhiyuan Liu, Maosong Sun, Jie Zhou |
| 2020 | ICDM | TADO: Time-varying Attention with Dual-Optimizer Model. | Yuexin Wu, Tianyu Gao, Sihao Wang, Zhongmin Xiong |
| 2020 | ICML | Few-shot Relation Extraction via Bayesian Meta-learning on Relation Graphs. | Meng Qu, Tianyu Gao, Louis-Pascal A. C. Xhonneux, Jian Tang |
| 2020 | IJCNLP | More Data, More Relations, More Context and More Openness: A Review and Outlook for Relation Extraction. | Xu Han, Tianyu Gao, Yankai Lin, Hao Peng, Yaoliang Yang, Chaojun Xiao, Zhiyuan Liu, Peng Li, Jie Zhou, Maosong Sun |
| 2019 | AAAI | Hybrid Attention-Based Prototypical Networks for Noisy Few-Shot Relation Classification. | Tianyu Gao, Xu Han, Zhiyuan Liu, Maosong Sun |
| 2019 | EMNLP | FewRel 2.0: Towards More Challenging Few-Shot Relation Classification. | Tianyu Gao, Xu Han, Hao Zhu, Zhiyuan Liu, Peng Li, Maosong Sun, Jie Zhou |
| 2019 | EMNLP | OpenNRE: An Open and Extensible Toolkit for Neural Relation Extraction. | Xu Han, Tianyu Gao, Yuan Yao, Deming Ye, Zhiyuan Liu, Maosong Sun |
| 2019 | SAC | Minimizing financial cost of scientific workflows under deadline constraints in multi-cloud environments. | Tianyu Gao, Chase Q. Wu, Aiqin Hou, Yongqiang Wang, Ruxia Li, Mingrui Xu |
| 2018 | AAAI | Lookine: Let the Blind Hear a Smile. | Yaohua Bu, Jia Jia, Yuhan Tang, Xuan Zang, Tianyu Gao |
| 2018 | ISPA | On Scheduling of High-Throughput Scientific Workflows under Budget Constraints in Multi-Cloud Environments. | Ruxia Li, Chase Q. Wu, Aiqin Hou, Yongqiang Wang, Tianyu Gao, Mingrui Xu |