About

Amir Kasaei

Biography

I'm an M.Sc. student in Computer Engineering at Sharif University of Technology and a Research Assistant at the RIML Lab, where I work on compositional visual generation, text-to-image diffusion models, and cross-modal reasoning.

My research focuses on teaching machines to combine visual elements in structured and creative ways — spanning generative modeling, evaluation, and semantic alignment across vision and language. I'm always open to connecting with others interested in machine learning, computer vision, and AI research.

Download CV
  • City: Tehran, Iran
  • Degree: M.Sc. Computer Engineering

Research Interests

  • Compositional & Multimodal Learning
  • Text-to-Image Generation & Diffusion
  • Visual Reasoning & Semantic Alignment
0

Publications

0

Certificates

0

Skills

0

Experiences

Publications

CARINOX: Inference-Time Scaling with Category-Aware Reward-Based Initial Noise Optimization and Exploration

Amir Kasaei, Ali Aghayari, Arash Marioriyad, Niki Sepasian, Shayan Baghayi Nejad, Mohammad Amin Fazli, Mahdieh Soleymani Baghshah, Mohammad Hossein Rohban

Transactions on Machine Learning Research (TMLR) 2026

A reward-guided optimization framework for initial noise in diffusion models, combining discrete exploration and continuous refinement for compositional alignment.

Weeding Out Bad Seeds: Noise-Robust Unlearning for Text-to-Image Diffusion Models

Arian Komaei Koma, Amir Kasaei, Ali Aghayari, Aida Aryafar, Matin Ghiasi, Amirhossein Souri, Mohammad Mosayyebi, Amir Mahdi Sadeghzadeh, Mohammad Hossein Rohban

Under review at NeurIPS 2026

Adaptive noise sampling for unlearning in text-to-image diffusion models.

Education

Master of Science

Computer Engineering

Sharif University of Technology

September 2023 – Present

GPA: 18.54 / 20

Supervisor: Dr. M. H. Rohban
Compositional Generation in T2I Diffusion Models

Bachelor of Science

Computer Engineering

University of Guilan

September 2019 – August 2023

GPA: 19.58 / 20 · Ranked 1st

Supervisor: Dr. S. A. Mirroshandel
Medical Image Analysis

Experiences

Research Assistant

Research Assistant

September 2023 – Present

RIML Lab (Sharif University of Technology)

Supervisor: Dr. M. H. Rohban

Focus: Compositional Generation · T2I Diffusion Models · Visual Reasoning · Multimodal Models

Research Assistant

September 2022 – August 2023

University of Guilan

Supervisor: Dr. S. A. Mirroshandel

Focus: Medical Image Analysis

Teaching Assistant

Head Teaching Assistant of Data Mining

September 2025 – January 2026

Instructor: Dr. Mohammad Amin Fazli

Led course coordination, quiz design, and grading; assisted with lectures, assignments, and student support.

Teaching Assistant of System 2 in AI

February 2025 – June 2025

Instructors: Dr. M. H. Rohban, Dr. M. Soleymani

Topics: Compositional Generation, Inference-time scaling, Test-time adaptation

Material preparation and grading.

Teaching Assistant of Deep Learning

February 2025 – June 2025

Instructor: Dr. M. Soleymani

Topics: Generative Models, VAE, GAN, Diffusion

Quiz and exam design, and grading.

Teaching Assistant of Intelligent Analysis of Biomedical Images

September 2024 – January 2025

Instructor: Dr. M. H. Rohban

Topics: Diffusion and Generative Models

Supported exercises, exam/quiz design, and grading.

Certificates

IELTS

IDP

Dec 2025 · Overall 7.5

Listening 9 · Reading 7.0 · Writing 6.5 · Speaking 7.0

Deep Learning Specialization

DeepLearning.AI

Instructor: Andrew Ng
Aug 2022

Machine Learning Specialization

DeepLearning.AI

Instructor: Andrew Ng
Sep 2022

Fundamentals of Reinforcement Learning

University of Alberta

Aug 2022

Crash Course on Python

Google

Instructor: Christine Rafla
Aug 2022

Honors & Awards

Exceptional Talent Admission at Sharif University of Technology

Sep 2023

Direct admission to M.Sc. in Computer Engineering as an exceptional talented student.

Ranked 1st among B.Sc. Computer Engineering Students

Sep 2019 – Aug 2023

University of Guilan · Highest GPA among all undergraduate Computer Engineering students · Certificate (Persian)

Skills

Machine Learning & Computer Vision

Generative Models Multimodal Models Diffusion Models VLMs Visual Reasoning

Programming & Frameworks

Python PyTorch TensorFlow Keras Java

Data Science & Developer Tools

NumPy Pandas Matplotlib W&B Git

Soft Skills & Leadership

Teamwork Leadership Technical Communication Mentorship Problem Solving Collaboration Time Management

Contact

My Address

Tehran, Tehran Province, Iran

Email Me

a.kasaei@me.com

Designed by Amir Kasaei