首页 > 技术文章 > CTC Algorithm Explained Part 1:Training the Network(CTC 算法详解之训练篇)

dahu-daqing 2020-08-18 17:08 原文

转载本文请注明出处:https://xiaodu.io/ctc-explained 作者:yudonglee

现实应用中许多问题可以抽象为序列学习(sequence learning)问题,比如词性标注(POS Tagging)、语音识别(Speech Recognition)、手写字识别(Handwriting Recognition)、机器翻译(Machine Translation)等应用,其核心问题都是训练模型把一个领域的(输入)序列转成另一个领域的(输出)序列。近年来基于 RNN 的序列到序列模型(sequence-to-sequence models)在序列学习任务中取得了显著的效果提升,本文介绍一种 RNN(Recurrent Neural Networks)的端到端训练方法 ——CTC(Connectionist Temporal Classification)算法,它可以让 RNN 直接对序列数据进行学习,而无需事先标注好训练数据中输入序列和输出序列的映射关系,打破了 RNN 应用于语音识别、手写字识别等领域的数据依赖约束,使得 RNN 模型在序列学习任务中取得更好的应用效果。

本文总共分为五部分来全面阐述 CTC 算法(本篇为 Part 1):
Part 1:Training the Network(训练算法篇),介绍 CTC 理论原理,包括问题定义、公式推导、算法过程等。Part 1 链接
Part 2:Decoding the Network(解码算法篇),介绍 CTC Decoding 的几种常用算法。Part 2 链接
Part 3:CTC Demo by Speech Recognition(CTC 语音识别实战篇),基于 TensorFlow 实现的语音识别代码,包含详细的代码实战讲解。Part 3 链接。
Part 4:CTC Demo by Handwriting Recognition(CTC 手写字识别实战篇),基于 TensorFlow 实现的手写字识别代码,包含详细的代码实战讲解。Part 4 链接。
Part 5:Conclusion(总结展望篇),总结 CTC 算法的理论局限性和适用场景,以及近年来相关的最新研究动态。Part 5 链接。

接下来,我们先从 “问题” 的背景说起。

1. 背景介绍

在序列学习任务中,RNN 模型对训练样本一般有这样的依赖条件:输入序列和输出序列之间的映射关系已经事先标注好了。比如,在词性标注任务中,训练样本中每个词(或短语)对应的词性会事先标注好,如下图(DT、NN 等都是词性的标注,具体含义请参考链接)。由于输入序列和输出序列是一一对应的,所以 RNN 模型的训练和预测都是端到端的,即可以根据输出序列和标注样本间的差异来直接定义 RNN 模型的 Loss 函数,传统的 RNN 训练和预测方式可直接适用。

然而,在语音识别、手写字识别等任务中,由于音频数据和图像数据都是从现实世界中将模拟信号转为数字信号采集得到,这些数据天然就很难进行 “分割”,这使得我们很难获取到包含输入序列和输出序列映射关系的大规模训练样本(人工标注成本巨高,且启发式挖掘方法存在很大局限性)。因此,在这种条件下,RNN 无法直接进行端到端的训练和预测。

如下图,输入是 “apple” 对应的一段说话音频和手写字图片,从连续的音频信号和图像信号中逐一分割并标注出对应的输出序列非常费时费力,在大规模训练下这种数据要求是完全不切实际的。而如果输入序列和输出序列之间映射关系没有提前标注好,那传统的 RNN 训练方式就不能直接适用了,无法直接对音频数据和图像数据进行训练。

因此,在语音识别、图像识别等领域中,由于数据天然无法切割,且难以标注出输入和输出的序列映射关系,导致传统的 RNN 训练方法不能直接适用。那么,如何让 RNN 模型实现端到端的训练成为了关键问题。

Connectionist Temporal Classification(CTC)[1] 是 Alex Graves 等人在 ICML 2006 上提出的一种端到端的 RNN 训练方法,它可以让 RNN 直接对序列数据进行学习,而无需事先标注好训练数据中输入序列和输入序列的映射关系,使得 RNN 模型在语音识别等序列学习任务中取得更好的效果,在语音识别和图像识别等领域 CTC 算法都有很比较广泛的应用。总的来说,CTC 的核心思路主要分为以下几部分:

  • 它扩展了 RNN 的输出层,在输出序列和最终标签之间增加了多对一的空间映射,并在此基础上定义了 CTC Loss 函数
  • 它借鉴了 HMM(Hidden Markov Model)的 Forward-Backward 算法思路,利用动态规划算法有效地计算 CTC Loss 函数及其导数,从而解决了 RNN 端到端训练的问题
  • 最后,结合 CTC Decoding 算法 RNN 可以有效地对序列数据进行端到端的预测

接下来,通过一个语音识别的实际例子来引出 CTC 的解决思路

2. 一个实际的例子–声学模型

语音识别的核心问题是把一段音频信号序列转化文字序列,传统的语音识别系统主要分为以下几部分,如下图。


其中,X 表示音频信号,O 是它的特征表示,一般基于 LPC、MFCC 等方法提取特征,也可以基于 DNN 的方式 “学到” 声学特征的表示。为了简化问题,我们暂且把 O 理解为是由实数数组组成的序列,它是音频信号的特征表示。Q 是 O 对应的发音字符序列,即建模单元,一般可以是音素、音节、字、词等。W 是音频信号 X 对应的文字序列,即我们最终的识别结果。

如图所示,核心问题是通过解码器找到令 P (W|X) 最大化的的 W,通过贝叶斯公式可将其分解为 P (O|Q)、P (Q|W)、P (W),分别对应声学模型、发音模型、语言模型。

其中,声学模型就是对 P (O|Q) 进行建模,通过训练可以 “学到” 音频信号和文字发音间的联系。为了简化问题,我们假定声学模型的建模单元 Q 选择的是音节,O 选择的是 MFCC 特征(由 39 维数组组成的序列)。

如下图,输入序列是一段 “我爱你中国” 的音频,输出序列是音节序列 “wo3 ai4 ni3 zhong1 guo2”,如果训练样本中已经 “分割” 好音频,并标注好它和音节的对应关系,则 RNN 模型如下:


然而,如前面所述,对音频 “分割” 并标注映射关系的数据依赖是不切实际的,实际情况是对音频按照时间窗口滑动来提取特征,比如按照每 10 毫秒音频提取特征得到一个 N 维数组。如下图所示:


由于人说话发音是连续的,且中间也会有 “停顿”,所以输出序列中存在重复的元素,比如 “wo3 wo3”,也存在表示间隔符号 “_”。需从输出序列中去除掉重复的元素以及间隔符,才可得到最终的音节序列,比如,“wo3 wo3 ai4 _ ni3 _ zhong1 guo2 _” 归一处理后得到 “wo3 ai4 ni3 zhong1 guo2”。因此,输出序列和最终的 label 之间存在多对一的映射关系,如下图:


RNN 模型本质是对

推荐阅读