Search papers, labs, and topics across Lattice.
This paper introduces TRIAGE, a novel framework that leverages Large Language Models (LLMs) to generate dialectical reasoning for risk prediction in irregularly sampled medical time series (ISMTS). By addressing the issue of risk polarization that leads to overconfident binary predictions, TRIAGE enables the generation of continuous risk scores accompanied by interpretable rationales, enhancing both calibration and comparability across patients. Evaluations demonstrate that TRIAGE outperforms competitive baselines with a 3.3% improvement in average AUPRC and an 81% reduction in calibration error, while also providing higher quality clinical reasoning in its rationales.
TRIAGE transforms risk prediction in healthcare by enabling LLMs to provide nuanced, interpretable risk scores rather than binary outcomes, significantly enhancing clinical decision-making.
Clinical early warning systems built on electronic health records, in which clinical observations are recorded as irregularly sampled medical time series (ISMTS), must deliver both calibrated risk scores for patient triage and interpretable rationales that clinicians can verify. Large Language Models (LLMs) have been explored for this task, yet they collapse graded clinical risk into overconfident binary predictions. This risk polarization undermines both calibration and cross-patient comparability. To address this, we propose TRIAGE, a framework that trains an LLM to generate dialectical reasoning over competing clinical outcomes by eliciting outcome-specific rationales. This dialectical formulation mitigates risk polarization, enabling a single LLM to yield continuous risk scores grounded in explicit clinical reasoning. Evaluated on three ISMTS benchmarks, TRIAGE achieves an average AUPRC improvement of 3.3% and reduces calibration error by 81% compared to the competitive baselines. An LLM-as-a-judge assessment further shows that our rationales surpass post-hoc explanations from the baseline by 20% in clinical reasoning quality. The source code is available at https://github.com/HyeongWon-Jang/TRIAGE .
