AutoHealth: An Uncertainty-Aware Multi-Agent System for Autonomous Health Data Modeling

Read full text
Authors: Tong Xia, Weibin Li, Gang Liu, Yong Xiang Li
Cited by: 0

Abstract

LLM-based agents have demonstrated strong potential for autonomous machine learning, yet their applicability to health data remains limited. Existing systems often struggle to generalize across heterogeneous health data modalities, rely heavily on predefined solution templates with insufficient adaptation to task-specific objectives, and largely overlook uncertainty estimation, which is essential for reliable decision-making in healthcare. To address these challenges, we propose \textit{AutoHealth}, a novel uncertainty-aware multi-agent system that autonomously models health data and assesses model reliability. \textit{AutoHealth} employs closed-loop coordination among five specialized agents to perform data exploration, task-conditioned model construction, training, and optimization, while jointly prioritizing predictive performance and uncertainty quantification. Beyond producing ready-to-use models, the system generates comprehensive reports to support trustworthy interpretation and risk-aware decision-making. To rigorously evaluate its effectiveness, we curate a challenging real-world benchmark comprising 17 tasks across diverse data modalities and learning settings. \textit{AutoHealth} completes all tasks and outperforms state-of-the-art baselines by 29.2\% in prediction performance and 50.2\% in uncertainty estimation.

Fields of Study & Subjects

Artificial Intelligence in Healthcare and Education Machine Learning in Healthcare Artificial Intelligence in Healthcare

Publication details

Journal / source
arXiv (Cornell University)
Publication date
1 February 2026
Document type
Preprint