About Me

I am a researcher at BAAI (Beijing Academy of Artificial Intelligence), a non-profit organization engaged in Al research and development.

My current research interests are focused on Multi-modal LLMs and Embodied Agents.

My supervisors during my Ph.D. study are Prof. Heyan Huang and Assoc. Prof. Yang Gao. I also visited McGill Univerisity as a visiting student for a year advised by Prof. Jackie Chi Kit Cheung.

Download my resumé.

Interests
  • In-context learning
  • Text summarization
  • Low resource language generation
Education

  • Ph.D. Student, 2019 - 2025

    Beijing Institute of Technology

  • B.S., 2015 - 2019

    China University of Geosciences (Beijing)

News

  • [July 1, 2025] I have graduated and joined BAAI as a researcher!
  • [Sep 23, 2024] CItruS has been accepted to be appear at the EMNLP 2024 main conference!
  • [Jun 17, 2024] A new preprint about the long sequence processing of large language models has been released on ArXiv!
  • [Jan 23, 2024] A new preprint about the working mechanism of in-context learning has been released on ArXiv!
  • [May 17, 2023] Finally confirmed to visit the McGill NLP Group and be supervised by Prof. Jackie C. K. Cheung for a year starting on July 1st. Hope to make some academic contributions during my time in Montreal!
  • [August 15, 2022] Another paper on which I am the third author has been accepted by COLING 2022!
  • [Apirl 20, 2022] A paper on which I am the third author has been accepted by IJCAI 2022!
  • [March 31, 2022] A paper has been accepted by SIGIR 2022! See you at the conference!
  • [June 28, 2021] I join Alibaba DAMO academy as a research intern!
  • [July 9, 2021] My volunteer application for ACL 2021 has been accepted. See you at the virtual conference!
  • [May 6, 2021] A paper has been accepted by ACL-IJCNLP 2021!

Recent Publications

Yu Bai, Xiyuan Zou, Heyan Huang, Sanxing Chen, Marc-Antoine Rondeau, Yang Gao, Jackie Chi Kit Cheung (2024). CItruS: Chunked Instruction-aware State Eviction for Long Sequence Modeling. EMNLP 2024.

PDF Cite Code

Yu Bai, Heyan Huang, Cesare Spinoso-Di Piano, Marc-Antoine Rondeau, Sanxing Chen, Yang Gao, Jackie Chi Kit Cheung (2024). Identifying and Analyzing Task-Encoding Tokens in Large Language Models. Arxiv Preprint.

PDF Cite

Jiaao Zhan, Qian Chen, Boxing Chen, Wen Wang, Yu Bai, Yang Gao (2023). DePA: Improving Non-autoregressive Translation with Dependency-Aware Decoder. IWSLT 2023.

PDF Cite

Yu Bai, Heyan Huang, Kai Fan, Yang Gao, Yiming Zhu, Jiaao Zhan, Zewen Chi, Boxing Chen (2022). Unifying Cross-lingual Summarization and Machine Translation with Compression Rate. SIGIR 2022.

PDF Cite Code

Xiaochen Liu, Yang Gao, Yu Bai, Jiawei Li, Yinan Hu, Heyan Huang, Boxing Chen (2022). PSP: Pre-trained Soft Prompts for Few-Shot Abstractive Summarization. COLING 2022.

PDF Cite

Jiaao Zhan, Yang Gao, Yu Bai, Qianhui Liu (2022). Stage-wise Stylistic Headline Generation: Style Generation and Summarized Content Insertion. IJCAI-2022.

PDF Cite

Yu Bai, Yang Gao, Heyan Huang (2021). Cross-Lingual Abstractive Summarization with Limited Parallel Resources. ACL-IJCNLP 2021.

PDF Cite Code

Haonan Wang, Yang Gao, Yu Bai, Mirella Lapata, Heyan Huang (2021). Exploring Explainable Selection to Control Abstractive Summarization. AAAI 2021.

PDF Cite Code

Mucheng Ren, Heyan Huang, Ran Wei, Hongyu Liu, Yu Bai, Yang Wang, Yang Gao (2019). Multiple perspective answer reranking for multi-passage reading comprehension. NLPCC 2019.

PDF Cite Code

Contact