Hao (Jack) BAI

Hao (Jack) BAI

CS M.S. Candidate

University of Illinois, Urbana Champaign

Hi, there! I’m Jack. I’m an M.S. student in Computer Science at UIUC, where I’m fully devoted to autonomous intelligence. I work on this problem upon two levels:

  • I study representation learning for foundation models in order to interpret their behaviors.
  • I study reinforcement learning algorithms for foudation models based on the so-derived understandings.

I’m also slightly involved in data augmentation to enhance intelligence.

Bio: Jack Bai is an M.S. student in Computer Science at UIUC, advised by Prof. Nan Jiang and Heng Ji, and a visiting scholar at BAIR, advised by Prof. Sergey Levine and Yi Ma. Before master’s, he obtained a dual degree from Zhejiang University and UIUC in Computer Engineering. During those wonderful years, he interned at Microsoft Research advised by Dr. Shilin He, and participated in the Amazon SocialBot Challenge advised by Prof. Chengxiang Zhai.

  • Representation Learning
  • Reinforcement Learning
  • Information Retrieval
  • MS in Computer Science, 2025

    University of Illinois, Urbana Champaign

  • BS in Computer Engineering, 2023

    University of Illinois, Urbana-Champaign


  • Only referred collaboration from UC Berkley will be considered (updated SP24)

Core Publications

The publications are not complete. Some work are protected and some work are still in the publication process.

White-Box Transformers via Sparse Rate Reduction: Compression Is All There Is? (JMLR)
CharmBana: Progressive Responses with Real-Time Internet Search for Knowledge-Powered Conversations (WSDM-24)
Social Commonsense-Guided Search Query Generation for Open-Domain Knowledge-Powered Conversations (EMNLP-23)