arXivDaily arXiv每日学术速递 周一至周五更新
稍安勿躁,官方尚未更新
arXiv 2401.06321cs.CL
收藏

面向TTS前端文本处理的多任务学习

Multi-Task Learning for Front-End Text Processing in TTS

Massachusetts Institute of Technology(麻省理工学院);AI at Meta(Meta人工智能部门)

Wonjune Kang, Yun Wang, Shun Zhang, Arthur Hinsvark, Qing He

AI总结:

本文提出一种树状多任务学习框架,联合处理TTS前端的文本规范化、词性标注和同形异义词消歧,并引入预训练语言模型及新的均衡HD数据集,显著提升整体性能。

摘要:

我们提出了一种多任务学习(MTL)模型,用于联合执行文本到语音(TTS)前端中通常解决的三个任务:文本规范化(TN)、词性(POS)标注和同形异义词消歧(HD)。我们的框架采用树状结构,主干学习共享表示,之后连接各自独立的任务专用头。我们进一步引入预训练语言模型,以利用其内置的词汇和上下文知识,并研究如何最好地使用其嵌入,从而最有效地提升我们的多任务模型。通过逐任务消融实验,我们表明,与仅在单个任务或部分任务组合上训练的模型相比,在全部三个任务上训练的完整模型取得了最强的整体性能,证实了我们MTL框架的优势。最后,我们引入了一个新的HD数据集,其中包含针对多种同形异义词及其发音、在多样上下文中的均衡数量的句子。我们证明,与仅使用一个常用但不平衡的已有数据集相比,将该数据集纳入训练可显著提升HD性能。

英文摘要:

We propose a multi-task learning (MTL) model for jointly performing three tasks that are commonly solved in a text-to-speech (TTS) front-end: text normalization (TN), part-of-speech (POS) tagging, and homograph disambiguation (HD). Our framework utilizes a tree-like structure with a trunk that learns shared representations, followed by separate task-specific heads. We further incorporate a pre-trained language model to utilize its built-in lexical and contextual knowledge, and study how to best use its embeddings so as to most effectively benefit our multi-task model. Through task-wise ablations, we show that our full model trained on all three tasks achieves the strongest overall performance compared to models trained on individual or sub-combinations of tasks, confirming the advantages of our MTL framework. Finally, we introduce a new HD dataset containing a balanced number of sentences in diverse contexts for a variety of homographs and their pronunciations. We demonstrate that incorporating this dataset into training significantly improves HD performance over only using a commonly used, but imbalanced, pre-existing dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑