A Conversational Application for Insomnia Treatment: Leveraging the ChatGLM-LoRA Model for Cognitive Behavioral Therapy
The aim of this study was to develop a mobile application for psychotherapy with insomnia patients using the ChatGLM-LoRA model, fine-tuned by Low-Rank Adaptation, and validated in a clinical trial.The dataset used to train the model was a collection of 764 dialogues related to sleep disorders. The corpus was randomly divided into three subsets: training, validation, and test sets. The hyperparameters used in this study to train the model were 450 epochs, betas ranging from 0.9 to 0.95, weight decay rate 5e-4, maximum learning rate 1e-5, and AdamW optimizer. Based on the test results of the above hyperparameters, the four metrics of BLEU-4, ROUGE-1, ROUGE-2, and ROUGE-L of the model reached 0.0340, 0.0451, and 0.0163; 0.2773, 0.3075, and 0.1986; 0.0592, 0.0735, and 0.0261; 0.2112, 0.2336, and 0.1500 for the training, validation, and test sets.These results indicate the technical feasibility and potential clinical utility of using an advanced language model-based application for psychotherapeutic intervention in insomnia.
Authors
- Qinghua Xia (ORCID: https://orcid.org/0000-0002-8261-6014)
- Luying Zhang (ORCID: https://orcid.org/0000-0001-8427-7063)
- Huanhuan Li (ORCID: https://orcid.org/0009-0006-6443-132X)
- Jianwei Shuai (ORCID: https://orcid.org/0000-0002-8712-0544)
- Hua Zhang (ORCID: https://orcid.org/0000-0002-4700-5971)
- Rongwen Yu (ORCID: https://orcid.org/0009-0008-1250-5280)
- Shuo Pan (ORCID: https://orcid.org/0000-0001-7934-572X)
- Jiahe Chen (ORCID: https://orcid.org/0000-0001-7584-7861)
- Yu Xia (ORCID: https://orcid.org/0000-0002-3111-8701)
- Yinda Chen
- Keqi Ren
- Zefei Mo
Institutions
- Ningbo University (CN)
- Wenzhou Medical University (CN)
- Affiliated Eye Hospital of Wenzhou Medical College (CN)
- Quzhou University (CN)
- University of Chinese Academy of Sciences (CN)
- Zhejiang University (CN)
Publication Details
- Published
- 2024-08-08
- DOI
- https://doi.org/10.1109/cis-ram61939.2024.10673360
- Citations
- 3
- Primary Topic
- Digital Mental Health Interventions
- Type
- article
- Field-Weighted Citation Impact
- 1.23
Funders
- National Natural Science Foundation of China
- Ministry of Science and Technology