MHGraphBench: Knowledge Graph-Grounded Benchmarking of Mental Health Knowledge in Large Language Models
Summary
MHGraphBench introduces a knowledge-graph (KG)-grounded benchmark designed to evaluate large language models (LLMs) on their mental health knowledge and application to clinically salient structured judgments. Derived from PrimeKG, this benchmark features nine task families, including entity recognition, relation judgment, and two-hop reasoning, all supported by KG-derived answers and controlled negative options. Experiments across 15 closed- and open-source LLMs revealed a consistent recognition-to-judgment gap; while leading models performed well on entity typing, they struggled significantly with relation prediction and two-hop reasoning. The study also found that short KG-derived snippets had varied effects, sometimes degrading performance, and that output-format reliability critically influenced measured performance in constrained multiple-choice settings. MHGraphBench specifically evaluates agreement with a curated mental-health slice of PrimeKG under a constrained multiple-choice interface, rather than real-world clinical safety.
Key takeaway
For machine learning engineers developing LLMs for mental health applications, you must prioritize robust evaluation beyond basic entity recognition. The MHGraphBench findings highlight a significant recognition-to-judgment gap, indicating that models struggle with complex relation prediction and two-hop reasoning despite strong entity typing. Therefore, focus your development and fine-tuning efforts on improving these advanced reasoning capabilities, and carefully consider how benchmark design and output format reliability influence your model's measured performance.
Key insights
LLMs exhibit a recognition-to-judgment gap in mental health knowledge, struggling with complex reasoning despite strong entity recognition.
Principles
- LLM mental health knowledge is inconsistent.
- Complex reasoning tasks challenge LLMs more.
- Benchmark design impacts measured performance.
Method
MHGraphBench assesses LLMs using a PrimeKG-derived benchmark with nine task families for entity recognition, relation judgment, and two-hop reasoning, employing KG-supported answers and controlled negative options.
In practice
- Evaluate LLMs on mental health entity recognition.
- Test LLM relation prediction capabilities.
- Assess two-hop reasoning in clinical contexts.
Topics
- Large Language Models
- Mental Health
- Knowledge Graphs
- LLM Benchmarking
- Clinical Reasoning
- PrimeKG
Best for: AI Scientist, Machine Learning Engineer, Research Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.