arrow
Return

Predicting explainable dementia types with LLM-aided feature engineering

delete2025-04-08
delete0
delete
OA
AI
A
Aditya Kashyap *
D
Delip Rao
M
Mary Regina Boland
L
Li Shen
C
Chris Callison-Burch
DOI:10.1093/bioinformatics/btaf156delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Motivation The integration of Machine Learning and Artificial Intelligence (AI) into healthcare has immense potential due to the rapidly growing volume of clinical data. However, existing AI models, particularly Large Language Models (LLMs) like GPT-4, face significant challenges in terms of explainability and reliability, particularly in high-stakes domains like healthcare.Results This paper proposes a novel LLM-aided feature engineering approach that enhances interpretability by extracting clinically relevant features from the Oxford Textbook of Medicine. By converting clinical notes into concept vector representations and employing a linear classifier, our method achieved an accuracy of 0.72, outperforming a traditional n-gram Logistic Regression baseline (0.64) and the GPT-4 baseline (0.48), while focusing on high-level clinical features. We also explore using Text Embeddings to reduce the overall time and cost of our approach by 97%.Availability and implementation All code relevant to this paper is available at: https://github.com/AdityaKashyap423/Dementia_LLM_Feature_Engineering/tree/main.
Keywords:
ALZHEIMERS-DISEASE

Journal

Bioinformatics cover
Bioinformatics
IF:
5.4
Papers:
1.1K
Citations:
17.9W

Organization

U
Univ Penn
Scholars:
4.0K
Papers: 2.6K
Citations: 726