Multi-Agent Learning for Adaptive Prompt Optimization in Heterogeneous Clinical Narratives

Abstract Free-text narratives within electronic health records, particularly echocardiography reports, contain vital physiological parameters that are frequently absent from structured flowsheet tables. While modern foundation language models offer strong textual comprehension, manual prompt engineering remains an empirical and non-robust procedure. Small prompt changes can cause formatting errors, missed unit conversions, and numerical hallucinations. To address these issues, we formulate prompt optimization as a discrete-action Markov Decision Process and evaluate it on 120,555 echocardiography notes from 38,022 adult ICU patients in the MIMIC-IV database. Unlike conventional rule-based parsers and naive prompts, the proposed reinforcement learning agents optimize extraction using a numeric reward with a strict \pm 2.5% relative tolerance. Both the off-policy Dueling Double Deep Q-Network (D3QN) and on-policy Deep SARSA agents converged within 15 training episodes, achieving a 7.24% relative improvement in extraction accuracy over conventional NLP methods.

Authors

Institutions

Publication Details

Journal
Research Square
Published
2026-11-02
DOI
https://doi.org/10.21203/rs.3.rs-11564090/v1
Primary Topic
Machine Learning in Healthcare
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Multi-Agent Learning for Adaptive Prompt Optimization in Heterogeneous Clinical Narratives

Kusum Deep, Millie Pant, DEEPAK YADAV
3 links
Research Square
Machine Learning in Healthcare
preprint

Multi-Agent Learning for Adaptive Prompt Optimization in Heterogeneous Clinical Narratives

Kusum Deep, Millie Pant, DEEPAK YADAV
preprint en

Abstract

Abstract Free-text narratives within electronic health records, particularly echocardiography reports, contain vital physiological parameters that are frequently absent from structured flowsheet tables. While modern foundation language models offer strong textual comprehension, manual prompt engineering remains an empirical and non-robust procedure. Small prompt changes can cause formatting errors, missed unit conversions, and numerical hallucinations. To address these issues, we formulate prompt optimization as a discrete-action Markov Decision Process and evaluate it on 120,555 echocardiography notes from 38,022 adult ICU patients in the MIMIC-IV database. Unlike conventional rule-based parsers and naive prompts, the proposed reinforcement learning agents optimize extraction using a numeric reward with a strict \pm 2.5% relative tolerance. Both the off-policy Dueling Double Deep Q-Network (D3QN) and on-policy Deep SARSA agents converged within 15 training episodes, achieving a 7.24% relative improvement in extraction accuracy over conventional NLP methods.

Research Square
Indian Institute of Technology Roorkee (IN)
Good health and well-being
Machine Learning in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon