BERT-语义对话预测-中文-说话人日志
  • 模型资讯
  • 模型资料

Dialogue Detection 模型

Dialogue Detection任务用于判读一段文本是否为一段对话。此信息将帮助到speaker diarization模型。

模型说明

我们的模型基于BERT训练得到,核心为一个二分类的Sentence Classification任务。

关于此模型的细节以及如何后续应用,请参考我们的论文:

数据集

我们的模型基于如下的数据集训练:

我们使用一个滑动窗策略来构造训练和测试数据。

模型效果评估

测试集 Precision Recall F1 Acc
AISHELL-4 Test 0.975 0.9444 0.9595 0.9304
ALIMEETING Test 0.9652 0.7686 0.8558 0.8463

在线体验

在Notebook中开发

待开发

使用Modelscope本地推理

待开发

相关论文以及引用信息

如果您觉得这个该模型有所帮助,请引用下面的相关的论文

@article{Cheng2023ExploringSI,
  title={Exploring Speaker-Related Information in Spoken Language Understanding for Better Speaker Diarization},
  author={Luyao Cheng and Siqi Zheng and Zhang Qinglin and Haibo Wang and Yafeng Chen and Qian Chen},
  journal={ArXiv},
  year={2023},
  volume={abs/2305.12927}
}