Large-Language-Models-for-Assisting-with-Mental-Health-Diagnostic-Assessments

Exploring The Potential of Large Language Models for Assisting with Mental Health Diagnostic Assessments

💿 Model Link

🔗 DiagnosticLlama Model Link

ℹ️ Code Snippet for Running the Model

from transformers import AutoModelForCausalLM, AutoTokenizer

model_path = "PATH_TO_MODEL_REPO"

tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(
    model_path,
    device_map="auto",
    torch_dtype='auto'
).eval()

# Prompt content: "<INSERT DIAGNOSIS PROMPT>"
messages = [
    {"role": "user", "content": "<INSERT DIAGNOSIS PROMPT>"}
]

input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
output_ids = model.generate(input_ids.to('cuda'))
response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)

''' Model Response
[['Feeling-bad-about-yourself-or-that-you-are-a-failure-or-have-let-yourself-or-your-family-down', 'yes'],
['Feeling-down-depressed-or-hopeless', 'yes'],
['Feeling-tired-or-having-little-energy', 'yes'],
['Little-interest-or-pleasure-in-doing ', 'yes'],
['Moving-or-speaking-so-slowly-that-other-people-could-have-noticed-Or-the-opposite-being-so-fidgety-or-restless-that-you-have-been-moving-around-a-lot-more-than-usual', 'no'],
['Poor-appetite-or-overeating', 'no'],
['Thoughts-that-you-would-be-better-off-dead-or-of-hurting-yourself-in-some-way', 'no'],
['Trouble-concentrating-on-things-such-as-reading-the-newspaper-or-watching-television', 'yes'],
['Trouble-falling-or-staying-asleep-or-sleeping-too-much', 'no']]
'''
print(response)

💾 Dataset Links

🔗 GPT-3.5-PHQ-9

🔗 GPT-4o_mini-PHQ-9

🔗 GPT-4o-PHQ-9

🔗 llama3.1_8b-PHQ-9

🔗 mixtral-8x7b-PHQ-9

🔗 GPT-4o_mini-GAD-7

🔗 GPT-4o-GAD-7

🔗 llama3.1_8b-GAD-7

🔗 mixtral-8x7b-GAD-7