Skip to content
AI Atlas
PaperActive

SafeImpute: Reliable Clinical Data Imputation via Conformal Selection

arxiv.org/abs/2607.05613

quality89

Updated 7 h ago · first seen 11 Sept 2026

paper_01M294FSA8WH7DEWYQ1BEQH1ZC

Published
11 Sept 2026
T1 · 7 h ago
arXiv
2607.05613
T1 · 7 h ago
Category
cs.LG
T1 · 7 h ago

Abstract

Clinical care often relies on key laboratory indicators, yet real-world patient visits are sparse and tests are ordered irregularly, leading to pervasive missingness. While many imputation methods improve average accuracy, they provide limited guidance on which imputed values are reliable enough for high-stakes downstream use. In this work, we study reliable clinical imputation, aiming to produce accurate imputations while selectively releasing the reliable results, with statistical control over clinically unacceptable errors. To achieve this goal, we propose SafeImpute, a reliable imputation framework for irregular and sparse clinical longitudinal records. SafeImpute constructs an event graph that captures both intra-patient temporal trajectories and inter-patient clinical similarity, and learns imputations with a two-relation GNN and adaptive fusion, regularized by an auxiliary masked reconstruction objective. For reliability guarantees, SafeImpute converts a proxy risk score into conformal p-values and applies the Benjamini--Hochberg procedure to control the false discovery rate (FDR) of unacceptable errors among released imputations at a user-specified tolerance. Experiments on our Mayo Clinic data, the public MIMIC-III and MIMIC-IV datasets show that SafeImpute achieves strong imputation accuracy while providing reliable error control, outperforming diverse baselines in both standard imputation evaluation and FDR-controlled selective-release evaluation.

Authors 5

Xinrui He, Mengting Ai, Junting Wang, Curtiss B. Cook, Jingrui He

Specification

Official page

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

Arxiv announce type
replace

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

arXiv id
2607.05613

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

Categories
cs.LG

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

DOI
10.1145/3770855.3817967

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

PDF

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

Primary category
cs.LG

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

Published
11 Sept 2026

Source:arXiv (Atom API + RSS)T1observed 7 h agohigh

Each value shows its source, tier and observation time. Conflicting claims are kept side by side and flagged — never averaged. How AI Atlas records facts →

Provenance

Attributed facts

10

Source tiers

T110

Freshest observation

7 h ago

Conflicts

None