Few-Shot Learning for Chronic Disease Management: Leveraging Large Language Models and Multi-Prompt Engineering with Medical Knowledge Injection

Haoxin Liu,Wenli Zhang,Jiaheng Xie,Buomsoo Kim,Zhu Zhang,Yidong Chai
2024-01-16
Abstract:This study harnesses state-of-the-art AI technology for chronic disease management, specifically in detecting various mental disorders through user-generated textual content. Existing studies typically rely on fully supervised machine learning, which presents challenges such as the labor-intensive manual process of annotating extensive training data for each disease and the need to design specialized deep learning architectures for each problem. To address such challenges, we propose a novel framework that leverages advanced AI techniques, including large language models and multi-prompt engineering. Specifically, we address two key technical challenges in data-driven chronic disease management: (1) developing personalized prompts to represent each user's uniqueness and (2) incorporating medical knowledge into prompts to provide context for chronic disease detection, instruct learning objectives, and operationalize prediction goals. We evaluate our method using four mental disorders, which are prevalent chronic diseases worldwide, as research cases. On the depression detection task, our method (F1 = 0.975~0.978) significantly outperforms traditional supervised learning paradigms, including feature engineering (F1 = 0.760) and architecture engineering (F1 = 0.756). Meanwhile, our approach demonstrates success in few-shot learning, i.e., requiring only a minimal number of training examples to detect chronic diseases based on user-generated textual content (i.e., only 2, 10, or 100 subjects). Moreover, our method can be generalized to other mental disorder detection tasks, including anorexia, pathological gambling, and self-harm (F1 = 0.919~0.978).
Computation and Language,Artificial Intelligence
What problem does this paper attempt to address?
The paper aims to address several key issues in chronic disease management, particularly the detection of various mental disorders through user-generated text content. Existing studies typically rely on fully supervised machine learning methods, which present some challenges, such as the need for large amounts of labeled data and the design of specialized deep learning architectures for each disease. To address these issues, the paper proposes a new framework that leverages advanced AI technologies, including large-scale language models (LLMs) and multi-prompt engineering, combined with medical knowledge injection to improve the efficiency and accuracy of chronic disease detection. Specifically, the paper addresses the following two key technical challenges: 1. Developing personalized prompts to represent the uniqueness of each user. 2. Integrating medical knowledge into the prompts to provide context for chronic disease detection, guide learning objectives, and achieve prediction goals. Moreover, this method performs excellently in the task of depression detection (F1 score of 0.975~0.978), significantly outperforming traditional supervised learning methods. Additionally, this method can effectively learn with a small number of samples (e.g., 2, 10, or 100 cases) and can be generalized to other mental disorder detection tasks, such as anorexia, pathological gambling, and self-harm behaviors. Overall, this work not only makes contributions at the technical level but also has the potential to improve patient well-being, control healthcare costs, and establish a more efficient and accessible healthcare system.