Yihao Xue

I am a Research Scientist at Google, working on building multi-agent capabilities for Gemini through post-training and inference-time techniques.

Previously, I completed my PhD in the Computer Science Department at UCLA, advised by Professor Baharan Mirzasoleiman, where I worked on robustness and out-of-distribution generalization in machine learning. I earned my undergraduate degree from Shanghai Jiao Tong University.

I am a recipient of the OpenAI Superalignment Grant and the Outstanding Graduate Student Research Award.

Portrait of Yihao Xue
Me and Brie

Publications

Preprints

Experience

Student Researcher, Google

Jun. 2025 - Jan. 2026

With Vincent Cohen-Addad. Designing optimal test-time scaling strategies to improve Gemini's reasoning.

Research Intern, Cisco

Jun. 2022 - Sep. 2022

With Ali Payani. Worked on few-shot domain adaptation.

More About Me

My last name is pronounced “sh-WEH”. I like playing tennis, ping-pong, and recently, running. I'm also passionate about music, especially late Romantic and early 20th-century classical music - these are my current top 7 favorite composers.