Reinforcement Learning for
Experimental Sciences

Bridging the Simulation‑to‑Reality Gap

Workshop at the Neural Information Processing Systems 2026 conference (Paris)

Workshop Motivation

Reinforcement learning has achieved major breakthroughs in simulated environments, yet transferring these advances to real experimental systems remains challenging. This workshop explores how RL can enable adaptive experimentation and real-world scientific discovery while addressing the simulation-to-reality gap.

Experimental Systems as Sequential Decisions

Many scientific experiments can naturally be formulated as sequential decision processes in which each experimental action influences future observations. Reinforcement learning provides a principled framework to optimize such adaptive decision loops.

Simulation-to-Reality Gap

Differences between simulations and physical systems create major challenges for deploying reinforcement learning in laboratory and field experiments.

Reducing Experimental Cost and Time

Many scientific experiments are costly, slow, or resource-constrained. Reinforcement learning methods can help allocate experimental effort more efficiently, accelerating discovery while reducing material, time, and financial costs.

Research Topics

RL for Experimental Platforms

Adaptive experiment design, autonomous laboratories, and sequential optimization in scientific experiments.

Bridging the Sim‑to‑Real Gap

Domain randomization, system identification, hybrid training pipelines, and robust policy transfer.

Efficient and Safe Learning

Sample-efficient reinforcement learning, Hypothesis testing, S-values (E-values), safe exploration, uncertainty-aware decision making.

Field & Clinical Experiments

Agricultural & Medical experimentation, ecological monitoring, and environmental sensing systems.

Human-in-the-Loop Science

Human–AI collaboration in experimentation and participatory research frameworks.

Benchmarks & Infrastructure

Simulation environments, digital twins, and benchmarking platforms for RL in experimental science.

Invited Speakers

Simon Hirländer

University of Salzburg

Koopman Stabilised Symplectic World Models for Safe Offline-to-Online RL in Particle Accelerators

Particle accelerators generate vast amounts of historical data from logs, yet learning-based control often still relies on risky online optimisation. To better utilise this data and avoid online exploration, we present an offline reinforcement learning (RL) workflow. First, we use XSuite to generate high-fidelity trajectories for steering tasks across representative scenarios, including optics variations, alignment errors and jitter, yielding a synthetic dataset of expert and non-expert behaviour. Second, we learn an uncertainty-aware, Koopman-stabilised world model from this data, in which nonlinear beam dynamics are lifted into a latent space with approximately linear, spectrally constrained evolution and a regularised residual term. This structure provides numerically stable long-horizon rollouts and estimates of epistemic uncertainty in latent space. The resulting surrogate environment enables model-based offline RL, where policies are optimised entirely on pre-generated data while epistemic uncertainty is used to detect distribution shift and enforce soft safety constraints. We benchmark these offline RL policies against a PPO agent trained directly in simulation. Results show that policies trained purely offline on the Koopman world model can match online PPO performance without requiring any interaction with the real machine. This demonstrates a safe, reproducible pathway for turning historical accelerator data into effective learning-based control policies.

Sofia Villar

University of Cambridge

RL, Bandits, and the Learning-Earning Dilemma for Clinical Trial Design

In machine learning literature, multi-armed bandits and reinforcement learning are heavily celebrated for optimizing the "learning-earning" trade-off—balancing the exploration of unknown actions to gather data with the exploitation of known actions to maximize immediate rewards. When applying these sequential decision-making frameworks to confirmatory clinical trial design, this trade-off manifests as a profound ethical and statistical tension: maximizing care for current trial participants versus optimizing scientific data gathered to benefit future patients.
However, translating multi-armed bandits and adaptive optimization algorithms into real-world clinical trial architectures introduces significant methodological bottlenecks. Most notably, adaptively collected data destroys the independent and identically distributed (i.i.d.) assumptions of classical statistics, making error control notoriously difficult. While clinical trial efficiency is paramount—especially given small sample size constraints—strict regulatory standardisation demands that Type I error rate inflation must be completely mitigated. To date, the adoption of these methods in confirmatory trials is practically zero, despite gaining some traction in early-phase drug development.
In this talk, I will provide a comprehensive overview of the mathematical, operational, and regulatory challenges that arise when bridging advanced reinforcement learning concepts with clinical trial design. Finally, I will present a suite of specific solutions developed in our group. These approaches successfully close the loop by blending multi-objective adaptive optimization during the design phase with robust, adjusted statistical analysis at the study’s conclusion, ensuring both trial efficiency and regulatory compliance.

Alex Hernandez-Garcia

Université de Montréal & Mila, Canada

GFlowNets to generate materials and molecules with desirable properties and constraints

Scientific progress is key to tackling some of the most pressing challenges for humanity, such as the climate crisis, the threat of pandemics and antibiotic resistance. Machine learning has the potential to help automate and facilitate parts of the scientific discovery cycle, but a large gap remains between the promising results in benchmarks and the practical application of machine learning in the lab. In this talk, I will offer an overview of our recent work on generative modelling based on reinforcement learning through GFlowNets for scientific discovery applications. I will first introduce the GFlowNet framework and highlight its utility for scientific discovery problems, specifically for discovering novel and diverse candidates with desirable properties and constraints. Then, I will illustrate these advantages through our recent work in materials and drug discovery applications, with a focus on how the generation of molecules via sequential decision making allows us to incorporate essential domain knowledge from crystallography, physics and chemistry. For example, we can introduce synthesisability and validity constraints that greatly reduce the vast chemical search space. Finally, I will present our algorithm for multi-fidelity active learning with GFlowNets, designed to better integrate machine learning in the scientific discovery loop.

Ola Engkvist

Discovery Sciences, R&D AstraZeneca & Department of Computer Science and Engineering, Chalmers University of Technology and University of Gothenburg, Gothenburg

Navigating the chemical space with reinforcement learning.

Accelerating the delivery of novel clinical candidates to address unmet medical need remains a central challenge in drug discovery. Machine learning, artificial intelligence, and automation offer a promising framework for addressing this challenge by enabling both the generation of novel molecular structures and the prediction of their properties. Once these generative and predictive components are established, a key remaining question is how to navigate the vast and relevant chemical space efficiently.
The number of potentially interesting molecules accessible in a drug discovery project is extraordinarily large, with the relevant chemical space often estimated to contain up to 10^60 molecules. This scale demands efficient and unbiased strategies for molecular design and prioritization. Reinforcement learning has emerged as a powerful approach for guiding the search toward molecules that satisfy multiple design objectives simultaneously.
The open-source platform REINVENT was among the first tools developed to address this challenge in small-molecule drug discovery. Since its introduction, REINVENT has evolved substantially, with continued methodological and practical enhancements. This presentation will discuss the development of REINVENT and highlight recent advances aimed at further improving chemical space navigation, including 3D molecular generation and human-in-the-loop methodologies.

Program

Morning Session

08:30 – 08:45
Opening remarks and introduction

08:45 – 09:30
Invited talk: Alex Hernandez-Garcia

09:30 – 10:15
Invited talk: Sofia Villar

10:15 – 10:30
No-coffee break

10:30 – 11:30
Contributed spotlight talks (9min+2min Questions)

  • Autonomous Focusing of a Capillary Optic for Angle-Resolved Photoemission Spectroscopy
  • Batch-to-Streaming Deep Reinforcement Learning for Continuous Control
  • Generative Language Modeling for Antibody CDR Grafting and Alignment
  • MARL-based System Optimization for Wildfire Sensing Networks
  • MA-BED: Coordinating Multiple Actors in Bayesian Experimental Design

11:30 – 11:45
Coffee break

11:45 – 12:15
Panel discussion: “Can RL Become a Standard Tool for Experimental Science?”

12:15 – 13:00
Poster session and discussion to facilitate networking

Afternoon Session

13:15 – 14:15
Lunch break

14:15 – 15:00
Invited talk: Ola Engkvist

15:00 – 15:45
Invited talk: Simon Hirländer

15:45 – 16:10
Coffee break

16:10 – 17:15
Contributed spotlight talks (9min+2min Questions)

  • Dynamic Plasma Shape Control Robust to Diagnostic Dropout
  • Heteroscedastic Gaussian Process Q-learning for Sample-Efficient Experimentation
  • In-Context Pure Exploration in Continuous Decision Spaces
  • Constrained sample efficient reinforcement learning for heavy ion accumulation in the Low Energy Ion Ring
  • [Virtual] WSI-Lab: A Pathologist-Inspired, High-Throughput RL Environment for Sequential Whole-Slide Analysis
  • [Virtual] Beyond Reduced-Order Control: Physics-Guided Nine-DOF Rollouts for Wave-Energy RL

17:15 – 17:30
Closing discussion: “Open challenges in RL for experimental sciences”

17:30 – 18:15
Poster session and discussion to facilitate networking

Call for Papers

We invite submissions describing reinforcement learning methods applied to real-world experimental systems, including research papers, system descriptions, benchmarks, and position papers. We will put efforts to include diverse participants. We encourage participation from both machine learning researchers and experimental scientists, practitioners, and interdisciplinary scientists.

Formatting Instructions: We solicit workshop papers following the NeurIPS 2026 main conference paper template. The main text of a submitted paper is limited to nine content pages, including all figures and tables. Additional pages containing references, optional technical appendices and mandatory paper checklist do not count as content pages. The maximum size of submissions is 50 MB. While your submission can contain a supplement or appendix, please note that reviewers are not obliged to review supplementary material.

Reviews: The review process will be double-blind. All submissions must be anonymized and the leakage of any identification information is prohibited. All submissions will undergo peer review by the Program Committee. Each submission will receive at least three reviews, followed by a discussion phase among reviewers and organizers.

Presentation: Accepted contributions will be presented as posters, with a subset selected for spotlight presentations.

What can be submitted: Following the convention of most workshops, we welcome submissions of papers already published in journals or presented at non–machine-learning conferences or workshops. If a paper has appeared in a machine-learning venue, we will consider it only if it includes substantial extensions or new results.

Submission Timeline

Submission deadline: September 01, 2026
Notification: September 29, 2026
Camera-ready: November 29, 2026
Workshop: December 12 (Saturday), 2026

OpenReview Portal

Organizers

Odalric‑Ambrym Maillard

INRIA & University of Lille, France

He is a senior researcher at Inria Lille, in the SCoOL team-project and CRIStAL laboratory, and head of the MARLEL consortium. He is working on the foundations of reinforcement learning and sequential statistics with a focus on regret, robustness, and scientific experimentation. He actively develops interdisciplinary applications of RL in fields such as agronomy and medicine.

Ronald Ortner

Technical University of Leoben, Austria

He is an Associate Professor at the Technical University of Leoben specializing in theoretical reinforcement learning, particularly regret analysis in Markov decision processes. His work focuses on designing sample-efficient algorithms suited to the stringent data regimes of experimental sciences.

Audrey Durand

Université Laval, Canada

She is an Associate Professor at Université Laval and Mila research affiliate, working on interactive machine learning, including reinforcement learning, bandits, and active learning. Her research emphasizes learning systems that actively shape their data acquisition, with applications in healthcare and real-world decision systems.

Sadegh Talebi

University of Copenhagen, Denmark

He is an Assistant Professor at the University of Copenhagen whose research focuses on uncertainty-aware, risk-sensitive reinforcement learning and decision-making. He develops methods for reliable adaptive experimentation under uncertainty, with applications to safety-critical and model-based scientific systems.

Samba Diaw

Higher Polytechnic School, Cheikh Anta DIOP University, Senegal

He is an Associate Professor at Cheikh Anta Diop University of Dakar, working on AI applications in environmental, agricultural, and health domains. His work includes reinforcement learning for sustainable agriculture, ecological modeling, and language technologies for local African languages.

Tristan Fauvel

BaysicLabs, France

He is a machine learning engineer and co-founder of Baysic Labs, with a PhD in computational neuroscience. He specializes in Bayesian modeling and human-in-the-loop AI, translating theoretical advances into real-world applications in biology, medicine, and industry.

Participating Institutions

This workshop brings together researchers from reinforcement learning and experimental sciences across multiple disciplines and institutions.

Program Committee

Area Chairs

  • Tristan Fauvel
  • Debabrota Basu
  • Audrey Durand
  • Ronald Ortner
  • Mohammad Sadegh Talebi

Reviewers

  • Christos Dimitrakakis
  • Elena Smirnova
  • Hippolyte Bourel
  • Yi-Shan Wu
  • Liam Schramm
  • Federico Pavone
  • Erinn M. Grigsby
  • Junya Honda
  • Régis Sabbadin
  • Oliver Mortensen
  • Tristan Fauvel
  • Roger Creus Castanyer
  • Adrienne Tuynman
  • Hyeonah Kim
  • Alexandre Larouche
  • Jiahui Xin
  • Moustapha MBAYE
  • Krishna Chaitanya Balusu
  • M. Sadegh Talebi
  • Samba diaw
  • Waris Radji
  • Jorge Ramirez-Ruiz
  • Udvas Das
  • Odalric-Ambrym Maillard
  • Srinivas Harish
  • Adrien Prevost
  • Simon Hirlaender
  • Dorian Baudry
  • Ronald Ortner
  • Emmanuelle Claeys
  • Caro Strickland

Accepted Papers

We are pleased to present the accepted contributions to the workshop on Reinforcement Learning for Experimental Sciences. Spotlight papers feature the highest-ranked contributions, followed by the poster presentations.

Spotlight Papers

11 papers

Selected contributions presented as spotlight talks.

Spotlight 53
Beyond Reduced-Order Control: Physics-Guided Nine-DOF Rollouts for Wave-Energy RL

Authors: Reuben Stanley

Read abstract

Wave energy is a substantial renewable resource, but commercial devices must extract energy competitively under changing waves. For the coupled D-Spar converter, this creates two linked problems: predicting nonlinear device motion accurately and choosing power-take-off (PTO) damping quickly enough for control. High-fidelity numerical models retain coupled motion, radiation memory and nonlinear forces, but are expensive to evaluate repeatedly. Rather than further reducing the device dynamics, we learn the expensive nonlinear-force correction inside the full nine-degree-of-freedom model. The remaining excitation, radiation, mechanics, PTO and RK4 calculations stay explicit. On development data, a free-recursive 10-s forecast obtains 0.382% macro NRMSE over all displacement and velocity channels. A rank-5 proper orthogonal decomposition (POD5) decoder then reconstructs a 15-action damping–energy curve from only three rollouts, with 0.202% RMS-relative curve error, 100/110 exact sampled optima, exact-or-adjacent selection in every case and 0.0596% maximum regret. The complete measured K3+POD5 controller is not yet 10-Hz real time. Residual RL benefits from POD context but remains 0.606% below greedy. By contrast, exact two-wave sequence MPC gains 4.21% over matched wave-synchronous greedy and 3.57% over canonical 10-Hz greedy across three development seeds, while losing on one seed. These perfect-preview results motivate learning a compact wave-level policy from the MPC value landscape; they do not yet establish RL superiority, unseen-seed performance or real-time deployment.

Spotlight 50
Autonomous Focusing of a Capillary Optic for Angle-Resolved Photoemission Spectroscopy

Authors: Hippolyte Bourel, Alfred J.H. Jones, Søren Ulstrup, Jill Miwa, Philip Hofmann, Panagiotis Karras

Read abstract

Low-dimensional optimization problems in complex experimental environments are common and often take the form of black-box optimization problems. We present a protocol for automated alignment of a capillary optic that focuses extreme-ultraviolet light for angle-resolved photoemission spectroscopy (ARPES). This alignment is a demanding procedure typically performed by expert beamline scientists and is generally inaccessible to non-expert users. In spatially resolved ARPES, a small beam spot is essential for measurements of challenging materials and functional devices. We compare several optimization methods in simulations based on experimental data and demonstrate successful automated focusing in experiments at the SGM4 microARPES beamline of the ASTRID2 synchrotron radiation source in Aarhus.

Spotlight 42
WSI-Lab: A Pathologist-Inspired, High-Throughput RL Environment for Sequential Whole-Slide Analysis

Authors: Ahmed Alagha, Hadi Otrok, Rabeb Mizouni, Jamal Bentahar, Mahdi S. Hosseini

Read abstract

Whole-slide images (WSIs) are gigapixel digital pathology slides where sparse diagnostic evidence makes exhaustive high-magnification inspection computationally expensive. Pathologists instead examine selected regions across magnifications, retain global context and examination history, update their assessment as evidence accumulates, and decide where to look next and when to stop. This naturally frames WSI examination as a partially observable sequential decision problem, for which reinforcement learning (RL) provides a principled framework to learn observation, navigation, and stopping policies. Existing RL and adaptive-sampling methods largely provide task-specific solutions, while existing environments expose limited examination states that do not reflect the richer information available to a pathologist during sequential slide examination and can be costly to execute at RL scale. We introduce WSI-Lab, a reusable, Gymnasium-compatible, pathologist-inspired environment combining the current viewport, global slide context, location and examination history, and an evolving machine diagnostic-belief estimate from acquired evidence. Agents navigate across magnifications and terminate with an explicit diagnosis. Tissue-aware preprocessing, precomputed navigation structures, caching, and efficient state updates reduce repeated online work. RL ablations show that WSI-Lab's complete examination state, tissue-aware navigation, and explicit diagnosis actions support substantially stronger training than reduced variants. Under matched environment workloads, WSI-Lab is additionally up to 2.24x faster while exposing a logical observation payload up to 10.7x larger than existing environments. Code is available at https://anonymous.4open.science/r/wsi-lab-0EEC/.

Spotlight 38
Generative Language Modeling for Antibody CDR Grafting and Alignment

Authors: Ferran Gonzalez, Oliver M. Turnbull, Maliha Sultana, Lorena Roldan Martin, Rohan J Kumar, Tom Diethe, Rebecca Croasdale-Wood, Charlotte Deane, Dino Oglic

Read abstract

Antibodies recognise their targets through hypervariable complementarity-determining regions (CDRs), which are interleaved with conserved frameworks in sequence space, making *de novo* CDR design an infilling problem. Autoregressive models generate residues left-to-right, which precludes full framework context during CDR generation and conflates framework and CDR likelihoods, leaving no natural prompt–response interface for feedback to steer generation. We present GenCDR, a family of LLaMa-based autoregressive language models that read all frameworks as a conditioning prompt and generate all CDRs jointly as a variable-length response, making CDR likelihoods a clean, separable target for reward attribution. The family comprises *IgGenCDR*, *p-IgGenCDR*, and *NanoGenCDR*, trained on unpaired, paired, and nanobody chains, respectively. GenCDR achieves the highest CDR recovery among autoregressive models and produces natural, diverse, human-like CDRs whose likelihoods correlate with fitness and developability assays. The prompt–response boundary also enables principled alignment: reward signals for binding affinity, expression, or developability can be composed to steer CDR generation. Over four rounds of alignment against antibody–antigen co-folding and developability objectives, we find that NanoGenCDR can reach *in silico* structural interface metrics competitive with those of a structure-conditioned diffusion pipeline at roughly half the sampling budget, with more natural, developable designs. The same interface can be extended to integrate experimental feedback, opening a path to closed-loop antibody *de novo* design.

Spotlight 36
MARL-based System Optimization for Wildfire Sensing Networks

Authors: River Bartz, Ramakrishnan Durairajan, Maggie Wigness

Read abstract

Terrestrial-based systems for early wildfire detection often use cameras deployed in remote environments, where images from these cameras are routed through the network and processed by ML models. However, determining an optimal policy for how to route and process collected images is challenging due to highly variable energy availability and network capacity at the edge. Existing methods typically adopt static approaches for determining such policies. They also rely on centralized execution, introducing a potential single point of failure. Multi-agent reinforcement learning (MARL) provides an attractive alternative. However, learned policy performance for MARL-based solutions is dependent on the quality of reward modeling, environment diversity, re-start strategies, and reward credit assignment, all of which can be difficult to define and if done poorly can lead to inconsistent and sub-optimal performance in live deployments. To address these challenges, we propose MARL-based System Optimization for Wildfire Sensing Networks (MARL-OptWSN), a framework for optimizing highly variable image-based wildfire sensing networks. MARL-OptWSN uses a custom training regime built around a Gurobi-based oracle, while leveraging decentralized execution to improve resilience and adapt to changing environmental conditions. Our training approach combines progressive training, oracle guidance, and an agent-specific blame system that identifies decisions responsible for poor outcomes. Empirical experiments show that our approach achieves near-oracle performance with substantially lower decision time, while retaining resilience under highly variable and constrained conditions that negatively affect centralized approaches.

Spotlight 24
MA-BED: Coordinating Multiple Actors in Bayesian Experimental Design

Authors: Marcel Hedman

Read abstract

To date, Bayesian experimental design (BED) has largely studied how a single adaptive policy should choose experiments. Yet many real-world data-acquisition systems distribute experimental capacity across multiple actors, such as laboratories, field teams, or decentralised robots. We introduce multi-actor Bayesian experimental design (MA-BED) as a general paradigm for BED in which experimental decisions are distributed across multiple actors, and within this paradigm focus on a fully cooperative setting with decentralised execution. In this regime, actors share an inferential objective but choose experiments using only their local histories, creating a coordination problem: individually informative policies may collectively acquire redundant information. We formalise this failure through an information-theoretic coordination gap and show that, under conditionally independent acquisition, it is exactly characterised by the total correlation among actor histories. To address this problem, we adapt centralised training with decentralised execution (CTDE) to BED, jointly optimising locally executable policies against the information contained in their pooled experimental histories. We instantiate this framework with DAD-based algorithms trained using a contrastive information objective. Empirically, joint training reduces coordination failure, induces greater actor specialisation under heterogeneous experimental costs, and outperforms independently trained policies without requiring communication at execution. These results establish coordination as an important challenge in Bayesian experimental design when data acquisition is distributed across multiple actors.

Spotlight 19
Constrained sample efficient reinforcement learning for heavy ion accumulation in the Low Energy Ion Ring

Authors: Borja Rodriguez Mateos, Verena Kain, Michael Schenk

Read abstract

The Low Energy Ion Ring (LEIR) at CERN turns long lead-ion pulses from Linac3 into the dense bunches required by the Large Hadron Collider's (LHC) high intensity ion program. It accumulates successive multi-turn injections through six-dimensional phase space painting under electron cooling. The accumulated intensity depends sensitively on nine parameters, and the optimum drifts on every relevant timescale: continuously, as the stripper foil that produces the injected charge state ages, and discontinuously, as the machine is recommissioned every yearly campaign or after critical hardware faults. In this paper, we describe an operational Reinforcement Learning (RL) pipeline deployed on LEIR that treats this drift as the central design constraint and outperforms routine operational performance by more than 10\%. Using four years of archived operational data and system identification between 2023 and 2026, we first quantify the drift: generative topographic maps of the nine-dimensional setting space show that both the sampled region and the high-efficiency region move from year to year, and surrogate models fitted on one campaign transfer to the next with near zero rank correlation. We then show how to choose a low-dimensional observation robust to drift: a joint denoising autoencoder with physics-informed auxiliary heads, benchmarked against variational, quantized, and joint embedding alternatives under measured acquisition noise, compresses $\sim$136\,000 bin longitudinal Schottky spectrograms into nine latents that remain informative across campaigns. Policies are trained on per-year heteroscedastic ensemble digital twins and deployed by warm finetuning on the machine: relocking to a new campaign takes a few hundred machine cycles (1 to 2 dedicated hours of training), $\approx$33$\times$ fewer than training from scratch. Finally, exploration is framed as a constrained Markov Decision Process (MDP) with a learned, per-campaign vacuum risk cost: with a fixed Lagrange multiplier the constrained agent explores an order of magnitude more safely than its unconstrained ancestor.

Spotlight 17
Heteroscedastic Gaussian Process Q-learning for Sample-Efficient Experimentation

Authors: Maximilian Bloor, Max Mowbray, Calvin Tsay, Antonio Del rio chanona

Read abstract

In process experimentation, chemical reactors, and bioprocesses, reinforcement learning agents may only have access to a few hundred trajectories. Each is a real experiment and therefore, the employed algorithm must use them efficiently. Furthermore, any tuning of algorithm hyperparameters can further waste experimental budget. Standard neural Q-learning offers no closed-form uncertainty estimates, and requires additional tuning to achieve satisfactory performance. Non-parametric alternatives can improve sample-efficiency through improved function approximation, however existing methods either fix a constant step or target exploration regret. We propose Heteroscedastic Gaussian Process Q-learning (HGPQ): a closed-form Bayesian gain, obtained by projecting both moments of the value posterior under the distributional Bellman operator, replaces the scalar learning rate of prior Gaussian Process Q-learning, with a heteroscedastic variance update making the gain itself state-action-dependent. Under a fixed, small interaction budget across chemical process-control and biological-growth case studies, we compare HGPQ against specialized low-data RL algorithms and standard vanilla RL algorithms. HGPQ matches or improves on every low-data comparator, and deep actor-critic algorithm.

Spotlight 10
Batch-to-Streaming Deep Reinforcement Learning for Continuous Control

Authors: Riccardo De Monte, Matteo Cederle, Gian Antonio Susto

Read abstract

State-of-the-art deep reinforcement learning (RL) methods have achieved remarkable performance in continuous control tasks, yet their computational complexity is often incompatible with the constraints of resource-limited hardware, due to their reliance on replay buffers, batch updates, and target networks. The emerging paradigm of streaming deep RL addresses this limitation through purely online updates. Existing streaming methods, however, are studied in isolation, trained from scratch until convergence, whereas on-device learning is most useful as a continuation of a pre-trained policy. In this work, we propose two novel streaming deep RL algorithms, Streaming Soft Actor-Critic (S2AC) and Streaming Deterministic Actor-Critic (SDAC), designed as streaming counterparts of SAC and TD3, with performance comparable to state-of-the-art streaming baselines on standard benchmarks. Building on them, we study the batch-to-streaming transition as a problem in its own right: we show that a naive transition can cause a severe and lasting loss of the pre-trained policy, trace the failure to a mismatch between the optimizers used on either side of the switch and show that aligning the optimizers enables streaming deep RL in settings such as Sim2Real.

Spotlight 09
Dynamic Plasma Shape Control Robust to Diagnostic Dropout

Authors: Dmitry Igorevich Sorokin, Maksim Stokolesov, Alexander A. Granovskiy, Igor Prokofyev, Evgeny Adishchev, Maxim Nurgaliev, Eduard Khairutdinov, Georgy Subbotin, Randall Clark, Dmitri M. Orlov

Read abstract

Plasma shape control in tokamaks requires a real-time controller that tracks dynamically changing shape targets while tolerating diagnostic failures. Classical approaches decompose the problem into equilibrium reconstruction followed by a linear controller, and assume a fixed, fully operational sensor set. We present a reinforcement learning agent that addresses both limitations simultaneously. The agent is trained in NSFsim, a high-fidelity tokamak simulator configured for DIII-D, on a curated dataset of 120 experimental plasma shapes. The shape targets are resampled as random step changes every 0.25\,s, exposing the agent to diverse transitions across the full shape envelope. At test time the agent zero-shot tracks dynamic shape sequences; on a held-out static configuration in simulation it achieves a mean shape error of 2.01\,cm, and dynamic trajectory following is demonstrated qualitatively in simulation and on the physical device. Diagnostic dropout randomly masks 30\% of magnetic sensors per episode, yielding a single policy robust to a distribution of sensor subsets used for tokamak operation, without backup controllers or mode-switching logic. An asymmetric actor-critic architecture with privileged equilibrium information improves value estimation under partial observability; an auxiliary shape reconstruction head on the actor enables end-to-end shape reconstruction from raw diagnostics and serves as an interpretability tool for policy analysis. The policy transfers to experimental DIII-D shots, where it directly commands the coil actuators on two dynamic shape maneuvers, and to the independent GSevolve simulator.

Spotlight 05
In-Context Pure Exploration in Continuous Decision Spaces

Authors: Alessio Russo, Yin-Ching Lee, Ryan Welch, Aldo Pacchiano

Read abstract

In active sequential testing, also termed pure exploration, a learner is tasked with the goal to adaptively acquire information so as to identify an unknown ground-truth hypothesis with as few queries as possible. This problem has several motivating applications, including Best-Arm Identification (BAI) in bandits, where actions index hypotheses, and generalized search problems, where strategically chosen queries reveal partial information about a hidden label. In many modern settings, however, the hypothesis, or recommendation space, is continuous: for example, identifying a near optimal action in a continuous-armed bandit, localizing an $\epsilon$-ball contained in a target region, or estimating the minimizer of a function from noisy observations. Existing methods are predominantly frequentist and model-specific, while learned approaches have been limited to finite recommendation spaces. We introduce C-ICPE, a theory-guided learned model for Bayesian fixed-confidence pure exploration with continuous recommendations. C-ICPE meta-trains sequential architectures over a task prior to jointly learn exploration, stopping and recommendations strategies. At inference time, it actively gathers evidence on tasks and identifies an -optimal recommendation without parameter updates.


Poster Papers

18 papers

Accepted contributions presented in the poster session.

Poster 52
Paired Significance Testing for Active Experimental Design Comparisons in Wildfire Model Calibration

Authors: To be added from OpenReview

Read abstract

Autonomous and adaptive experimentation systems are increasingly proposed for the physical sciences, often on the intuition that choosing which measurement to take next, rather than measuring at random or on a fixed schedule, should make better use of a limited observation budget. This project tests this intuition directly on a physical-parameter calibration problem: a stochastic wildfire simulator with three unknown physical parameters, calibrated from field observations under a fixed budget. This project compares an active, uncertainty-seeking sensing policy against random sampling and a fixed monitoring-grid sweep, evaluating all three on held-out calibration loss, burned-area agreement with the true fire, and parameter-recovery error. An initial run with 5 independent fire draws suggested active sensing had a real edge. A properly-powered rerun with 12 independent fire draws did not reproduce it: none of the three strategies are statistically distinguishable on any of the three metrics (paired Wilcoxon signed-rank test, $p \geq 0.456$ pooled over the full budget sweep). A second scenario with a different wind direction, ignition point, and terrain map does show a significant active-sensing advantage on calibration loss ($p=0.0009$), so the effect is not universally absent, it is scenario-dependent and easy to mistake for a general property of the method from a single underpowered pilot. This project argues that claims about adaptive experimental design improving physical-parameter calibration should be treated as empirical, checkable questions specific to a deployment scenario, not assumed a prior, and it reports the initial false positive as a worked example of why.

Poster 45
Bandit Simulation for Average Reward Inference

Authors: To be added from OpenReview

Read abstract

Multi-arm bandit algorithms are increasingly used in online platforms, clinical trials, and social science experiments, but valid statistical inference on their performance remains an open challenge. After deploying bandits, a natural question is whether one can construct a confidence interval for its mean reward and assess whether it reliably outperforms a baseline policy. More broadly, one may wish to assess a new candidate algorithm's expected reward before committing to deployment. Standard inference methods break down because bandit algorithms introduce complex dependencies in the collected data, leading to inflated Type-I error. Moreover, existing inference methods for bandit data only apply to estimands such as the mean reward under a fixed action, which do not depend on the data-collection algorithm. We propose Bandit Simulation for Inference (BSI), a framework that fits a simulator of the bandit environment from on- or off-policy data and uses it to estimate the mean reward under any evaluation policy, including adaptive blackbox algorithms. BSI formally propagates simulator estimation error into the confidence interval construction, requires only weak exploration assumptions on the behavior policy, and avoids importance weighting. We prove that BSI yields asymptotically valid confidence intervals, and demonstrate empirically that it maintains nominal coverage in settings where standard off-policy evaluation methods fail.

Poster 41
From Learned Values to Simple Rules: RL-Guided Strategy Discovery in Food Production

Authors: To be added from OpenReview

Read abstract

Food factories rely on fixed operating procedures because they are stable andeasy to audit. These procedures work well in normal production, but they canreact slowly to rare contamination events. We study whether reinforcementlearning can find when a factory should clean earlier. Our simulator follows areported factory practice of 18 hours of production followed by six hours ofcleaning. It also includes hidden failures, noisy sensor readings, and delayedtests. The current procedure earns a steady profit in normal operation, butloses an average of 3,276 dollars in a simulated crisis. We use simulated futureoutcomes to learn the value of cleaning after an alarm. The learned policyreduces crisis loss by 182 dollars compared with a strong two-alarm rule whilepreserving routine profit. The learned values also point to a simplestrategy: clean after the first strong alarm. On new simulations, this rulereduces crisis loss by 586 dollars and gains 5.83 dollars per day. The main lesson is notthat learning should replace the factory procedure. It is that learning canhelp find a small and useful change to a mature process.

Poster 40
Correlation-aware reward scalarization for reinforcement learning-guided generative molecular design

Authors: To be added from OpenReview

Read abstract

Generative molecular design is often framed as a reinforcement learning (RL) problem in which a policy proposes candidate molecules and is rewarded according to desired properties. Because these properties cannot be measured during generative optimization, they are instead estimated using predictive models, where the reward is typically constructed from multiple in silico molecular property objectives formulated as a multi-parameter optimization (MPO) scalarization. The choice of weights is an important design decision, yet assigning them is difficult: equal weighting implicitly assumes objectives are independent and equally important, and yet drug-discovery objectives are frequently correlated or antagonistic, while manually specified weights rely on expert assumptions that may not reflect statistical dependencies among objectives. We integrate correlation-aware MPO (caMPO), a data-driven scalarization that down-weights redundant objectives, into a molecular scaffold-based RL generator (LibINVENT), recomputing weights online at every optimization step from the molecules generated so far, allowing the reward to adapt to the evolving correlation structure among property objectives. Across 17 scaffolds derived from a public PDE10A dataset, and comparing to a standard equal-weight approach, correlation-aware reward shaping improved property-space quality and alignment with target property profiles, with no significant impact on chemical novelty for both aggregations, and broadened Pareto coverage under arithmetic mean aggregation. These results suggest correlation-aware scalarization may serve as a simple, generator-agnostic reward-design mechanism for RL in experimental molecular science, enabling objective weights to be inferred from data rather than specified manually.

Poster 39
Curriculum Learning with Internal-State-Aware Reward Allocation for Animal Behavior Shaping

Authors: To be added from OpenReview

Read abstract

Training animals to perform complex behavioral tasks is a major bottleneck in systems neuroscience. A common training strategy is shaping: animals begin with easier versions of a task and progress toward the target behavior as their performance improves. This process resembles automated curriculum learning (ACL), where task difficulty is adapted to the learner's performance. Standard ACL formulations, however, typically assume that the learner remains available for continued training. In animal shaping, participation is not guaranteed. Animals may disengage after repeated failures or effortful trials. Reward can help sustain participation, but its motivational value declines with satiety and each session has a finite reward budget. Here, we formulate shaping as a sequential control problem over task difficulty and reward magnitude. Difficulty determines what the animal practices, whereas reward allocation influences whether future training opportunities remain available. We characterize how these decisions should be coordinated using a tractable three-step simulation, then develop a scalable strategy that adapts difficulty from recent performance and reward from the animal's within-session state. Adaptive reward allocation approaches a full-state dynamic-programming benchmark and provides the greatest benefit when declining participation would otherwise shorten training sessions. The same principle reduces sessions to mastery in longer sequential and perceptual-learning tasks. These results establish limited trial availability as a key feature of animal shaping and suggest that efficient training requires managing not only what an animal practices, but also how limited reward is allocated to sustain future learning opportunities.

Poster 35
Learning to Trace: Zero-Shot Connectivity Repair for Vessel Segmentation via Procedural Reinforcement Learning

Authors: To be added from OpenReview

Read abstract

Structured prediction tasks such as tubular segmentation require topology preservation over long-range geometric dependencies. Pixel-wise classification methods are limited in preserving fine-scale connectivity and recovering vascular structure. Connectivity is therefore the property that matters: a segmentation broken into disconnected fragments misrepresents the network even when its pixel overlap is high. Reinforcement learning (RL) offers an alternative by explicitly tracing image geometry, but policies trained without domain adaptation may become overly sensitive to low-level image statistics and generalize poorly under distribution shift. We propose a memory-augmented actor--critic framework that traces tubular structures sequentially, using accumulated trajectory context to resolve local ambiguities. The policy jointly models exploration and branching to reconnect fragmented structures. To enable zero-shot generalization, we train the policy through an adaptive curriculum on synthesized data covering an expansive distribution of noise and geometry.When applied without adaptation to the outputs of a DRIVE-trained segmenter, the policy reduces the number of connected components ($\beta_0$) from 29.1 to 4.5 while recovering vessels missed by the segmenter, increasing centerline recall from 85.8\% to 87.7\% at 94.6\% on-vessel precision. Without retraining or modification, the same policy transfers to ISBI-12 electron-microscopy membranes, despite their distinct appearance and topology, reducing $\beta_0$ from 10.8 to 4.2 while maintaining 94.1\% recall. These results demonstrate that a sequential tracing policy learned entirely from synthetic data can improve topological coherence across substantially different imaging domains.

Poster 33
Stochastic Decision Horizons for Survival-Based Constrained Reinforcement Learning

Authors: To be added from OpenReview

Read abstract

We propose stochastic decision horizons (SDH), a theoretically grounded framework for survival-based constrained reinforcement learning from per-step violation signals. Rather than treating violations as expenditure of a cumulative budget, SDH shortens the effective decision horizon when violations occur, reducing current reward credit and future bootstrapping. We show that this construction represents a geometric-horizon survival constraint and, for a fixed Lagrange multiplier, remains the same SDH problem up to a constant reward offset that acts as an implicit incentive for survival. Using control as inference, we derive two off-policy and replay-compatible, regularized algorithms with post-violation semantics. Absorbing-state ends the decision process, so only surviving decisions pay policy KL, yielding AS-SAC. Virtual termination stops reward credit while the process continues, retaining ordinary discounted KL and yielding VT-MPO. To connect SDH with cumulative-cost CMDPs, we introduce violation-depth profiles. Under exponential-tail single-scale profiles, the SDH statistic determines cumulative cost; departures from this form mark where the connection breaks. Experiments validate both the method and this predicted scope. On the 90-muscle H2190 humanoid in Hyfydy, VT-MPO matches the EWA reference in peak gait realism, with its highest observed gait-match checkpoint at 21M environment steps versus 90M for EWA and markedly more stable training. On Safety Gymnasium, violation-depth profiles predict the regimes in which SDH achieves strong reward-violation trade-offs.

Poster 32
Trust-region constraints govern improvement in a closed-loop protein engineering system

Authors: To be added from OpenReview

Read abstract

Closed-loop design systems are now common in the experimental sciences, but few of them report which part of the loop produced the improvement.We report such a system to increase the activity of large serine recombinases (LSRs), which catalyze unidirectional, site-specific integration of multi-kilobase DNA cargoes, making them attractive for therapeutic gene insertion.Each round aligns a 3B-parameter masked protein language model to activity data with a learn-to-rank objective,draws multi-mutant variants by Gibbs MCMC under a soft trust region on mutational distance,assays them by in vitro transcription-translation, and returns the measurements.Across two campaigns, five rounds and 392 assayed designs, the best variant carries fivesubstitutions and reads 165 times parent activity on the same plate.We then ask which component was responsible.Of the components we tested, only the trust region has an effect we can measure:distance to the previously measured set is inversely correlated with activity, absorbing the whole difference between policy-update schedules.We report these findings as a template for future closed-loop campaigns.

Poster 31
Does an Expert Teacher Help PPO? A Controlled Comparison of Training-Time Guidance Mechanisms on a Reactor Control Benchmark

Authors: To be added from OpenReview

Read abstract

Facilities that consider reinforcement learning for control almost always possess a trusted classical controller, and a recurring proposal is to keep that controller in the training loop as a *teacher*. Many teacher designs exist, but no controlled experiment compares them: published evidence comes from different learners, tasks, and tuning budgets. We provide that experiment. On a nuclear-microreactor load-following benchmark, one on-policy learner (PPO) is trained under six guidance mechanisms and an unguided control, with a one-step nonlinear model-predictive controller (NMPC) as teacher and classical baseline. Every configuration receives its own hyperparameter search, then twenty evaluation seeds disjoint from the tuning seed, with best-checkpoint selection. Peak accuracy does not separate the methods; reliability separates them sharply, in an order given by one design property: how much of the teacher's behavior enters the learner's training data. Mechanisms that inject teacher actions into the on-policy gradient succeed on 0/20 (action blending) and 6/20 (warmstart) seeds, at or below the unguided control's 8/20, while mechanisms that confine the teacher to shaping states, rewards, or a base action succeed on 19/20 to 20/20 ($p \le 4.5\times10^{-4}$, Fisher exact, Holm-corrected). Guidance reduces the training-time constraint-violation rate by up to $17\times$ and eliminates overpower events; a residual configuration trains to teacher-level accuracy with *zero* violations in 200,600 episodes. Deployment-robustness ablations complete the accounting: under plant–model mismatch the learned policies degrade gracefully while the nominal-model teacher incurs errors one to two orders of magnitude larger; under sensor noise the relationship reverses; under process noise the two are comparable. Resampling our own seeds shows a 3-seed version of this study could have reported any unguided success rate from 0% to 100%, so small-seed comparisons of guidance methods are uninformative.

Poster 26
Uncertainty-Aware Kernels as a Practical Alternative to Moment Matching in Gaussian-Process MPC

Authors: To be added from OpenReview

Read abstract

Analytic Moment Matching enables multistep uncertainty propagation in Gaussian-Process Model Predictive Control, but requires specialised derivations. We study an uncertainty-aware RBF kernel over Gaussian-distributed inputs as a practical alternative. For deterministic training inputs, we show that the resulting GP predictor has exactly the same posterior mean and marginal-likelihood objective as analytic Moment Matching. On Pendulum-v1 our method shows comparable performance with an $8\times$ wall-clock speedup and an $11\times$ lower memory footprint.

Poster 22
Compositional Perturbation Planning with a Generative Structural Causal World Model

Authors: To be added from OpenReview

Read abstract

Multi-gene intervention planning is critical to achieve a desired final cell state from either an embryonic stem or an intermediary cell state, while simultaneously modeling epistatic gene interactions. Current perturbation foundation models are not reliably trained to predict multi-gene perturbations and are conditional-mean estimators rather than intervention operators, so composing perturbations under a reinforcement learning (RL) policy queries the model off-distribution. Here, we introduce a generative structural causal world model in which transcription factor (TF) perturbations are modeled as latent interventions with support for multi-perturbations where any k-factor combination is solved in closed form rather than rolled out. On held-out TF overexpressions, the model matches or outperforms baselines on both single and combination perturbations, with cocktails of up to eight TFs, and predicts held-out gene-set epistasis. Epistatic interaction is explicitly learned through a low-rank interaction head trained on multi-TF perturbations by injecting set-dependent latent shifts through the causal graph, allowing the model to predict non-additive effects in held-out, experimentally validated TF combinations. Over this pre-trained world model, we built a policy to design perturbation cocktails that steer human embryonic stem cells (hESCs) to downstream differentiated fates including hematopoietic stem and progenitor cells (HSPC). We implement a goal-conditioned GFlowNet policy that proposes TF cocktails scored by the frozen world model's do-operator under an ensemble-pessimistic reward, penalizing cocktails whose value depends on simulator error. The policy generalizes to held-out fate signatures and recovers the driver TFs of a human embryonic stem cell to HSPC screen alongside other human fetal atlas annotated fates, matching model-predictive-control (MPC) search on broad recovery and exceeding it on top-rank metrics, while searching at roughly 36-fold lower inference cost and proposing far more diverse cocktails. Furthermore, since both the gene-sets that differentiate to the HSPC fate and the fate itself were held out in the world model and the policy, respectively, its zero-shot proposals for HSPC surface more validated answer-key drivers than the search and heuristic baselines at matched budget, nominating canonical blood-lineage regulators including LYL1, KLF1, and KLF2. Together, our work provides a framework for an exact, compositional causal world model allowing the implementation of combinatorial cell-fate design as an amortized, highly efficient policy.

Poster 20
Chain-of-Molecules: Agentic Reasoning in Chemical Space for Constrained Molecular Optimization

Authors: To be added from OpenReview

Read abstract

Chain-of-thought reasoning has enabled language models to solve complex problems through explicit intermediate steps. We ask whether this paradigm can be transferred from language into chemical space. We introduce \emph{Chain-of-Molecules} (CoM), a molecular-generation paradigm in which the intermediate reasoning states are molecules and the transitions between them are explicit chemical edits. In Agentic CoM, a symbolic environment executes valid edits, rejects infeasible ones, and returns the resulting molecule, allowing the model to adapt its subsequent actions, explore alternatives, backtrack, and commit. We evaluate CoM on constrained molecular optimization across five property objectives, requiring generated molecules to attain a requested property value while remaining within a specified graph-edit-distance interval. A controlled comparison of four generation formats isolates the effects of representing intermediate molecular states and interleaving generation with environment execution. Each format follows the same three-stage training pipeline: pretraining teaches molecular syntax and search, supervised finetuning grounds the true property objectives, and GRPO directly optimizes property and structural success. GRPO increases in-distribution joint success by a factor of $4.2$--$4.4$ across all formats. Agentic CoM combines the property success enabled by intermediate molecular states with the structural control provided by environment execution, reaching joint success rates of $0.547$ in distribution and $0.573$ on held-out interpolation targets while producing the most structurally diverse successful solutions across all three target regimes.

Poster 16
Better Than Clinician Care? A Stress Test of Off-Policy Evaluation in ICU Transfusion

Authors: To be added from OpenReview

Read abstract

Artificial intelligence systems are increasingly proposed for treatment decisions in the intensive care unit (ICU), raising the question of how such systems should be evaluated.Treatment decisions by clinicians are assessed in randomized trials and large prospective studies. Decisions produced by learned policies are instead evaluated retrospectively, either by concordance with human experts or by off-policy evaluation (OPE). We therefore test whether \OPE{} supports a stable, credible, and clinically sensible choice among policies for red-blood-cell transfusion in the ICU.We treat clinical decision support as a reinforcement learning policy learned from offline data, and we score 21 learned, guideline-based, and planted control policies with six value estimators on two large databases under a fixed evaluation protocol.Our results show that the estimators do not agree. Every setting produces between three and six different winners, the median policy moves 8 to 12 rank positions across estimators, and a planted control takes the first rank 14 times out of 144. The strongest agreement occurs among importance-weighted estimators, but it is not independent evidence, since all four reuse the same likelihood ratios. Among the policies repeatedly ranked among the best, the database-specific median effective sample sizes are in the single digits despite thousands of held-out stays. Changing only the behaviour model changes the selected policy in 26 of 32 estimator--setting comparisons. Highly ranked policies can still recommend transfusion at normal hemoglobin levels. Retrospective evaluation therefore does not substantiate a claim that an offline policy improves on recorded clinician care.

Poster 13
Reducing cross-sample prediction churn in scientific machine learning

Authors: To be added from OpenReview

Read abstract

Scientific machine learning reports predictive performance. It does not report whether the same prediction would survive a different draw of training data. Across 9 chemistry benchmarks, two classifiers trained on independent bootstraps of the same training set agree on aggregate accuracy to within 1.3–4.2 percentage points but disagree on the class label of 8.0–21.8% of test molecules. We call this gap cross-sample prediction churn. The standard parameter-side techniques (deep ensembles, MC dropout, stochastic weight averaging) do not reduce this gap; two data-side methods do. The first is K-bootstrap bagging, which cuts the rate 40–54% on every dataset at no accuracy cost (K×-ERM compute). The second is twin bootstrap, our proposal: two networks trained jointly on independent bootstraps with a sym-KL consistency loss between their predictions, which at matched 2×-ERM compute reduces churn a further median 45% beyond bagging-K=2. Cross-sample prediction churn deserves a column alongside predictive performance in scientific-ML benchmark reports, because without it the parameter-side and data-side methods are indistinguishable on the metric they actually differ on.

Poster 11
Preference-Graph Policy Improvement for Noisy Biological Sequence Design

Authors: To be added from OpenReview

Read abstract

We introduce Preference-Graph Policy Improvement (PGPI), a method forlearning biological sequence edit policies directly from noisy localcomparisons. PGPI uses per-variant measurement uncertainties to constructa confidence-gated graph of local improvements, converts its edges intoweighted state--action transitions, and trains an iterative edit policy byconfidence-weighted behaviour cloning, without fitting an absolute scalarreward. The construction is motivated by two properties: localdifferencing is insensitive to smoothly varying absolute bias, and acalibrated confidence gate bounds the expected fraction of misdirectededges, subject to a graph-coverage trade-off.Across seven measured fitness landscapes and four noise models, PGPImatches or improves upon direct surrogate ascent most clearly in long,sparse, and noisy regimes, but direct surrogate ascent remains stronger on short, densely sampled landscapes.A controlled corruption study shows that robustness comes from learningover graph-supported transitions rather than from reward-freeness alone;component ablations identify the hard confidence gate and transitionrepresentation as the main source of the effect.

Poster 8
ICMAPE: In-Context Multiagent Pure Exploration

Authors: To be added from OpenReview

Read abstract

In some multi-agent systems, the quantity to be optimized is not an externally specified reward but the information acquired about unknown properties of the environment as done in active sequential hypothesis testing (ASHT) problems. However, the ASHT literature tends to focus on finite single-agent problems with well-specified models, while there is currently a gap for practical multi-agent methods that can perform active sequential testing. We fill this gap with ICMAPE, a Bayesian learning-based framework for decentralized multi-agent pure-exploration driven by inference objectives. ICMAPE converts the fixed-confidence identification objective into a reward derived from inference confidence, so that standard reinforcement learning machinery can be applied to decentralized pure exploration. It jointly learns a centralized neural inference network that estimates a posterior distribution over hypotheses from global trajectory data, and decentralized policies that select actions from local observation histories and learn when to stop collecting data once the target confidence is reached.On two synthetic benchmarks and a Maryland nitrate concentration monitoring task based on real-world data, ICMAPE-TD3 achieves target accuracy with fewer exploration steps.

Poster 6
Bayesian optimization with amortized search-space reparametrization for precision dosing

Authors: To be added from OpenReview

Read abstract

Personalizing drug dosing is a problem of sample-efficient optimization: candidate regimens are tried one at a time on a patient, and evaluations are expensive and potentially risky. Here, we use a pharmacokinetic/pharmacodynamic (PK/PD) model of thedrug, a mechanistic model calibrated on population data with quantified parameter uncertainty, to reparameterize thissearch. A dosing network is trained offline against the model, amortizing the map from patient PK/PD parameters to the regimen predicted to be optimal, and online Bayesian optimization then searches over this learned, low-dimensional representation rather than over raw dosages. Because the network's image is a subspace of pharmacologically coherent regimens, the search remains sensible by construction, improving safety and convergence speed. Unlike model-informed precision dosing, ourmethod admits heterogeneous feedback (symptom scores, preferences, wearables), and applies toany drugs for chronic disease with a PK/PD model where no real-time biomarkers can be measured that provide a comprehensive measure of patients’ state, and disease management implies taking into account patient reported outcomes. On an in-silico population of Parkinson's patients under Levodopa treatment, itconverges to better regimens with fewer trials than baselines optimization methods.

Poster 3
Between Life and Death: Examining Failure Modes of Sparse Reward Designs in Healthcare RL

Authors: To be added from OpenReview

Read abstract

In reinforcement learning (RL) for healthcare, reward functions often encode clinical endpoints like survival and death. This results in a sparse reward structure with non-zero rewards only at terminal transitions. However, the exact reward magnitude and combination of survival rewards and death penalties vary across studies, with the implicit assumption that these choices are interchangeable. In this work, we theoretically and empirically examine three common sparse reward designs: survival-only, death-only, and mixed. We prove that, under the assumptions of terminal-only rewards, guaranteed absorption, and no discounting, the corresponding value functions of the three designs have an equivalence relationship and lead to the same optimal policy; while breaking any of these conditions can break the equivalence in consequential ways. On simulated environments, we verify the theoretical results and demonstrate how relaxing these assumptions affect the equivalence relationship. Finally, we consider a healthcare domain of sepsis treatment based on real-world clinical data, in which all theoretical assumptions are violated. Overall, we find that the death-only reward design consistently underperforms and leads to indecisive policies, while survival-only and mixed rewards perform similarly with differences arising from how they handle trap states and interact with intermediate rewards and discounting. Our work provides the first systematic characterization of these failure modes in healthcare RL, offering theoretically grounded guidance for practitioners on reward design choices.