Research

I have been involved in various research groups, mainly focused on different areas of machine learning. They were all a lot of fun, but I'm not currently planning to pursue research in the future.

Computational Cameras Group @ Caltech
SIGNAL Lab @ Colorado State University
Mechanistic interpretability research w/ Prof. Tony Yue Yu @ Caltech
Climate Modeling Alliance (CliMA) @ Caltech

Publications

Text-Guided Image Restoration via a Unified Plug-and-Play Diffusion Framework

Brady Bhalla*, Zachary Huang*, Elyes Serghine*, Bingliang Zhang, Zihui Wu, Katherine L. Bouman (* denotes equal contribution)

CVPR Computational Cameras and Displays Workshop, 2025

TRACE: Real-Time Multimodal Common Ground Tracking in Situated Collaborative Dialogues

Hannah VanderHoeven, Brady Bhalla, Ibrahim Khebour, Austin C. Youngren, Videep Venkatesha, Mariah Bradford, Jack Fitzgerald, Carlos Mabrey, Jingxuan Tu, Yifan Zhu, Kenneth Lai, Changsoo Jung, James Pustejovsky, and Nikhil Krishnaswamy

NAACL: System Demonstrations, 2025

Propositional Extraction from Collaborative Naturalistic Dialogues

Videep Venkatesha, Abhijnan Nath, Ibrahim Khebour, Avyakta Chelle, Mariah Bradford, Jingxuan Tu, Hannah VanderHoeven, Brady Bhalla, Austin Youngren, Jack Fitzgerald, James Pustejovsky, Nathaniel Blanchard, Nikhil Krishnaswamy

Journal of Educational Data Mining, 17(1), 183-216., 2025

Higher Embedding Dimension Creates a Stronger World Model for a Simple Sorting Task

Brady Bhalla, Honglu Fan, Nancy Chen, Tony Yue YU

NeurIPS Mechanistic Interpretability Workshop, 2025