AI Researcher · Robust & Distributed Intelligence

Mohammad Hossein Badiei

I work on robust optimization, information theory, federated learning, computer vision, and robotics, with a focus on building learning systems that remain reliable under uncertainty, heterogeneity, and adversarial conditions.

Robust OptimizationInformation TheoryFederated LearningComputer VisionRobotics
Mohammad Hossein Badiei
01 / Profile

Research at the intersection of learning, optimization, and trustworthy autonomy.

My work spans robust and decentralized learning, adversarial resilience, generative modeling, autonomous systems, and AI for science. I am particularly interested in methods that combine theoretical structure with deployment-aware machine learning.

RESEARCH

Robust & Federated Learning

Resilience to poisoning, heterogeneity, uncertainty, and distributed decision-making constraints.

INTELLIGENCE

Generative & Vision Models

Generative modeling and vision systems for scientific, biomedical, and engineering applications.

AUTONOMY

Optimization & Robotics

Constrained planning, model predictive control, and uncertainty-aware autonomous systems.

02 / Education

Simultaneous tri-degree program with two honours degrees.

2020–2023

M.Sc. in Electrical & Computer Engineering

University of Tehran · Electrical and Computer Engineering Department

GPA 17.66/20
US GPA 3.88/4
2016–2021

B.Sc. (+Honours) in Computer Engineering

Tehran Polytechnic (Amirkabir University of Technology) · Dual-Degree

GPA 18.27/20
Top 3%
2016–2020

B.Sc. (+Honours) in Electrical Engineering

Tehran Polytechnic (Amirkabir University of Technology) · Dual-Degree

GPA 18.58/20
Class Rank: 1st
03 / Recognition

Selected awards & distinctions.

01

National Electrical Engineering Olympiad Gold Medal

Center of Student Olympiad and International Exam Affairs of Iran · 2021

02

UNESCO Distinguished Young Scientist of Basic Science and Technology

UNESCO Club · 2023

03

Full Scholarship for AI Excellence Internship

Tsinghua University, China · 2024

04

Ranked 1st — 6th Young Scientists Festival

Mathematics and Computer Science · 2022

05

Simultaneous Dual Degree and Dual Level Honors

Iran's National Elites Foundation · 2018 & 2020

06

Ranked 1st in the B.Sc. Major

Tehran Polytechnic · 2020

04 / Publications

Selected publications.

01

Robust Peer-to-Peer Federated Learning with Deep Reinforcement Learning Based Client Selection Against Data Poisoning Attacks

K. Kazemi, M. H. Badiei, H. Kebriaei
IEEE Transactions on Artificial Intelligence 7 (1), 262–271 · 2025
02

Resilient Federated Adversarial Learning with Auxiliary-Classifier GANs and Probabilistic Synthesis for Heterogeneous Environments

Y. Haghbin, M. H. Badiei, N. H. Tran, M. J. Piran
IEEE Transactions on Network and Service Management 22 (5), 4998–5014 · 2025
03

Uncertainty-Aware Path Planning with Multiple Constraints for Autonomous Vehicles Based on Nonlinear Reduced-Order Model Predictive Control

M. H. Badiei, S. M. Dashtaki, N. Vafamand, M. J. Piran
IEEE Transactions on Consumer Electronics 71 (2), 3802–3816 · 2024
04

NeuroAtlas: Benchmarking Foundation Models for Clinical EEG and Brain-Computer Interfaces

K. Kontras, T. Osselaer, S. G. Mouslech, A. I. Karaiskou, G. Gagliardi, et al.
arXiv preprint arXiv:2605.14698 · 2026
05

Design of Thin, Wideband Electromagnetic Absorbers with Polarization and Angle Insensitivity Using Deep Learning

A. Shahsavaripour, M. H. Badiei, A. Kalhor, L. Yousefi
Scientific Reports 15 (1), 9426 · 2025
06

Resilient Federated Vision Transformers for Alzheimer's Disease Prediction with Brain Imaging Data Jittering

M. Mohammadi, M. H. Badiei, M. Mashreghi, K. Kazemi, H. Kebriaei
IEEE Transactions on Consumer Electronics · 2026
07

Partial Federated Primal-Dual Optimization for Network-Based Android Malware Detection

M. Mashreghi, M. H. Badiei
2025 IEEE 45th International Conference on Distributed Computing Systems · 2025
08

Generative Robust Design of UWB Metamaterial Structures with Polarization and Angular Stability

A. Shahsavaripour, M. H. Badiei, L. Yousefi, A. Kalhor
IEEE AP-S/URSI · 2025
09

GAN-Driven Image Generation for Metamaterial Absorbers Using Mean and Variance Encoding

A. Shahsavaripour, M. H. Badiei, L. Yousefi, A. Kalhor
33rd International Conference on Electrical Engineering (ICEE), 99–103 · 2025