Tianning Chai

EMAIL GOOGLE SCHOLAR

I am currently a research intern at BAIR (Berkeley Artificial Intelligence Research), advised by Dr. Roei Herzig and Prof. Trevor Darrell. Previously I obtained my B.Eng from Wuhan University, advised by Prof. Mang Ye.

Input Stream (Window: 10)

Attention Map

Layer 11 / Head 4 (Softmax)

> forward_pass(hidden_states)

> extracting_activations...

Next Token Probs (Top-K)

Selected Publications

Activation Rewards Models for Few-Shot Model Alignment

Preprint 2025

Tianning Chai*, Chancharik Mitra*, Brandon Huang, Gautam Rajendrakumar Gare, Zhiqiu Lin, Assaf Arbelle, Leonind Karlinsky, Rogerio Feris, Trevor Darrell, Deva Ramanan, Roei Herzig
Enhancing Few-Shot Vision-Language Classification with Large Multimodal Model Features

ICCV 2025

Chancharik Mitra*, Brandon Huang*, Tianning Chai, Zhiqiu Lin, Assaf Arbelle, Rogerio Feris, Leonid Karlinsky, Trevor Darrell, Deva Ramanan, Roei Herzig
Decentralized Federated Learning via Third-Party Bridge Collaboration

Preprint 2024

Tianning Chai, Mang Ye, Wenke Huang, Chun-Mei Feng