Javad Rajabi

I am a CS PhD Student at University of Toronto (UofT), and a Faculty Affiliate Researcher at Vector Institute, co-supervised by David B. Lindell and Babak Taati. Currently, I am a research intern at Samsung AI Center Toronto.

Currently, my research focuses on inference-time guidance and scaling for diffusion and flow-matching models, using targeted search, sampling, and steering signals to improve generation quality and control specific attributes. I am exploring novel guidance signals (e.g., intrinsic confidence, perturbation-based methods, reward models, textual feedback, etc.) that can score partial states or full trajectories.

Profile photo
Research Interest

My research interests lie in generative modeling (flows and diffusion) for image/video/3D/4D generation and their potential to understand and construct sophisticated environments. I also have a deep interest in mathematics, optimization, and dynamic systems, particularly in their applications to generative models and machine learning.

News
Selected Publications
SEGA: Spectral-Energy Guided Attention for Resolution Extrapolation in Diffusion Transformers
Javad Rajabi, Kimia Shaban, Koorosh Roohi, David B. Lindell, Babak Taati
Token Perturbation Guidance for Diffusion Models
Javad Rajabi, Soroush Mehraban, Seyedmorteza Sadat, Babak Taati
NeurIPS 2025
TCARE-PD: A Multi-Site Anonymized Clinical Dataset for Parkinson's Disease Gait Assessment
Vida Adeli, Ivan Klabucar, Javad Rajabi, Benjamin Filtjens, Soroush Mehraban et al.
NeurIPS 2025
STARS: Self-supervised Tuning for 3D Action Recognition in Skeleton Sequences
Soroush Mehraban, Javad Rajabi, Andrea Iaboni, Babak Taati
WACV 2026
Teaching Assistant
  • CSC420: Introduction to Image Understanding — Winter 2025, Fall 2025, Winter 2026
  • CSC311: Introduction to Machine Learning — Summer 2026
  • CSC2515: Introduction to Machine Learning — Winter 2026
  • CSC263: Data Structures and Analysis — Winter 2025
Academic Service
  • Conference reviewer: CVPR 2026, ECCV 2026, NeurIPS 2026