Early Warning of Intraoperative Adverse Events via Transformer-Driven Multi-Label Learning
Organizations: Chengdu Institute of Computer Application, Chinese Academy of Sciences, Chengdu 610213, China · University of Chinese Academy of Sciences, Beijing 100049, China · China Zhenhua Research Institute Co., Ltd., Guiyang 550014, China
Abstract
Early warning of intraoperative adverse events plays a vital role in reducing surgical risk and improving patient safety. While deep learning has shown promise in predicting the single adverse event, several key challenges remain: overlooking adverse event dependencies, underutilizing heterogeneous clinical data, and suffering from the class imbalance inherent in medical datasets. To address these issues, we construct the first Multi-label Adverse Events dataset (MuAE) for intraoperative adverse events prediction, covering six critical events. Next, we propose a novel Transformerbased multi-label learning framework (IAENet) that combines an improved Time-Aware Feature-wise Linear Modulation (TAFiLM) module for static covariates and dynamic variables robust fusion and complex temporal dependencies modeling. Furthermore, we introduce a Label-Constrained Reweighting Loss (LCRLoss) with co-occurrence regularization to effectively mitigate intra-event imbalance and enforce structured consistency among frequently co-occurring events. Extensive experiments demonstrate that IAENet consistently outperforms strong baselines on 5, 10, and 15-minute early warning tasks, achieving improvements of +5.05%, +2.82%, and +7.57% on average F1 score. These results highlight the potential of IAENet for supporting intelligent intraoperative decision-making in clinical practice.
Figures & tables
| Models | IAENet | iTransformer | Crossformer | PatchTST | FEDformer | Autoformer | Informer | DLinear | Non-stationary | TFT | SegRNN | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | F1 | AUC | ||
| Hypotension | 5 | 51.45 | 74.83 | 45.64 | 65.76 | 48.32 | 67.29 | 24.30 | 57.58 | 45.90 | 66.69 | 3.02 | 50.69 | 47.65 | 67.96 | 47.65 | 67.69 | 48.37 | 68.66 | 36.71 | 62.59 | 29.23 | 59.19 |
| 10 | 36.21 | 65.85 | 16.91 | 54.79 | 19.14 | 55.50 | 10.58 | 52.80 | 11.59 | 53.11 | 1.07 | 50.10 | 25.16 | 58.35 | 17.81 | 55.20 | 23.10 | 57.40 | 19.09 | 55.54 | 14.47 | 54.55 | |
| 15 | 28.54 | 62.08 | 9.79 | 52.57 | 15.24 | 54.28 | 5.45 | 51.29 | 10.13 | 52.67 | 0.03 | 49.95 | 11.17 | 53.02 | 8.01 | 52.03 | 16.10 | 54.84 | 16.81 | 55.31 | 10.17 | 52.68 | |
| Low DoA | 5 | 61.45 | 76.69 | 54.03 | 69.49 | 37.70 | 66.77 | 2.28 | 50.43 | 51.42 | 68.08 | 6.11 | 50.31 | 49.53 | 67.19 | 49.53 | 67.19 | 47.92 | 66.19 | 6.18 | 51.06 | 54.65 | 70.35 |
| 10 | 56.54 | 73.42 | 49.80 | 67.35 | 43.18 | 63.80 | 0.95 | 50.17 | 43.85 | 64.14 | 11.62 | 50.57 | 37.89 | 61.17 | 43.77 | 64.09 | 31.35 | 58.56 | 7.08 | 51.21 | 43.59 | 64.01 | |
| Loss | ACC | PRE | REC | F1 | AUC | HM |
|---|---|---|---|---|---|---|
| LCRLoss | 91.72 | 61.77 | 69.39 | 65.36 | 80.88 | 0.0828 |
| WBCE | 92.17 | 68.22 | 56.99 | 62.10 | 71.12 | 0.0783 |
| BCE | 92.69 | 71.70 | 57.83 | 64.03 | 74.21 | 0.0731 |
| FL | 92.73 | 73.76 | 54.91 | 62.96 | 73.75 | 0.0727 |
| ASL | 91.22 | 58.98 | 72.08 | 64.87 | 79.86 | 0.0878 |
| PolyLoss | 92.66 | 70.90 | 58.91 | 64.35 | 75.07 | 0.0734 |
| Module | ACC | PRE | REC | F1 | AUC | HM |
|---|---|---|---|---|---|---|
| Informer | 91.28 | 65.74 | 46.92 | 54.76 | 71.54 | 0.0872 |
| + LCRLoss | 90.24 | 55.87 | 62.89 | 59.17 | 79.36 | 0.0976 |
| + FiLM | 92.05 | 70.10 | 51.14 | 59.13 | 70.62 | 0.0795 |
| + TAFiLM | 91.95 | 67.89 | 53.96 | 60.13 | 71.34 | 0.0805 |
| Crossformer | 92.39 | 73.03 | 51.36 | 60.31 | 71.56 | 0.0761 |
| + LCRLoss | 90.80 | 57.88 | 66.80 | 62.02 | 81.17 | 0.0920 |
| Factor | Setting | F1 | AUC |
|---|---|---|---|
| Weighting (G / L) | none | 63.97 | 74.30 |
| inverse | 40.55 / 60.82 | 78.80 / 85.35 | |
| log_inverse | 24.40 / 51.02 | 60.23 / 83.94 | |
| sqrt_inverse | 59.51 / 65.36 | 83.68 / 80.88 | |
| cubic_inverse | 63.91 / 64.26 | 82.33 / 79.44 | |
| Batch Size | 64 | 65.21 | 81.35 |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Training dataset | Validation dataset | Test dataset | P-value | Test | |
|---|---|---|---|---|---|
| (n = 610, 70%) | (n = 88, 10%) | (n = 175, 20%) | |||
| Age, years SD | 59.59 14.15 | 61.22 15.48 | 60.88 12.98 | 0.502 | Kruskal-Wallis |
| Sex, n(%) | 0.776 | Chi-squared | |||
| Female | 257 (42.1) | 38 (43.2) | 79 (45.1) | ||
| Male | 353 (57.9) | 50 (56.8) | 96 (54.9) | ||
| Weight | 62.94 11.99 | 62.87 11.76 | 61.53 10.57 | 0.379 | Kruskal-Wallis |
| Device/Variable | Description | Covariate |
|---|---|---|
| Age | Age of the patient | |
| Sex | Gender of the patient | |
| Weight | Body weight of the patient | |
| Height | Height of the patient | |
| ASA | ASA classification | |
| Orchestra/PPF20_VOL | Propofol infusion volume |
Clinical Risk Prediction
Clinical risk prediction estimates patients’ future health outcomes from medical records, physiological signals, imaging, and other health data.
See all 217 papers