As foundation models continue to transform the landscape of artificial intelligence, demonstrating impressive capabilities across vision and language tasks, the need for interpretability, transparency, and control becomes increasingly critical. We develop methods to understand AI models by studying their representation spaces, the role of their internal architectural components, and the role of training data in their test-time behavior. Our work spans three main areas: interpretability of vision models, where we propose methods for mapping internal representations to human-understandable concepts and explaining failure modes; knowledge localization and editing in text-to-image generative models, where we propose techniques to identify and modify the layers responsible for specific concepts; and understanding the impact of data on models' training trajectories through the problem of machine unlearning, where we introduce benchmarks for data-level unlearning and new algorithms that improve unlearning efficacy, particularly through the use of intermediate checkpoints. Through these efforts, we contribute to building more transparent, controllable, and adaptable AI systems.
Keivan Rezaei is a Ph.D. student at the University of Maryland, where he is advised by Prof. Feizi and Prof. Hajiaghayi. His research centers on the interpretability of generative AI models, approached from two angles: a model perspective, where he localizes knowledge inside models and detects and explains their failure modes, and a data perspective, where he studies how individual data points shape a model through problems such as unlearning and data selection for language model pretraining. In addition, he has proposed methods for integrating ads into LLM outputs as an effective monetization strategy, along with methods for speeding up tool-calling language model agents. His work has appeared at ICML, ICLR, NeurIPS, TMLR, and EC, and he has held research internships at Google Research, the Allen Institute for AI, Adobe Research, and Susquehanna International Group.

