背景痛点:传统方案为什么总被用户吐槽“答非所问”
做智能客服的同学都遇到过这种尴尬场景:用户问“我昨天买的空调今天能不能退”,机器人却回复“退货需保持商品完好”。看似相关,其实完全没理解“昨天买”“今天退”的时间诉求。背后元凶就是意图识别(Intent Classification)不准,尤其是长尾意图。
我最早用规则引擎(关键词+正则)做兜底,维护成本爆炸:每上新业务就要加一堆“if-else”,还得处理各种口语化表达。后来换成浅层神经网络——BiLSTM+Attention,F1-score 在头部 30 类能到 0.88,但尾部 200 多类只有 0.54,且多轮对话里一旦用户换说法,上下文就“失忆”。总结下来,传统方案三大短板:
- 长尾意图样本少,模型懒得学
- 缺乏深层语义,同义词/口语化鲁棒性差
- 无法利用大规模预训练知识,泛化靠“堆数据”
技术选型:为什么最终敲定 BERT
在同样 3 万条客服语料上,我横向对比了三种结构:
| 模型 | 头部 F1 | 尾部 F1 | 平均推理延迟(CPU) |
|---|---|---|---|
| TextCNN | 0.85 | 0.51 | 4 ms |
| BiLSTM+Attention | 0.88 | 0.54 | 11 ms |
| BERT-base-chinese | 0.93 | 0.77 | 18 ms |
BERT 尾部 F1 直接提升 20+ 个百分点,而延迟只增加 7 ms,仍在 20 ms 以内的业务容忍度。加上 Transformers 库一行代码就能调,团队上手成本最低,于是拍板。
核心实现:30 行代码搭一个可微调 Intent 分类器
下面代码基于 PyTorch 1.13 + Transformers 4.27,已跑通生产 200 QPS。
1. 数据预处理:Tokenization 最佳实践
from transformers import BertTokenizer from typing import List, Tuple import torch class IntentDataset(torch.utils.data.Dataset): """ 客服意图数据集封装 """ def __init__(self, texts: List[str], labels: List[int], tokenizer: BertTokenizer, max_len: int = 32): self.texts = texts self.labels = labels self.tokenizer = tokenizer self.max_len = max_len def __getitem__(self, idx): # 中文客服场景保留标点,有助于识别反问、疑问语气 text = self.texts[idx].lower() encoded = self.tokenizer( text, add_special_tokens=True, max_length=self.max_len, padding='max_length', truncation=True, return_tensors='pt' ) item = {k: v.squeeze(0) for k, v in encoded.items()} item['labels'] = torch.tensor(self.labels[idx], dtype=torch.long) return item def __len__(self): return len(self.texts)要点:
- 保留标点,尤其“?”“!”对客服情绪意图帮助大
max_len先统计 95% 分位长度,再取 2 的整次幂,减少 padding 浪费
2. 模型结构:给 BERT 接一个“小脑袋”
from transformers import BertModel import torch.nn as nn class BertForIntent(nn.Module): """ 基于 BERT 的意图分类头 """ def __init__(self, bert_dir: str, num_classes: int, dropout: float = 0.3): super().__init__() self.bert = BertModel.from_pretrained(bert_dir) self.drop = nn.Dropout(dropout) self.classifier = nn.Linear(self.bert.config.hidden_size, num_classes) def forward(self, input_ids, attention_mask, token_type_ids=None, labels=None): pooled = self.bert( input_ids=input_ids, attention_mask=attention_mask, token_type_ids=token_type_ids ).pooler_output # [batch, 768] logits = self.classifier(self.drop(pooled)) loss = None if labels is not None: loss_fn = nn.CrossEntropyLoss() loss = loss_fn(logits, labels) return loss, logits训练脚本就是常规 PyTorch Lightning,不再赘述。唯一提醒:客服数据往往类别不平衡,用class_weight='balanced'或 Focal Loss 都能再提 2-3 个点。
生产考量:让 0.77 的尾部 F1 真正跑在线上
1. 量化部署:ONNX Runtime 提速 2.3×
# 导出 ONNX dummy = ( torch.ones(1, 32, dtype=torch.long), torch.ones(1, 32, dtype=torch.long) ) torch.onnx.export( model, dummy, 'intent_bert.onnx', input_names=['input_ids', 'attention_mask'], output_names=['logits'], opset_version=11, dynamic_axes={'input_ids': {0: 'batch'}, 'logits': {0: 'batch'}} )用 ONNX Runtime-GPU 推理,batch=8 时延迟从 18 ms 降到 8 ms,且 F1 无损。
2. OOV 补偿:领域新词自动回退
客服常冒出“以旧换新”“价保”等内部缩写。我把词汇表外(OOV)词做 sub-word 拼接后,再加一层 Embedding 补偿:若 token 仍被<UNK>,用领域词向量字典做替换。实现很简单,在__getitem__里加一段:
for i, id_ in enumerate(encoded['input_ids']): if id_ == tokenizer.unk_token_id: word = tokenizer.decode([id_]) if word in domain_vocab: encoded['input_ids'][i] = domain_vocab[word]线上实测,尾部意图召回率又涨 4%。
避坑指南:踩过的坑提前帮你埋好
Early Stopping 阈值
客服数据头部类别易过拟合,我设patience=3,监控“尾部加权 F1”而非全局准确率,防止模型偷懒只学头部。标点符号处理
中文全角/半角混写会把“?”切成“?”,导致情绪识别失效。统一用unicodedata.normalize('NFKC', text)后再转半角,再进 tokenizer。学习率
BERT 底层用 2e-5,分类头用 1e-3,差一个量级,能加速收敛且不掉点。
延伸思考:知识图谱 + Few-shot,让冷启动不再痛苦
BERT 再强,遇到全新业务线只有 30 条样本也白搭。我的下一步计划:
- 把商品知识图谱(SKU、属性、售后政策)做成节点向量,拼接在
[CLS]后,让模型“带着知识”做意图判断 - 用 Prototypical Networks 做 Few-shot Learning,新意图只需 5 例就能达到 0.8+ F1,配合主动学习,人工标注成本降 70%
写在最后的碎碎念
整套流程从 baseline 0.54 提到 0.77,客服团队实测转人工率下降 30%,老板终于不再天天拉会“优化机器人”。如果你也在为长尾意图头疼,不妨先跑通上面的 30 行代码,再逐步把量化、知识图谱、Few-shot 往里面加。BERT 不是银弹,但用对了,确实能让用户少骂两句“人工智障”。祝各位调参愉快,有问题评论区一起交流。