背景:规则引擎的“天花板”
做客服的同学都懂,早期用正则+关键词的“规则引擎”上线快,但三个月后就陷入“打补丁地狱”:
- 用户问“我昨天买的那个东西能不能退”,规则里没写“那个东西”,就匹配不到退货意图
- 多轮对话要来回确认订单号、手机号,状态写在内存里,服务一重启,用户就得从头再来
- 长尾问题占比 18%,却吃掉 60% 人力维护,老板一句话:“加条规则”,开发就得连夜上线
痛点总结:覆盖率瓶颈、状态易失、维护成本高。于是我们把目光投向了数据驱动的“模型+状态机”方案。
技术选型:Rasa、DialogFlow 还是自研?
先做一轮“小范围 benchmark”,用同一批 2.3 万条真实语料,在 4C8G 容器里跑:
| 方案 | 意图准确率 | 平均 QPS | 年度 License 成本 | 备注 |
|---|---|---|---|---|
| DialogFlow ES | 87.2 % | 180 | 3.2 万 USD | 黑盒,不可微调 |
| Rasa 3.x | 90.1 % | 220 | 0 | 组件多,学习曲线陡 |
| 自研 BERT+Transformer | 92.6 % | 310 | 服务器折旧 | 可控,需算法团队 |
结论:对并发和领域定制要求高的场景,自研最划算;Rasa 适合 POC;DialogFlow 快速上线但容易被“锁喉”。
核心实现一:BERT+BiLSTM 意图识别
模型结构很简单:BERT 取 [CLS] 向量,接 128 维 BiLSTM,再 Softmax。PyTorch 代码如下(PEP8,带类型注解)
import torch import torch.nn as nn from transformers import BertModel from typing import List, Dict class IntentClassifier(nn.Module): def __init__(self, bert_dir: str, hidden_size: int = 128, num_intents: int = 64): super().__init__() self.bert = BertModel.from_pretrained(bert_dir) self.lstm = nn.LSTM( input_size=self.bert.config.hidden_size, hidden_size=hidden_size, num_layers=1, batch_first=True, bidirectional=True ) self.fc = nn.Linear(hidden_size * 2, num_intents) def forward(self, input_ids: torch.Tensor, attention_mask: torch.Tensor) -> torch.Tensor: bert_out = self.bert(input_ids, attention_mask)[0] # [B, L, H] lstm_out, _ = self.lstm(bert_out) # [B, L, 2*H] # 取最后一个时间步 logits = self.fc(lstm_out[:, -1, :]) return logits训练 30 epoch,AdamW lr=2e-5,早停 patience=3,最终在验证集达到 92.6% F1。推理时把动态图剪掉,转 TorchScript,延迟从 280 ms 降到 90 ms。
核心实现二:Redis 驱动的分布式对话状态机
多轮对话最怕“状态漂移”。我们把每轮事件抽象成三元组:(user_id, slot_key, slot_value),用 Redis Hash 存储,并加上 15 min TTL。
分布式锁采用 Redlock,防止横向扩容时两个 Pod 同时修改状态:
import redis import uuid from typing import Optional class DialogueState: def __init__(self, redis_url: str): self.rdb = redis.from_url(redis_url, decode_responses=True) def lock(self, user_id: str, timeout: int = 5) -> Optional[str]: ticket = str(uuid.uuid4()) ok = self.rdb.set( f"lock:{user_id}", ticket, nx=True, ex=timeout ) return ticket if ok else None def save_slots(self, user_id: str, ticket: str, slots: Dict[str, str]) -> bool: if self.rdb.get(f"lock:{user_id}") != ticket: return False key = f"ds:{user_id}" self.rdb.hset(key, mapping=slots) self.rdb.expire(key, 900) self.rdb.delete(f"lock:{user_id}") return True压测 1 k 并发,锁冲突率 < 0.3%,对体验无感知。
性能优化:GPU 量化与负载均衡
1. INT8 量化
用 NVIDIA TensorRT 把上述 TorchScript 再量化,INT8 校准 500 条语料,模型从 418 MB → 108 MB,T4 GPU 单卡 QPS 由 310 提到 720,精度掉 0.4%,可接受。
2. 负载均衡策略对比
| 策略 | 平均 TPS | P99 延迟 | 容灾表现 |
|---|---|---|---|
| Round-Robin | 2100 | 340 ms | 节点故障 30 s 后恢复 |
| Least-RTT | 2350 | 260 ms | 同上 |
| 基于会话粘滞 | 2180 | 290 ms | 用户无感知,但热点明显 |
最终线上采用“Least-RTT + 会话粘滞 fallback”,保证性能与体验平衡。
避坑指南:日志脱敏与 AB 测试
1. 日志脱敏
客服日志里身份证号、手机号、订单号遍地走,一旦泄露就“喜提热搜”。我们的脱敏流程:
- 在 Annotator 阶段先跑一个 NER,把敏感实体替换成
<PHONE>、<ID>等占位符 - 写日志用占位后文本,原始文本进加密 ES,权限白名单管控
- 定期跑敏感数据扫描脚本,发现明文立即告警
2. AB 测试框架
模型迭代频繁,线上要跑 AB。我们把流量按 user_id 哈希分桶,保证同一用户始终命中同一模型版本;指标看“意图准确率 + 任务完成率 + 平均轮数”三件套。灰度 5% → 20% → 100%,七天无异常才全量。回滚策略:切换模型版本号,30 秒内置热完成。
上线效果与复盘
- 意图准确率从 87% → 92.6%,人工转接率下降 5.2 %
- 大促峰值 4 万 QPS,Kubernetes HPA 根据 GPU 利用率 70% 弹性扩容,单卡成本降低 38 %
- 规则维护人力由 6 人月缩减到 1.5 人月,算法同学终于不用每天“加正则”了
开放讨论
如何平衡预训练模型精度与推理延迟?我们在量化、剪枝、蒸馏之间反复横跳,仍没银弹。你的场景会倾向大模型+高算力,还是小模型+CPU 集群?欢迎留言碰撞。