|
|
--- |
|
|
tags: |
|
|
- Multi-task Learning |
|
|
- Explainable AI |
|
|
- Prognosis |
|
|
--- |
|
|
# Multi-task Learning Approach for Intracranial Hemorrhage Prognosis |
|
|
|
|
|
This repository contains the checkpoints of deep learning intracranial hemorrhage (ICH) prognosis models, used in the paper: 'Multi-task Learning Approach for Intracranial Hemorrhage Prognosis'. |
|
|
Particularly, it contains the checkpoints of our proposed method, where we introduced a multi-task learning approach to jointly predict ICH, Glasgow Coma Scale and age, improving performance and interpretability. |
|
|
|
|
|
The checkpoints are trained on the open-source Head-CT 2D/3D images with and without ICH prepared for Deep Learning dataset. You can access the dataset at this [link](https://digital.csic.es/handle/10261/275792). |
|
|
|
|
|
Please refer to the [original GitHub repository](https://github.com/Trustworthy-AI-UU-NKI/MultitaskLearning_ICH_Prognosis) for the code. |
