RobIn: A Robust Interpretable Deep Network for Schizophrenia Diagnosis

Organisciak, Daniel, Shum, Hubert, Nwoye, Ephraim and Woo, Wai Lok (2022) RobIn: A Robust Interpretable Deep Network for Schizophrenia Diagnosis. Expert Systems with Applications, 201. p. 117158. ISSN 0957-4174

[img]
Preview
Text
1-s2.0-S0957417422005498-main.pdf - Published Version
Available under License Creative Commons Attribution 4.0.

Download (2MB) | Preview
[img]
Preview
Text
Schizophrenia_manuscript.pdf - Accepted Version
Available under License Creative Commons Attribution 4.0.

Download (1MB) | Preview
Official URL: https://doi.org/10.1016/j.eswa.2022.117158

Abstract

Schizophrenia is a severe mental health condition that requires a long and complicated diagnostic process. However, early diagnosis is vital to control symptoms. Deep learning has recently become a popular way to analyse and interpret medical data. Past attempts to use deep learning for schizophrenia diagnosis from brain-imaging data have shown promise but suffer from a large trainingapplication gap - it is difficult to apply lab research to the real world. We propose to reduce this training-application gap by focusing on readily accessible data. We collect a data set of psychiatric observations of patients based on DSM-5 criteria. Because similar data is already recorded in all mental health clinics that diagnose schizophrenia using DSM-5, our method could be easily integrated into current processes as a tool to assist clinicians, whilst abiding by formal diagnostic criteria. To facilitate real-world usage of our system, we show that it is interpretable and robust. Understanding how a machine learning tool reaches its diagnosis is essential to allow clinicians to trust that diagnosis. To interpret the framework, we fuse two complementary attention mechanisms, ?squeeze and excitation? and ?self-attention?, to determine global attribute importance and attribute interactivity, respectively. The model uses these importance scores to make decisions. This allows clinicians to understand how a diagnosis was reached, improving trust in the model. Because machine learning models often struggle to generalise to data from different sources, we perform experiments with augmented test data to evaluate the model?s applicability to the real world. We find that our model is more robust to perturbations, and should therefore perform better in a clinical setting. It achieves 98 accuracy with 10-fold cross-validation.

Item Type: Article
Additional Information: Funding information: The project was supported in part by the Royal Society (Ref: IES\R2\181024).
Uncontrolled Keywords: Schizophrenia, Deep Learning, Self-Attention, Interpretability, Out-of-Distribution
Subjects: G400 Computer Science
G500 Information Systems
Department: Faculties > Engineering and Environment > Computer and Information Sciences
Depositing User: Elena Carlaw
Date Deposited: 11 Apr 2022 16:11
Last Modified: 17 May 2022 11:15
URI: http://nrl.northumbria.ac.uk/id/eprint/48871

Actions (login required)

View Item View Item

Downloads

Downloads per month over past year

View more statistics