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.
Amir Kasaei, Ali Aghayari, Arash Marioriyad, Niki Sepasian, Mohammad Amin Fazli, Mahdieh Soleymani Baghshah, Mohammad Hossein Rohban
NeurIPS 2025 Workshop: Generative and Protective AI for Content Creation (GenProCC)
A correlation study of compositional text-to-image metrics, benchmarking category-specific and cross-category agreement with human evaluations.
Amir Kasaei, Mohammad Hossein Rohban
NeurIPS 2025 Workshop: Generative and Protective AI for Content Creation (GenProCC)
A taxonomy of attribute, relation, and object hallucinations in text-to-image models, framing hallucination as an upper bound for richer evaluation.
Amir Kasaei, Arash Marioriyad, Mahbod Khaleti, Mohammad Amin Fazli, Mahdieh Soleymani Baghshah, Mohammad Hossein Rohban
ICLR 2026 Workshop: From Human Cognition to AI Reasoning (HCAIR)
RebusBench is a benchmark of 1,164 rebus puzzles for evaluating abstract cognitive visual reasoning in LVLMs.
Arian Komaei Koma, Amir Kasaei, Ali Aghayari, AmirMahdi Sadeghzadeh, Mohammad Hossein Rohban
CVPR 2026 Workshop: Machine Unlearning for Computer Vision (MUC)
Evaluating unlearned diffusion models through the lens of compositionality.
Arian Komaei Koma, Amir Kasaei, Aida Aryafar, Matin Ghiasi, Ali Aghayari, Amirhossein Souri, Mohammad Mosayyebi, Amir Mahdi Sadeghzadeh, Mohammad Hossein Rohban
Under review at ICLR 2027
An adaptive, concept-conditioned noise sampling strategy that improves robust concept erasure across random initializations while preserving non-target content and generation quality.
Arian Komaei Koma, Amir Kasaei, Amir Mahdi Sadeghzadeh, Mohammad Hossein Rohban
Under review at IEEE Conference on Secure and Trustworthy Machine Learning
A black-box embedding-aware adversarial prompting framework for attacking unlearned text-to-image diffusion models.