Explainable multi-modal fusion-based federated learning for mortality prediction in energy-constrained healthcare systemsShow others and affiliations
2026 (English)In: IEEE Access, E-ISSN 2169-3536, Vol. 14, p. 6146-6166
Article in journal (Refereed) Published
Abstract [en]
The growing demand for privacy-preserving and reliable AI solutions in healthcare has made Federated Learning (FL) a captivating approach for clinical decision support systems. However, despite its promise, many FL approaches struggle to balance predictive accuracy with explainability, an essential requirement for clinical approval. To address this gap, this study presents an explainable multi-modality FL framework that consolidates a transformer model for unstructured clinical notes with a tree-based model for structured patient data. Using both data modalities, the proposed hybrid architecture empowers robust mortality prediction while ensuring traceability in model decision-making. The framework is evaluated in the large-scale MIMIC-IV dataset within a decentralized hospital environment, where the global fusion-based model achieves an AUC of 0.89 that surpasses traditional baselines while providing interpretable insights on clinical outcomes. These findings validate a scalable and reliable solution for the deployment of explainable AI (XAI) in federated healthcare systems, spanning the critical gap between model performance and clinical confidence.
Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2026. Vol. 14, p. 6146-6166
Keywords [en]
attention-based fusion, explainable AI (XAI), Federated learning, mortality prediction, multi-modality data, transformer models
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:umu:diva-249160DOI: 10.1109/ACCESS.2025.3650610Scopus ID: 2-s2.0-105027756830OAI: oai:DiVA.org:umu-249160DiVA, id: diva2:2033963
2026-01-302026-01-302026-01-30Bibliographically approved