M. M. Azarbeik
CV

PhD candidate, TU WienResearcher, Medical University of Vienna

Vienna, Austria

Mohammad Mahdi Azarbeik

I work on offline reinforcement learning for clinical decision support in intensive care, and on evaluating when its recommendations can be trusted.

Research interests

  • Offline reinforcement learning
  • Off-policy evaluation
  • Clinical decision support
  • Large language models
  • State estimation
Scroll

About

Portrait of Mohammad Mahdi Azarbeik

I’m Mahdi, a machine-learning researcher working on offline reinforcement learning for critical care.

At the Medical University of Vienna, I develop and validate deep reinforcement learning methods for decision support in intensive care, and use off-policy evaluation to assess learned treatment policies on retrospective ICU data. I work with intensive-care physicians to check that the recommendations are clinically plausible.

I am also a PhD candidate in Computer Science at TU Wien. My background is in mechanical engineering, where I worked on state estimation and sensor fusion; the common thread is quantifying how reliable an estimate is.

Current position
Researcher, Medical University of Vienna since July 2025
Doctorate
PhD candidate, Computer Science (Informatics), TU Wien since 2023
Education
M.Sc. and B.Sc., Mechanical Engineering Sharif University of Technology · K. N. Toosi University of Technology, 2016 to 2023

Research approach

  1. 01

    Explicit uncertainty

    From Kalman filtering to off-policy evaluation, an estimate is reported together with how far it could be wrong.

  2. 02

    Multi-method evaluation

    A treatment policy that looks best under one off-policy estimator may not under another, so I compare several before drawing conclusions.

  3. 03

    Clinical collaboration

    Learned treatment strategies are reviewed with intensive-care physicians for clinical plausibility.

Research

Machine learning for decisions under uncertainty.

My work spans state estimation from noisy sensors, reinforcement learning and language models, and clinical decision support. Across all three, the aim is to learn from imperfect data and to quantify how far the results can be trusted.

  1. 01

    Sensing & estimation

    Kalman filtering and sensor fusion that turn low-cost, noisy sensor signals into reliable estimates of motion, together with their remaining uncertainty.

    • Kalman filtering
    • Sensor fusion
    • SLAM

    M.Sc. · Sharif University of Technology

  2. 02

    Learning systems

    Offline reinforcement learning, which learns decision policies from logged data rather than by trial and error, and off-policy evaluation of those policies; also hallucination detection in large language models.

    • Offline RL
    • Off-policy evaluation
    • LLMs

    PhD · TU Wien

  3. 03

    Clinical decision support

    Treatment strategies for sepsis and renal replacement therapy, learned from intensive-care records, evaluated with several off-policy methods and reviewed with physicians for clinical plausibility.

    • ICU time series
    • MIMIC-IV
    • Decision support

    Medical University of Vienna

Selected projects

Five projects across three areas.

Objectives and outcomes in brief; the papers, code and data are linked from each project.

  1. Schematic

    Clinical decision support2025First author

    Treatment strategies for sepsis

    Learning fluid and vasopressor dosing policies for sepsis from intensive-care records (MIMIC-IV), and comparing off-policy evaluation methods, whose choice can change which policy appears best.

    Conference abstract in Journal of Critical Care (ICCAI 2025, Vienna); full paper in preparation.

    AbstractCitation

  2. Schematic

    Clinical decision support2024 to 2025Co-author

    Timing renal replacement therapy

    Learning when to start renal replacement therapy in critically ill patients with acute kidney injury, using reinforcement learning with high-confidence off-policy evaluation.

    Conference abstract in Journal of Critical Care (ICCAI 2024); preprint on Research Square (2025).

    AbstractPreprint

  3. Schematic

    Learning systems2024TU Wien team

    Hallucination detection in language models

    Detecting hallucinations in language-model outputs (SemEval-2024 Task 6) with an ensemble that combines black-box detectors and detectors reading the model’s internal activations.

    Ranked 3rd in the model-aware track and 8th in the model-agnostic track; paper in the SemEval-2024 proceedings (ACL).

    Paper (PDF)DOITeam code

  4. Graphical abstract: estimated arm trajectories (EKF and SRUKF, with position measurements from SLAM or VICON). Silent.

    Sensing & estimation2023First author

    Inertial motion capture with SLAM

    Fusing drifting inertial sensors with camera-based SLAM in extended and square-root unscented Kalman filters to track a human arm, validated against an optical motion-capture system.

    Journal article in Measurement (2023), with an open dataset and a graphical-abstract video (MP4).

    DOIarXivDataset

  5. Schematic

    Sensing & estimation2025 to 2026Horizon Europe

    AI-enabled wearables: EU-TRAINS

    Lead researcher for the Ludwig Boltzmann Institute for Digital Health and Patient Safety on a Horizon Europe project developing AI-enabled wearables for multisport training and remote healthcare.

    Funded by Horizon Europe, grant agreement 101130495.

    CORDISProject site

Publications

Papers, preprints & abstracts.

Six publications in machine learning for intensive care, language models and sensor fusion. Also listed on Google Scholar and ORCID.

Showing 6 of 6

2026

  1. Learning and evaluating improved reinforcement learning-based policies for sepsis treatment on MIMIC-IV

    M. M. Azarbeik, L. Kapral, R. Weiss, C. Heitzinger, O. Kimberger

    Journal of Critical Care 92, 155385 Conference abstract · ICCAI 2025

    A full-length manuscript is in preparation.

    DOIProject

2025

  1. Optimized renal replacement therapy decisions in intensive care: A reinforcement learning approach

    L. Kapral, R. Bologheanu, M. M. Azarbeik, A. Bilir, R. Weiss, S. Bartos, S. Schaller, C. Heitzinger, E. Schaden, O. Kimberger

    Research Square Preprint

    DOIProject

  2. Optimal timing for renal replacement therapy in critically ill patients using reinforcement learning algorithms

    L. Kapral, M. M. Azarbeik, R. Weiss, R. Bologheanu, C. Heitzinger, O. Kimberger

    Journal of Critical Care 86, 154964 Conference abstract · ICCAI 2024

    DOI

2024

  1. TU Wien at SemEval-2024 Task 6: Unifying model-agnostic and model-aware techniques for hallucination detection

    V. Arzt, M. M. Azarbeik, I. Lasy, T. Kerl, G. Recski

    Proceedings of the 18th International Workshop on Semantic Evaluation (SemEval-2024), ACL, pp. 1183 to 1196 Workshop paper

    PDFDOICode

2023

  1. Augmenting inertial motion capture with SLAM using EKF and SRUKF data fusion algorithms

    M. M. Azarbeik, H. Razavi, K. Merat, H. Salarieh

    Measurement 222, 113690 Journal article

    DOIarXivDataVideo

2022

  1. An overview of the design experience and group analysis of a spinning ride from the perspective of engineering education

    A. Meghdari, S. M. J. Zolanvari, H. Izanlo, M. S. Tohidi Nafe, M. M. Azarbeik, et al.

    Iranian Journal of Engineering Education 24(94), pp. 39 to 60 Journal article · in Persian

    DOI

Experience

Positions and education.

  1. B.Sc., Mechanical EngineeringK. N. Toosi University of Technology, Tehran2016 to 2020
  2. M.Sc., Mechanical EngineeringSharif University of Technology
    State estimation and sensor fusion
    2020 to 2023
  3. PhD & University AssistantTU Wien
    Offline reinforcement learning, off-policy evaluation, language models
    PhD since 2023 · Assistant Oct 2023 to Jan 2025
  4. Researcher, EU-TRAINSLudwig Boltzmann Institute for Digital Health and Patient SafetyFeb 2025 to Apr 2026
  5. ResearcherMedical University of Vienna
    Anaesthesia & intensive care medicine
    Since Jul 2025

Teaching assistant

  • Reinforcement LearningTU Wien
  • Generative AITU Wien
  • Data-oriented Programming ParadigmsTU Wien
  • Numerical AnalysisK. N. Toosi University of Technology

Peer review

  • PLOS Digital Health
  • Measurement
  • Journal of Intelligent & Robotic Systems
  • IEEE Sensors Journal

Toolkit

  • Python
  • PyTorch
  • TensorFlow
  • scikit-learn
  • SQL
  • BigQuery
  • PostgreSQL
  • Google Cloud
  • Git
  • MIMIC-IV
  • ViennaAIdb
  • MATLAB
  • ROS
  • Arduino