中文

基于经过优化的Transformer进行线程检测与响应生成

计算与语言 2024-03-12 v1 机器学习

摘要

对话系统在人机交互中发挥着关键作用,通过识别线程并优先确定响应来管理复杂的对话。这在多方对话中尤为重要,因为精确识别线程和战略性地优先确定响应有助于确保高效的对话管理。为应对这些挑战,开发了一种端到端模型,用于识别线程并根据其重要性对其响应生成进行优先排序,这涉及将问题系统地分解为离散组件——线程检测、优先排序和性能优化——并对这些细化组件进行精心分析和优化。这些精炼后的组件无缝集成到对话系统中的统一框架中。由于其高度的通用性,Llama2 7b被用于该系统,但该系统可以与任何开源大型语言模型(LLM)更新集成。通过采用微调方法和战略性提示技术来增强Llama2模型的计算能力,以优化模型的性能,减少计算时间并提高模型的准确性。该模型实现了最高可达10倍的速度提升,同时生成的結果比现有模型更具连贯性。

关键词

引用

@article{arxiv.2403.05931,
  title  = {Thread Detection and Response Generation using Transformers with Prompt Optimisation},
  author = {Kevin Joshua T and Arnav Agarwal and Shriya Sanjay and Yash Sarda and John Sahaya Rani Alex and Saurav Gupta and Sushant Kumar and Vishwanath Kamath},
  journal= {arXiv preprint arXiv:2403.05931},
  year   = {2024}
}

备注

6 pages, 4 figures, submitted to 2024 IEEE International Conference on Signal Processing and Communications (SPCOM)