arXiv:2505.00032cs.CLcs.AI2025-05被引 12

用大模型分析真实人群数据,提升抑郁症诊断准确率

MDD-LLM: Towards Accuracy Large Language Models for Major Depressive Disorder Diagnosis

  • 基于英国生物银行27万条数据,用表格转文本训练大模型
  • 70B版本准确率达83.78%,AUC达89.19%,显著优于现有方法
  • 适合精神健康研究者和医疗AI开发者参考

重度抑郁症(MDD)影响全球超3亿人,但医疗资源分布不均与诊断复杂性导致许多地区重视不足。本文提出MDD-LLM,一个基于微调大语言模型(LLM)的AI诊断框架,利用英国生物银行队列中的274,348条个体记录进行训练与评估。通过设计表格数据转换方法构建大规模语料库,实现对真实世界数据的有效建模。实验表明,MDD-LLM(70B)在多项指标上表现优异,准确率为0.8378,AUC为0.8919(95%置信区间:0.8799–0.9040),显著超越现有机器学习与深度学习模型。同时,研究探索了表格转换方式、微调策略等因素对性能的影响,为大模型在抑郁症诊断中的应用提供新思路。

原文摘要 · Abstract (English)

Major depressive disorder (MDD) impacts more than 300 million people worldwide, highlighting a significant public health issue. However, the uneven distribution of medical resources and the complexity of diagnostic methods have resulted in inadequate attention to this disorder in numerous countries and regions. This paper introduces a high-performance MDD diagnosis tool named MDD-LLM, an AI-driven framework that utilizes fine-tuned large language models (LLMs) and extensive real-world samples to tackle challenges in MDD diagnosis. Therefore, we select 274,348 individual information from the UK Biobank cohort to train and evaluate the proposed method. Specifically, we select 274,348 individual records from the UK Biobank cohort and design a tabular data transformation method to create a large corpus for training and evaluating the proposed approach. To illustrate the advantages of MDD-LLM, we perform comprehensive experiments and provide several comparative analyses against existing model-based solutions across multiple evaluation metrics. Experimental results show that MDD-LLM (70B) achieves an accuracy of 0.8378 and an AUC of 0.8919 (95% CI: 0.8799 - 0.9040), significantly outperforming existing machine learning and deep learning frameworks for MDD diagnosis. Given the limited exploration of LLMs in MDD diagnosis, we examine numerous factors that may influence the performance of our proposed method, such as tabular data transformation techniques and different fine-tuning strategies.

抑郁症诊断大语言模型医疗AIUK Biobank

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。