This repository implements LLMCARE, a speech-based NLP pipeline for early detection of Alzheimer’s disease and related dementias (ADRD). The pipeline integrates:
- Transformer embeddings with handcrafted linguistic features.
- Synthetic data augmentation using large language models (LLMs).
- LLMs as classifiers (text-only and multimodal).
- External generalizability evaluation using the DementiaBank Delaware Corpus.
The repository is organized into five components, each with its own README and runnable code.
- Evaluates ten transformer models (general-purpose + biomedical/clinical).
- Compares frozen, last-layer, and full fine-tuning.
- Incorporates 110 handcrafted linguistic features.
- Develops a fusion classifier combining embeddings + features. 👉 See Component 1 README
- Uses LLMs (e.g., LLaMA-8B/70B, MedAlpaca-7B, Ministral-8B, GPT-4o) to generate label-conditioned synthetic speech transcripts.
- Augmented data boosts F1 scores (e.g., MedAlpaca +2× augmentation improved F1 to 85.7). 👉 See Component 2 README
- Evaluates fine-tuned LLMs as direct classifiers for speech transcripts.
- Demonstrates significant performance gains with fine-tuning. 👉 See Component 3 README
-
Benchmarks multimodal LLMs (GPT-4o, Qwen 2.5 Omni, Phi-4) for audio-text classification.
-
Includes two subfolders:
-
Qwen → Fine-tuning with LLaMA-Factory.
- Train with:
llamafactory-cli train train.yaml - Inference with:
test_audio_classification.py
- Train with:
-
Phi4 → Fine-tuning with native scripts.
- Train with:
bash run.shorpython finetune.py - Inference with:
bash test.shorpython test.py👉 See Component 4 README
- Train with:
-
- Tests pipeline generalizability on an MCI-only cohort (n=205).
- Tasks: Cookie Theft, Cinderella recall, procedural discourse.
- Pipeline achieved F1 = 72.8 (AUC = 69.6) with MedAlpaca augmentation.
- Note: This codebase is the same as Component 1 (ADReSSo pipeline). To run, simply update data paths in configs to point to the Delaware dataset.