FreezeEmpath: Efficient Training for Empathetic Spoken Chatbots with Frozen LLMs
📄 FreezeEmpath: Efficient Training for Empathetic Spoken Chatbots with Frozen LLMs #语音对话系统 #多模态模型 #迁移学习 #语音情感识别 🔥 评分:10.0/10 | arxiv 👥 作者与机构 第一作者:Yun Hong(中国科学院计算技术研究所,智能信息处理国家重点实验室;中国科学院大学) 通讯作者:Yang Feng(中国科学院计算技术研究所,智能信息处理国家重点实验室;中国科学院大学) 其他作者:Yan Zhou(中国科学院计算技术研究所,智能信息处理国家重点实验室;中国科学院大学) 机构详情:所有作者均隶属于中国科学院计算技术研究所的“智能信息处理国家重点实验室”和“人工智能安全国家重点实验室”,以及中国科学院大学。 💡 毒舌点评 亮点是“冻结LLM”这个思路简直是懒人智慧的巅峰——让模型自己懂共情,我们只负责搭个桥,数据和训练成本直接砍半。槽点嘛,虽然生成的语音情感挺到位,但毕竟用的是现成的TTS模块,情感表达的上限可能被预训练模型锁死了,想让它“影帝级”爆发估计有点难。 📌 核心摘要 本文旨在解决训练共情语音聊天机器人时面临的共情语音数据稀缺、模型泛化能力弱、以及微调导致LLM通用能力退化三大难题。作者提出了FreezeEmpath,一种高效的端到端训练框架。其核心方法是冻结基础LLM,采用语义-情感解耦编码策略,通过独立的语义适配器和情感提取器从语音中分别提取内容和情感特征,并设计三阶段训练(语义对齐、情感对齐、语音生成)将这些特征与LLM的嵌入空间对齐,从而将LLM内在的文本共情能力迁移到语音模态。整个训练仅需现有的中性语音指令数据和语音情感识别数据,无需人工构建的共情语音数据。实验表明,FreezeEmpath在共情对话、语音情感识别和口语问答任务上均显著优于现有SOTA模型,证明了其方法的有效性和高效性。 🏗️ 模型架构 FreezeEmpath的整体架构由三部分组成:语音理解模块、基础LLM和语音生成模块。 完整输入输出流程: 输入:用户的语音指令。 语音理解模块: 语音编码器(Whisper-large-v3):将原始语音波形编码为隐藏状态序列 X ∈ ℝ^{L×T×D}(L=层数,T=序列长度,D=维度)。 语义适配器:由一个下采样层和一个2层前馈网络(FFN)组成,将编码器输出映射到LLM的嵌入空间,得到语义特征序列 S。 情感提取器:包含两步池化。 层级池化:使用一个门控网络 g 对编码器所有层的隐藏状态进行加权平均,得到压缩后的特征 X̂。 帧级池化:使用一个可学习的查询 Q,通过多头交叉注意力机制(MHA,4个头)聚合 X̂ 的时间维度信息,再通过一个2层FFN(隐藏维度2048)映射为LLM嵌入空间的情感特征向量 E。 序列拼接:将语义特征序列 S、固定连接词嵌入 F1、情感特征向量 E、固定连接词嵌入 F2 拼接,形成最终输入序列 X_S = [S, F1, E, F2] 送入LLM。 基础LLM(Qwen2.5-7B-Instruct):全程参数冻结。接收 X_S,基于其内在的语义理解和共情能力,生成文本响应 r 的隐藏状态序列。 语音生成模块: 流式语音解码器:一个解码器Transformer(初始化自Qwen2.5-0.5B)。它包含一个门控融合模块,聚合来自LLM的上下文隐藏状态和已解码文本 token 的精确语义信息。解码器以流式方式自回归生成语音 token 序列 u(词汇表大小8192,频率50Hz)。流式参数为:每读入 R=3 个输入嵌入,生成 W=15 个语音 token。 Token2Wav模块:使用预训练的IndexTTS2的流匹配模型和声码器,将语音 token 序列 u 转换为最终的共情语音响应。 关键设计理由: ...