~ $ whoami

Soroush Mahdi

I work on world models and representation learning, especially on learning representations at several levels of abstraction, so an agent can plan better and further ahead and adapt when the world changes.

~ $ cat status

● looking for a PhD position in world models / representation learning

~ $ ls papers/

evict3r/ mode-tta/ memloss/

~ $ cd ~/lab

$ cat about.md

I’m interested in world models: learned representations an agent can carry forward in time, use to predict what happens next, and keep reliable when the world changes. I think a good world model has to learn multiple levels of abstraction, with fine-grained states for the next few steps and coarser ones for long horizons. That hierarchy is what lets an agent plan better and further ahead, and adapt to a change without relearning everything. Getting there is largely a representation learning problem: what to encode at each level, what to remember, and how to adapt when the data drifts.

My recent work covers the last two. Evict3R asks what a streaming 3D transformer actually needs to remember; a training-free eviction policy roughly halves its peak memory on 7-Scenes with almost no loss in accuracy. MODE-TTA adapts a frozen 3D vision-language model at test time by modelling each class’s features as a Gaussian mixture updated online. My M.Sc. at Amirkabir University of Technology (GPA 4.0/4.0) was on adversarial robustness and led to MemLoss.

I’m a research assistant at AUT’s Autonomous & Intelligent Systems Lab, working with Mahdi Javanmardi, and I collaborate with Ali Cheraghian’s group at the University of Sydney. If our interests overlap, I’d be glad to hear from you.

interests/
├── world-models/
│   ├── abstraction/      multi-level states for longer, better plans
│   ├── memory/           what to keep from a long stream, under a fixed budget
│   ├── dynamics/         predicting what happens next, in latent space
│   └── embodied-ai/      using learned models for planning and control
└── representation-learning/
    ├── self-supervised/  learning from raw video and 3D observations
    ├── adaptation/       staying reliable under distribution shift
    └── generative/       diffusion and flow-based models

$ ls -l papers/

  1. under review ICLR 2027

    MODE-TTA: Multi-Mode Density Modeling for Test-Time Adaptation of 3D Vision-Language Foundation Models

    Hamidreza Dastmalchi, Soroush Mahdi, Mohammadreza Alimoradijazi, Mehran Tamjidi, Morteza Saberi, Mojtaba Golzan, Aijun An, Ali Cheraghian

    per-class Gaussian mixtures updated online with hierarchical EM, plus residual text alignment

  2. arXiv 2025 · 2509.17650

    Evict3R: Training-Free Token Eviction for Memory-Bounded Streaming Visual Geometry Transformers

    Soroush Mahdi, Fardin Ayar, Ehsan Javanmardi, Manabu Tsukada, Mahdi Javanmardi

    peak memory 18.63 GB → 9.39 GB on 7-Scenes; accuracy and completeness drop by only 0.003

  3. arXiv 2025 · 2510.09105

    MemLoss: Enhancing Adversarial Training with Recycling Adversarial Examples

    Soroush Mahdi, Maryam Amirmazlaghani, Saeed Saravani, Zahra Dehghanian

    reuses adversarial examples from past epochs; better clean and robust accuracy on CIFAR-10/100 and SVHN

$ contact --all

email
soroushm1377@gmail.com
github
soroush-mim
linkedin
soroush-mahdi
cv
CV_soroush_mahdi.pdf