Ahmad Rahimi

PhD student at VITA, EPFL

prof_pic.jpg

I am a PhD student at the VITA lab at EPFL advised by Prof. Alexandre Alahi. My research interests include deep learning and computer vision for autonomous vehicles. I have experience in vehicle trajectory prediction methods, and more recently, I’m growing an interest in video generation and world models.

I got my B.Sc. in Computer Science from Sharif University of Technology, Iran, in 2022.

news

Feb 26, 2025 Two of my papers got accepted at CVPR 2025! Cheers 🍺!
Sep 15, 2022 I joined VITA Lab to do a PhD in Computer Science at EPFL. Very exciting!
Jun 01, 2022 I finished my B.Sc. degree in Computer Science at Sharif University of Technology.
Mar 29, 2022 My project on robustness of vehicle trajectory prediction models which I did during my internship at VITA lab is accepted at CVPR 2022!

selected publications

  1. CVPR
    gem-control.gif
    GEM: A Generalizable Ego-vision Multimodal World Model for Fine-Grained Ego-Motion, Object Dynamics, and Scene Composition Control
    Mariam Hassan*, Sebastian Stapf*, Ahmad Rahimi*, and 17 more authors
    2025
    arXiv Code Website
  2. under review
    offroad-loss.png
    A Multi-Loss Strategy for Vehicle Trajectory Prediction: Combining Off-Road, Diversity, and Directional Consistency Losses
    Ahmad Rahimi, and Alexandre Alahi
    2024
    arXiv Code
  3. CVPR
    social-causality.jpg
    Sim-to-Real Causal Transfer: A Metric Learning Approach to Causally-Aware Interaction Representations
    Ahmad Rahimi, Po-Chien Luan, Yuejiang Liu, and 2 more authors
    2025
    arXiv Code
  4. CVPR
    scene-attack.gif
    Vehicle trajectory prediction works, but not everywhere
    Mohammadhossein Bahari, Saeed Saadatnejad, Ahmad Rahimi, and 4 more authors
    In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2022
    arXiv Code Website