TRACE: Transformer-based Risk Assessment for Clinical Evaluation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Christopoulos, Dionysis, Spanos, Sotiris, Ntouskos, Valsamis, Karantzalos, Konstantinos
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908374581379072
author Christopoulos, Dionysis
Spanos, Sotiris
Ntouskos, Valsamis
Karantzalos, Konstantinos
author_facet Christopoulos, Dionysis
Spanos, Sotiris
Ntouskos, Valsamis
Karantzalos, Konstantinos
contents We present TRACE (Transformer-based Risk Assessment for Clinical Evaluation), a novel method for clinical risk assessment based on clinical data, leveraging the self-attention mechanism for enhanced feature interaction and result interpretation. Our approach is able to handle different data modalities, including continuous, categorical and multiple-choice (checkbox) attributes. The proposed architecture features a shared representation of the clinical data obtained by integrating specialized embeddings of each data modality, enabling the detection of high-risk individuals using Transformer encoder layers. To assess the effectiveness of the proposed method, a strong baseline based on non-negative multi-layer perceptrons (MLPs) is introduced. The proposed method outperforms various baselines widely used in the domain of clinical risk assessment, while effectively handling missing values. In terms of explainability, our Transformer-based method offers easily interpretable results via attention weights, further enhancing the clinicians' decision-making process.
format Preprint
id arxiv_https___arxiv_org_abs_2411_08701
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TRACE: Transformer-based Risk Assessment for Clinical Evaluation
Christopoulos, Dionysis
Spanos, Sotiris
Ntouskos, Valsamis
Karantzalos, Konstantinos
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
We present TRACE (Transformer-based Risk Assessment for Clinical Evaluation), a novel method for clinical risk assessment based on clinical data, leveraging the self-attention mechanism for enhanced feature interaction and result interpretation. Our approach is able to handle different data modalities, including continuous, categorical and multiple-choice (checkbox) attributes. The proposed architecture features a shared representation of the clinical data obtained by integrating specialized embeddings of each data modality, enabling the detection of high-risk individuals using Transformer encoder layers. To assess the effectiveness of the proposed method, a strong baseline based on non-negative multi-layer perceptrons (MLPs) is introduced. The proposed method outperforms various baselines widely used in the domain of clinical risk assessment, while effectively handling missing values. In terms of explainability, our Transformer-based method offers easily interpretable results via attention weights, further enhancing the clinicians' decision-making process.
title TRACE: Transformer-based Risk Assessment for Clinical Evaluation
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2411.08701