Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Causal-Transformer with Adaptive Mutation-Locking for Early Prediction of Acute Kidney Injury

About

Accurate early prediction of Acute Kidney Injury (AKI) is critical for timely clinical intervention. However, existing deep learning models struggle with irregularly sampled data and suffer from the opaque "black-box" nature of sequential architectures, strictly limiting clinical trust. To address these challenges, we propose CT-Former, integrating continuous-time modeling with a Causal-Transformer. To handle data irregularity without biased artificial imputation, our framework utilizes a continuous-time state evolution mechanism to naturally track patient temporal trajectories. To resolve the black-box problem, our Causal-Attention module abandons uninterpretable hidden state aggregation. Instead, it generates a directed structural causal matrix to identify and trace the exact historical onset of severe physiological shocks. By establishing clear causal pathways between historical anomalies and current risk predictions, CT-Former provides native clinical interpretability. Training follows a decoupled two-stage protocol to optimize the causal-fusion process independently. Extensive experiments on the MIMIC-IV cohort (N=18,419) demonstrate that CT-Former significantly outperforms state-of-the-art baselines. The results confirm that our explicitly transparent architecture offers an accurate and trustworthy tool for clinical decision-making.

Weizhi Nie, Haolin Chen• 2026

Related benchmarks

TaskDatasetResultRank
Acute Kidney Injury PredictionMIMIC-IV 0h lead time
AUROC88.72
10
Acute Kidney Injury PredictionMIMIC-IV 6h lead time
AUROC0.8471
10
Acute Kidney Injury PredictionMIMIC-IV 12h lead time
AUROC0.8098
10
Acute Kidney Injury PredictionMIMIC-IV 18h lead time
AUROC0.7782
10
Acute Kidney Injury PredictionMIMIC-IV 24h lead time
AUROC76.48
10
Showing 5 of 5 rows

Other info

Follow for update