update README
Browse files
README.md
CHANGED
@@ -15,10 +15,6 @@ language:
|
|
15 |
|
16 |
# 灵心健康大模型SoulChat:通过长文本咨询指令与多轮共情对话数据集的混合微调,提升大模型的“共情”能力
|
17 |
|
18 |
-
<p align="center">
|
19 |
-
<img src="https://github.com/scutcyr/SoulChat/ProactiveHealthGPT.png" width=900px/>
|
20 |
-
</p>
|
21 |
-
<p align="center">
|
22 |
<a href="./LICENSE"><img src="https://img.shields.io/badge/license-Apache%202-red.svg"></a>
|
23 |
<a href="support os"><img src="https://img.shields.io/badge/os-linux%2C%20win%2C%20mac-pink.svg"></a>
|
24 |
<a href=""><img src="https://img.shields.io/badge/python-3.8+-aff.svg"></a>
|
@@ -43,10 +39,7 @@ language:
|
|
43 |
|
44 |
|
45 |
## 简介
|
46 |
-
我们调研了当前常见的心理咨询平台,发现,用户寻求在线心理帮助时,通常需要进行较长篇幅地进行自我描述,然后提供帮助的心理咨询师同样地提供长篇幅的回复(见[https://github.com/scutcyr/SoulChat/figure/single_turn.png](https://github.com/scutcyr/SoulChat/figure/single_turn.png)
|
47 |
-
<p align="center">
|
48 |
-
<img src="https://github.com/scutcyr/SoulChat/figure/multi_turn.png" width=900px/>
|
49 |
-
</p>
|
50 |
|
51 |
考虑到当前十分欠缺多轮共情对话数据集,我们一方面,构建了超过15万规模的 **单轮长文本心理咨询指令与答案(SoulChatCorpus-single_turn)** ,回答数量超过50万(指令数是当前的常见的心理咨询数据集 [PsyQA](https://github.com/thu-coai/PsyQA) 的6.7倍),并利用ChatGPT与GPT4,生成总共约100万轮次的 **多轮回答数据(SoulChatCorpus-multi_turn)** 。特别地,我们在预实验中发现,纯单轮长本文驱动的心理咨询模型会产生让用户感到厌烦的文本长度,而且不具备引导用户倾诉的能力,纯多轮心理咨询对话数据驱动的心理咨询模型则弱化了模型的建议能力,因此,我们混合SoulChatCorpus-single_turn和SoulChatCorpus-multi_turn构造成超过120万个样本的 **单轮与多轮混合的共情对话数据集SoulChatCorpus** 。所有数据采用“用户:xxx\n心理咨询师:xxx\n用户:xxx\n心理咨询师:”的形式统一为一种指令格式。
|
52 |
|
@@ -99,9 +92,6 @@ response, history = model.chat(tokenizer, query=input_text, history=None, max_le
|
|
99 |
```
|
100 |
|
101 |
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
* 启动服务
|
106 |
本项目提供了[soulchat_app.py](./soulchat_app.py)作为SoulChat模型的使用示例,通过以下命令即可开启服务,然后,通过http://<your_ip>:9026访问。
|
107 |
```bash
|
@@ -118,30 +108,6 @@ model_name_or_path = 'scutcyr/SoulChat'
|
|
118 |
```
|
119 |
|
120 |
|
121 |
-
## 示例
|
122 |
-
* 样例1:失恋
|
123 |
-
*
|
124 |
-
<p align="center">
|
125 |
-
<img src="https://github.com/scutcyr/SoulChat/figure/example_shilian.png" width=600px/>
|
126 |
-
</p>
|
127 |
-
|
128 |
-
* 样例2:宿舍关系
|
129 |
-
|
130 |
-
<p align="center">
|
131 |
-
<img src="https://github.com/scutcyr/SoulChat/figure/example_sushe.png" width=600px/>
|
132 |
-
</p>
|
133 |
-
|
134 |
-
* 样例3:期末考试
|
135 |
-
|
136 |
-
<p align="center">
|
137 |
-
<img src="https://github.com/scutcyr/SoulChat/figure/example_kaoshi.png" width=600px/>
|
138 |
-
</p>
|
139 |
-
|
140 |
-
* 样例4:科研压力
|
141 |
-
|
142 |
-
<p align="center">
|
143 |
-
<img src="https://github.com/scutcyr/SoulChat/figure/example_keyan.png" width=600px/>
|
144 |
-
</p>
|
145 |
|
146 |
## 声明
|
147 |
* 本项目使用了ChatGLM-6B 模型的权重,需要遵循其[MODEL_LICENSE](https://github.com/THUDM/ChatGLM-6B/blob/main/MODEL_LICENSE),因此,**本项目仅可用于您的非商业研究目的**。
|
|
|
15 |
|
16 |
# 灵心健康大模型SoulChat:通过长文本咨询指令与多轮共情对话数据集的混合微调,提升大模型的“共情”能力
|
17 |
|
|
|
|
|
|
|
|
|
18 |
<a href="./LICENSE"><img src="https://img.shields.io/badge/license-Apache%202-red.svg"></a>
|
19 |
<a href="support os"><img src="https://img.shields.io/badge/os-linux%2C%20win%2C%20mac-pink.svg"></a>
|
20 |
<a href=""><img src="https://img.shields.io/badge/python-3.8+-aff.svg"></a>
|
|
|
39 |
|
40 |
|
41 |
## 简介
|
42 |
+
我们调研了当前常见的心理咨询平台,发现,用户寻求在线心理帮助时,通常需要进行较长篇幅地进行自我描述,然后提供帮助的心理咨询师同样地提供长篇幅的回复(见[https://github.com/scutcyr/SoulChat/figure/single_turn.png](https://github.com/scutcyr/SoulChat/figure/single_turn.png)),缺失了一个渐进式的倾诉过程。但是,在实际的心理咨询过程当中,用户和心理咨询师之间会存在多轮次的沟通过程,在该过程当中,心理咨询师会引导用户进行倾诉,并且提供共情,例如:“非常棒”、“我理解你的感受”、“当然可以”等等。
|
|
|
|
|
|
|
43 |
|
44 |
考虑到当前十分欠缺多轮共情对话数据集,我们一方面,构建了超过15万规模的 **单轮长文本心理咨询指令与答案(SoulChatCorpus-single_turn)** ,回答数量超过50万(指令数是当前的常见的心理咨询数据集 [PsyQA](https://github.com/thu-coai/PsyQA) 的6.7倍),并利用ChatGPT与GPT4,生成总共约100万轮次的 **多轮回答数据(SoulChatCorpus-multi_turn)** 。特别地,我们在预实验中发现,纯单轮长本文驱动的心理咨询模型会产生让用户感到厌烦的文本长度,而且不具备引导用户倾诉的能力,纯多轮心理咨询对话数据驱动的心理咨询模型则弱化了模型的建议能力,因此,我们混合SoulChatCorpus-single_turn和SoulChatCorpus-multi_turn构造成超过120万个样本的 **单轮与多轮混合的共情对话数据集SoulChatCorpus** 。所有数据采用“用户:xxx\n心理咨询师:xxx\n用户:xxx\n心理咨询师:”的形式统一为一种指令格式。
|
45 |
|
|
|
92 |
```
|
93 |
|
94 |
|
|
|
|
|
|
|
95 |
* 启动服务
|
96 |
本项目提供了[soulchat_app.py](./soulchat_app.py)作为SoulChat模型的使用示例,通过以下命令即可开启服务,然后,通过http://<your_ip>:9026访问。
|
97 |
```bash
|
|
|
108 |
```
|
109 |
|
110 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
111 |
|
112 |
## 声明
|
113 |
* 本项目使用了ChatGLM-6B 模型的权重,需要遵循其[MODEL_LICENSE](https://github.com/THUDM/ChatGLM-6B/blob/main/MODEL_LICENSE),因此,**本项目仅可用于您的非商业研究目的**。
|