LLM-based Agent Simulation for Maternal Health Interventions: Uncertainty Estimation and Decision-focused Evaluation

Sarah Martinson, Lingkai Kong, Cheol Woo Kim, Aparna Taneja, Milind Tambe
3/25/2025
cs.AI

Official code: sarahmart/LLM-ABS-ARMMAN-prediction (2 stars, MIT, last commit 2025-02-22).

Abstract

Agent-based simulation is crucial for modeling complex human behavior, yet traditional approaches require extensive domain knowledge and large datasets. In data-scarce healthcare settings where historic and counterfactual data are limited, large language models (LLMs) offer a promising alternative by leveraging broad world knowledge. This study examines an LLM-driven simulation of a maternal mobile health program, predicting beneficiaries' listening behavior when they receive health information via automated messages (control) or live representatives (intervention). Since uncertainty quantification is critical for decision-making in health interventions, we propose an LLM epistemic uncertainty estimation method based on binary entropy across multiple samples. We enhance model robustness through ensemble approaches, improving F1 score and model calibration compared to individual models. Beyond direct evaluation, we take a decision-focused approach, demonstrating how LLM predictions inform intervention feasibility and trial implementation in data-limited settings. The proposed method extends to public health, disaster response, and other domains requiring rapid intervention assessment under severe data constraints. All code and prompts used for this work can be found at https://github.com/sarahmart/LLM-ABS-ARMMAN-prediction.

View on arXivView PDF

Code Implementations(1)

MIT
21Shell, PythonOct 29, 20241 years agoMIT

Similar Papers

Cite this paper

@article{martinson2025llmbased,
  title  = {LLM-based Agent Simulation for Maternal Health Interventions: Uncertainty Estimation and Decision-focused Evaluation},
  author = {Sarah Martinson and Lingkai Kong and Cheol Woo Kim and Aparna Taneja and Milind Tambe},
  year   = {2025},
  eprint = {2503.22719},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2503.22719v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2503.22719.svg)](https://codeofpaper.com/paper/2503.22719)

Discussion