赞
踩
我们的感知模式、语言和艺术作品都具有统计结构。学习这种结构是深度学习算法所擅长的。机器学习模型能够对图像、音乐和故事的统计潜在空间 ( l a t e n t s p a c e ) (latent space) (latentspace)进行学习,然后从这个空间中采样,创造出于模型在训练数据中所见到的艺术作品具有相似特征的新作品。当然,这种采样本身并不是艺术创作行为。它只是一种数学运算,算法并没有关于人类生活、人类情感或我们人生经验的基础知识。相反,它从一种与我们的经验完全不同的经验中进行学习。作为人类旁观者,只能靠我们的解释才能对模型生成的内容赋予意义。但在技艺高超的艺术家手中,算法生成可以变得非常有意义,并且很美。潜在空间采样会变成一支画笔,能够提高艺术家的能力,增强我们的创造力,并拓展我们的想象空间。——《Python 深度学习》
作曲家从繁琐的计算中解脱出来,从而能够全神贯注于解决新音乐形式所带来的一般性问题,并在修改输入数据值的同时探索这种形式的鲜为人知之处。例如,它可以测试所有的乐器组合,从独奏到室内管弦乐队再到大型管弦乐队。在电子计算机的帮助下,作曲家变成了一名飞行员:他按下按钮,引入坐标,并监控宇宙飞船在“声音空间”中的航行,飞船穿越声波的星座和星系,这是以前只能在遥不可及的梦中出现的场景。 —— Iannis Xenakis
基于马尔科夫的语言模型在数据驱动下的自然语言生成。利用数据和文字间的对齐语料,使用 N-gram 语言模型生成文本。在语言模型上加入句法分析
(
关
系
抽
取
、
实
体
识
别
、
依
存
句
法
、
短
语
结
构
等
)
(关系抽取、实体识别、依存句法、短语结构等)
(关系抽取、实体识别、依存句法、短语结构等)可以显著改善文本生成效果。因为这些都是建立在句子理解的基础上,文本生成的过程需要考虑历史信息,处理长距离的依赖关系情况,如语义连贯性。
标准定义: 对于语言序列
w
1
,
w
2
,
.
.
.
,
w
n
w_1, w_2, ..., w_n
w1,w2,...,wn ,语言模型就是计算该序列出现的概率,即:
P
(
w
1
,
w
2
,
.
.
.
,
w
n
)
P(w_1, w_2,...,w_n)
P(w1,w2,...,wn) 也即语言模型描述了这些单词组成这个语言序列句子的概率分布。
首先,由链式法则 chain rule 可以得到:
P ( w 1 , w 2 , . . . , w n ) = P ( w 1 ) P ( w 2 ∣ w 1 ) ⋅ ⋅ ⋅ P ( w n ∣ w 1 , . . . , w n − 1 ) P(w_1, w_2,...,w_n) = P(w_1)P(w_2|w_1)···P(w_n|w_1,...,w_{n-1}) P(w1,w2,...,wn)=P(w1)P(w2∣w1)⋅⋅⋅P(wn∣w1,...,wn−1)
马尔科夫假设,当前词只依赖前面 n-1 个词。
P ( w i ∣ w 1 , w 2 , . . . , w i − 1 ) = P ( w i ∣ w i − n + 1 , . . . , w i − 1 ) P(w_i|w_1,w_2,...,w_{i-1}) = P(w_i|w_{i-n+1},...,w_{i-1}) P(wi∣w1,w2,...,wi−1)=P(wi∣wi−n+1,...,wi−1)
基于上式,定义 n-gram 语言模型如下:
n=1 unigram: P ( w 1 , w 2 , . . . , w n ) = ∏ i = 1 n P ( w i ) P(w_1,w_2,...,w_n) = \prod_{i=1}^{n}P(w_i) P(w1,w2,...,wn)=i=1∏nP(wi)
n=2 bigram: P ( w 1 , w 2 , . . . , w n ) = ∏ i = 1 n P ( w i ∣ w i − 1 ) P(w_1,w_2,...,w_n) = \prod_{i=1}^{n}P(w_i|w_{i-1}) P(w1,w2,...,wn)=i=1∏nP(wi∣wi−1)
n=3 trigram: P ( w 1 , w 2 , . . . , w n ) = ∏ i = 1 n P ( w i ∣ w i − 2 , w i − 1 ) P(w_1,w_2,...,w_n) = \prod_{i=1}^{n}P(w_i|w_{i-2},w_{i-1}) P(w1,w2,...,wn)=i=1∏nP(wi∣wi−2,wi−1)
P ( w i ∣ w i − 1 ) = P ( w i w i − 1 ) P ( w i − 1 ) = c o u n t ( w i w i − 1 ) c o u n t ( w i − 1 ) P(w_i|w_{i-1}) = \frac{P(w_iw_{i-1})}{P(w_{i-1})} = \frac{count(w_iw_{i-1})}{count(w_{i-1})} P(wi∣wi−1)=P(wi−1)P(wiwi−1)=count(wi−1)count(wiwi−1)
应用场景:
候选句打分:常应用在文本纠错、语音识别等场景中。比如: P ( I a m h a p p y ) > P ( I a m h a p p e n ) P(I\;am\;happy) > P(I\;am\;happen) P(Iamhappy)>P(Iamhappen)
文本生成:比如广告文案、机器人写作、机器翻译。
按照输入数据的不同,文本生成任务可以分为三类:
基于深度学习的文本生成,常用的是 Seq2Seq 模型 ( E n c o d e r − D e c o d e r ) (Encoder-Decoder) (Encoder−Decoder)。利用 Attention 机制的 Seq2Seq 模型可以加强单词和特征之间的对齐,在生成文字的时候,模拟人类注意力转移的过程,生成更加复合人类习惯的文本。
用深度学习生成序列数据的通用方法,就是使用前面的标记作为输入,训练一个网络(通常是循环神经网络或卷积神经网络)来预测序列中接下来的一个或多个标记。标记通常是单词或字符,给定前面的标记,能够对下一个标记的概率进行建模的任何网络都叫作语言模型。语言模型能够捕捉到语言的潜在空间
(
l
a
t
e
n
t
s
p
a
c
e
)
(latent space)
(latentspace),即语言的统计结构。
一旦训练好了这样一个语言模型,就可以从中采样
(
s
a
m
p
l
e
,
即
生
成
新
序
列
)
(sample,即生成新序列)
(sample,即生成新序列)。向模型中输入一个初始文本字符串
(
c
o
n
d
i
t
i
o
n
i
n
g
d
a
t
a
,
即
条
件
数
据
)
(conditioning data, 即条件数据)
(conditioningdata,即条件数据),要求模型生成下一个字符或下一个单词(甚至可以同时生成多个标记),然后将生成的输出添加到输入数据中,并多次重复这一过程。这个循环可以生成任意长度的序列,这些序列反应了模型训练数据的结构,它们与人类书写的句子几乎相同。
我们将会用到一个 LSTM 层,向其输入从文本语料中提取的 N 个字符组成的字符串,然后训练模型来生成第 N+1 个字符。模型的输出是对所有可能的字符做 Softmax,得到下一个字符的概率分布。这个 LSTM 叫作字符级的神经语言模型。
基于 S e q 2 S e q Seq2Seq Seq2Seq 模型的文本生成有各种不同的 d e c o d i n g s t r a t e g y decoding\;strategy decodingstrategy。文本生成中的 d e c o d i n g s t r a t e g y decoding\;strategy decodingstrategy 主要可以分为两大类:
A r g m a x D e c o d i n g Argmax\;Decoding ArgmaxDecoding:
主要包括: b e a m s e a r c h beam\;search beamsearch, c l a s s − f a c t o r e d s o f t m a x class-factored\;softmax class−factoredsoftmax 等。
S t o c h a s t i c D e c o d i n g Stochastic\;Decoding StochasticDecoding:
主要包括 t e m p e r a t u r e s a m p l i n g temperature\;sampling temperaturesampling, t o p K s a m p l i n g topK\;sampling topKsampling 等。
在 S e q 2 S e q Seq2Seq Seq2Seq 模型中, R N N E n c o d e r RNN\;Encoder RNNEncoder 对输入句子进行编码,生成一个大小固定的 h i d d e n s t a t e h c hidden\;state\;h_c hiddenstatehc;基于输入句子的 h i d d e n s t a t e h c hidden\;state\;h_c hiddenstatehc 和先前生成的第 1 1 1 到 t − 1 t-1 t−1 个词 x 1 : t − 1 x_{1 : t−1} x1:t−1, R N N D e c o d e r RNN\;Decoder RNNDecoder 会生成当前第 t t t 个词的 h i d d e n s t a t e h t hidden\;state\;h_t hiddenstateht,最后通过 s o f t m a x softmax softmax 函数得到第 t t t 个词 x t x_t xt 的 v o c a b u l a r y p r o b a b i l i t y d i s t r i b u t i o n P ( x ∣ x 1 : t − 1 ) vocabulary\;probability\;distribution\; P(x|x_{1:t−1}) vocabularyprobabilitydistributionP(x∣x1:t−1)。
两类 d e c o d i n g s t r a t e g y decoding\;strategy decodingstrategy 的主要区别就在于,如何从 v o c a b u l a r y p r o b a b i l i t y d i s t r i b u t i o n P ( x ∣ x 1 : t − 1 ) vocabulary\;probability\;distribution\; P(x|x_{1:t−1}) vocabularyprobabilitydistributionP(x∣x1:t−1) 中选取一个词 x t x_t xt :
在做 s e q p r e d c i t i o n seq\;predcition seqpredcition 时,需要根据假设模型每个时刻 s o f t m a x softmax softmax 的输出概率来 s a m p l e sample sample 单词,合适的 s a m p l e sample sample 方法可能会获得更有效的结果。
核心思想: 每一步取当前最大可能性的结果,作为最终结果。
具体方法: 获得新生成的词是vocab中各个词的概率,取 a r g m a x argmax argmax 作为需要生成的词向量索引,继而生成后一个词。
核心思想: b e a m s e a r c h beam\;search beamsearch尝试在广度优先基础上进行进行搜索空间的优化(类似于剪枝)达到减少内存消耗的目的。
具体方法: 在 d e c o d i n g decoding decoding 的每个步骤,我们都保留着 t o p − K top-K top−K 个可能的候选单词,然后到了下一个步骤的时候,我们对这 K K K 个单词都做下一步 d e c o d i n g decoding decoding,分别选出 t o p K top K topK,然后对这 K 2 K^2 K2 个候选句子再挑选出 t o p K top K topK 个句子。以此类推一直到 d e c o d i n g decoding decoding 结束为止。当然 B e a m S e a r c h Beam\;Search BeamSearch 本质上也是一个 g r e e d y d e c o d i n g greedy\;decoding greedydecoding 的方法,所以我们无法保证自己一定可以得到最好的 d e c o d i n g decoding decoding 结果。
G r e e d y S e a r c h Greedy\;Search GreedySearch 和 B e a m S e a r c h Beam\;Search BeamSearch 存在的问题:
核心思想: 根据单词的概率分布随机采样。
**具体方法:**在
s
o
f
t
m
a
x
softmax
softmax 中引入一个
t
e
m
p
e
r
a
t
u
r
e
temperature
temperature 来改变
v
o
c
a
b
u
l
a
r
y
p
r
o
b
a
b
i
l
i
t
y
d
i
s
t
r
i
b
u
t
i
o
n
vocabulary\;probability\;distribution
vocabularyprobabilitydistribution,使其更偏向
h
i
g
h
p
r
o
b
a
b
i
l
i
t
y
w
o
r
d
s
high\;probability\;words
highprobabilitywords:
另一种表示:假设 p ( x ) p(x) p(x) 为模型输出的原始分布,给定一个 t e m p e r a t u r e temperature temperature 值,将按照下列方法对原始概率分布(即模型的 s o f t m a x softmax softmax 输出) 进行重新加权,计算得到一个新的概率分布。
当 t e m p e r a t u r e → 0 temperature→0 temperature→0,就变成 g r e e d y s e a r c h greedy\;search greedysearch;当 t e m p e r a t u r e → ∞ temperature→\infty temperature→∞,就变成均匀采样 ( u n i f o r m s a m p l i n g ) (uniform\;sampling) (uniformsampling)。详见论文:The Curious Case of Neural Text Degeneration
可以缓解生成罕见单词的问题。比如说,我们可以每次只在概率最高的 50 50 50 个单词中按照概率分布做采样。我只保留 t o p k topk topk 个 p r o b a b i l i t y probability probability 的单词,然后在这些单词中根据概率做 s a m p l i n g sampling sampling。
核心思想: 对概率进行降序排序,然后对第k个位置之后的概率转换为 0 0 0。
具体方法: 在 d e c o d i n g decoding decoding 过程中,从 P ( x ∣ x 1 : t − 1 ) P(x|x_{1:t−1}) P(x∣x1:t−1) 中选取 p r o b a b i l i t y probability probability 最高的前 k k k 个 t o k e n s tokens tokens,把它们的 p r o b a b i l i t y probability probability 加总得到 p ′ = ∑ P ( x ∣ x 1 : t − 1 ) p′=∑P(x|x_{1:t−1}) p′=∑P(x∣x1:t−1) ,然后将 P ( x ∣ x 1 : t − 1 ) P(x|x_{1:t−1}) P(x∣x1:t−1) 调整为 P ′ ( x ∣ x 1 : t − 1 ) = P ( x ∣ x 1 : t − 1 ) p ′ P′(x|x_{1:t−1})=\frac{P(x|x_{1:t−1})}{p′} P′(x∣x1:t−1)=p′P(x∣x1:t−1) ,其中 x ∈ V ( k ) ! x∈V(k)! x∈V(k)! ,最后从 P ′ ( x ∣ x 1 : t − 1 ) P′(x|x_{1:t−1}) P′(x∣x1:t−1) 中 s a m p l e sample sample 一个 t o k e n token token 作为 o u t p u t t o k e n output\;token outputtoken。详见论文:Hierarchical Neural Story Generation
但 T o p − k S a m p l i n g Top-k\;Sampling Top−kSampling 存在的问题是,常数 k k k 是提前给定的值,对于长短大小不一,语境不同的句子,我们可能有时需要比 k k k 更多的 t o k e n s tokens tokens。
核心思想: 通过对概率分布进行累加,然后当累加的值超过设定的阈值 p p p,则对之后的概率进行置 0 0 0。
具体方法: 提出了 T o p − p S a m p l i n g Top-p\;Sampling Top−pSampling 来解决 T o p − k S a m p l i n g Top-k\;Sampling Top−kSampling 的问题,基于 T o p − k S a m p l i n g Top-k\;Sampling Top−kSampling,它将 p ′ = ∑ P ( x ∣ x 1 : t − 1 ) p′=∑P(x|x_{1:t−1}) p′=∑P(x∣x1:t−1) 设为一个提前定义好的常数 p ′ ∈ ( 0 , 1 ) p′∈(0,1) p′∈(0,1) ,而 s e l e c t e d t o k e n s selected\;tokens selectedtokens 根据句子 h i s t o r y d i s t r i b u t i o n history\;distribution historydistribution 的变化而有所不同。详见论文:The Curious Case of Neural Text Degeneration
本质上 T o p − p S a m p l i n g Top-p\;Sampling Top−pSampling 和 T o p − k S a m p l i n g Top-k\;Sampling Top−kSampling 都是从 t r u n c a t e d v o c a b u l a r y d i s t r i b u t i o n truncated\;vocabulary\;distribution truncatedvocabularydistribution 中 s a m p l e t o k e n sample\;token sampletoken,区别在于置信区间的选择。
随机采样存在的问题:
生成文本时,如何选择下一个字符至关重要。一种简单的方法是贪婪采样 ( g r e e d y s a m p l i n g ) (greedy\;sampling) (greedysampling),就是始终选择可能性最大的下一个字符。但这种方法得到重复的、可预测的字符串,看起来不像是连贯的语言。
在采样的过程中引入随机性,即从下一个字符的概率分布中进行采样。这叫作随机采样
(
s
t
o
c
h
a
s
t
i
c
s
a
m
p
l
i
n
g
)
(stochastic\;sampling)
(stochasticsampling),在这种情况下,根据模型结果,如果下一个字符是 e 的概率为 0.3,那么就有 30% 的概率选择它。假设
p
(
x
)
p(x)
p(x) 为模型输出的原始分布,则基于 softmax 采样得到的新分布:
π
(
x
k
)
=
e
l
o
g
(
p
(
x
k
)
)
∑
i
=
1
n
e
l
o
g
(
p
(
x
i
)
)
\pi(x_k) = \frac{e^{log(p(x_k))}}{\sum_{i=1}^{n}e^{log(p(x_i))}}
π(xk)=∑i=1nelog(p(xi))elog(p(xk))
从模型的 Softmax 输出中进行概率采样是一种很巧妙的方法,它甚至可以在某些时候采样到不常见的字符,从而生成看起来更加有趣的句子,而且有时会得到训练数据中没有的、听起来像是真实存在的新单词,从而表现出创造性。但这种方法有一个问题,就是它在采样过程中无法控制随机性的大小。
考虑一个极端的例子——纯随机采样: 即从均匀概率分布中抽取下一个字符,其中每个字符的概率相同。这种方案具有最大的随机性,换句话说,这种概率分布具有最大的熵。当然,他不会生成任何有趣的内容。
再来看另一个极端——贪婪采样:贪婪采样也不会生成任何有趣的内容,它没有任何随机性,即相应的概率分布具有最小的熵。从"真实概率分布"(即模型 Softmax 函数输出的分布)中进行采样,是这两个极端之间的一个中间点。但是,还有许多其他中间点具有更大或更小的熵,你可能都想研究一下。更小的熵可以让生成的序列具有更加可预测的结构,因此看起来更真实。而更大的熵会得到更加出人意料且更有创造性的序列。从生成式模型中进行采样时,在生成过程中探索不同的随机性大小总是好的做法。我们人类是生成数据是否有趣的最终判断者,所以有趣是非常主观的,我们无法提前知道最佳熵的位置。
为了在采样过程中控制随机性的大小,我们引入一个叫作Softmax 温度的参数,用于表示采样概率分布的熵,即表示所选择的下一个字符会有多么出人意料或多么可预测。给定一个 temperature 值,将按照下列方法对原始概率分布(即模型 Softmax 输出)进行重新加权,计算得到一个新的概率分布。 π ( x k ) = e l o g ( p ( x k ) ) / t e m p e r a t u r e ∑ i = 1 n e l o g ( p ( x i ) ) / t e m p e r a t u r e , t e m p e r a t u r e ∈ [ 0 , 1 ) \pi(x_k) = \frac{e^{log(p(x_k))} / temperature}{\sum_{i=1}^{n}e^{log(p(x_i))} / temperature}, temperature \in [0,1) π(xk)=∑i=1nelog(p(xi))/temperatureelog(p(xk))/temperature,temperature∈[0,1)
import numpy as np
def reweight_distribution(original_distribution, temperature=0.5):
# original_distribution 是概率值组成的一维 Numpy 数组。temperature 是一个因子,用于定量描述输出分布的熵。
distribution = np.log(original_distribution) / temperature
distribution = np.exp(distribution)
# 返回原始分布重新加权后的结果。distribution 的求和可能不再等于1,因此需要归一化
return distribution / np.sum(distribution)
更高的温度得到的是熵更大的分布,会生成更加出人意料、更加无结构的生成数据,而更低的温度对应更小的随机性,以及更加可预测的生成数据。
文本序列生成的程序流程:
生成批次(Batch generation)
数据格式的预处理: b a t c h _ g e n e r a t o r ( d a t a , b a t c h _ s i z e , n _ s t e p s ) batch\_generator(data, batch\_size, n\_steps) batch_generator(data,batch_size,n_steps).
数据的连续采样,要求相邻批次的序列是连续的。需要设置一个批次/周期内的序列个数 b a t c h _ s i z e batch\_size batch_size 以及每个序列中所含字符数 n _ s t e p s n\_steps n_steps ,这样可以得到一个批次/周期数量。
def batch_generator(data, batch_size, n_steps):
data = copy.copy(data)
batch_steps = batch_size * n_steps
n_batches = int(len(data) / batch_steps)
data = data[:batch_size * n_batches]
data = data.reshape((batch_size, -1))
while True:
np.random.shuffle(data)
for n in range(0, data.shape[1], n_steps):
x = data[:, n:n + n_steps]
y = np.zeros_like(x)
y[:, :-1], y[:, -1] = x[:, 1:], x[:, 0]
yield x, y
假设训练集含有 30 30 30 个字符,调用 b a t c h _ g e n e r a t o r ( d a t a , 2 , 6 ) batch\_generator(data, 2, 6) batch_generator(data,2,6).
每个 b a t c h batch batch 的行数就是每个周期的句子数,列数就是 l e n ( d a t a ) b a t c h s i z e \frac{len(data)}{batch_size} batchsizelen(data)得到的数值。
这个网络是由多层 L S T M LSTM LSTM 堆叠组成,然后根据所有时刻的输出 l s t m _ o u t p u t s lstm\_outputs lstm_outputs,对词典中所有词的 s o f t m a x softmax softmax 得到概率。
采样策略:贪婪采样,随机采样
采样函数可以选择 T e m p e r a t u r e Temperature Temperature S a m p l i n g Sampling Sampling 或 T o p K TopK TopK S a m p l i n g Sampling Sampling,分别根据模型得到的原始概率分布进行重新加权或限定 T o p K TopK TopK 个概率构建新的概率分布,并从中抽取一个字符索引。
给定一个训练好的模型和一个种子文本片段,可以通过重复以下操作生成新的文本。
如何对生成的文本进行评价也是文本生成研究中重要的一环。
内在评价关注文本的正确性、流畅度和易理解性。常见的内在评价方法又可分为两类:
找出输出句子与参考句子之间的 n − g r a m n-gram n−gram 重叠部分并对(比参考句子)更短的输出句子施以惩罚的评价方法。
论文地址:BLEU: a Method for Automatic Evaluation of Machine Translation
它基于 N g r a m Ngram Ngram 的稀缺性对其进行加权。这就意味着对某个稀缺 N g r a m Ngram Ngram 的正确匹配能提高的分数,要多于对某个常见的 N g r a m Ngram Ngram 的正确匹配。
论文地址:Automatic Evaluation of Machine Translation QualityUsing N-gram Co-Occurrence Statistics
由于 B L E U BLEU BLEU 只考虑了精确率而没有考虑召回率,因此,在 2004 2004 2004 年,Chin-Yew Lin 提出了一种新的评估方法 R O U G E ROUGE ROUGE,该方法主要是从召回率的角度计算生成文本与参考文本之间的相似性,比较适用于文本摘要任务,因为文本摘要我们更考察生成文本包括了多少参考文本中包含的信息。作者在论文中总共提出了 4 4 4 种不同的计算方式: R O U G E − N ROUGE-N ROUGE−N、 R O U G E − L ROUGE-L ROUGE−L、 R O U G E − W ROUGE-W ROUGE−W、 R O U G E − S ROUGE-S ROUGE−S。
它对 B L E U BLEU BLEU 进行了修改,聚焦于召回率而非准确率。换句话说,该方法看重的是参考翻译句中有多少 N g r a m Ngram Ngram 出现在输出句中,而不是输出句中有多少 N g r a m Ngram Ngram 出现在参考翻译句中。
论文地址:ROUGE: A Package for Automatic Evaluation of Summaries
B L E U BLEU BLEU 只考虑精确率, M E T E O R METEOR METEOR 则同时考虑精确率和召回率,采用加权的 F F F 值来作为评估指标。
论文地址:Meteor: An Automatic Metric for MT Evaluation with High Levels of Correlation with Human Judgments
外在评价则关注生成文本在实际应用中的可用性。
困惑度(perplexity):是交叉熵的指数形式。给定一个包含 n n n 个词的测试文本 W = ( w 1 , w 2 , . . . , w n ) W =(w_1, w_2, ..., w_n) W=(w1,w2,...,wn) 和语言模型 N g r a m Ngram Ngram: P ( w 1 , w 2 , . . . , w n ) = ∏ i = 1 n P ( w i ∣ w i − N + 1 , . . . , w i − 1 ) P(w_1,w_2,...,w_n)=\prod ^n_{i=1}P(w_i|w_{i−N+1},...,w_{i−1}) P(w1,w2,...,wn)=∏i=1nP(wi∣wi−N+1,...,wi−1) 的困惑度 p e r p l e x i t y perplexity perplexity 可定义为交叉熵的指数形式:
p e r p l e x i t y perplexity perplexity 和交叉熵一样都可以用来评价语言模型的好坏。 对于测试集其困惑度越小,准确率也就越高,语言模型也就越好。
p e r p l e x i t y perplexity perplexity 刻画的是语言模型预测一个语言样本的能力,比如已经知道了 W = ( w 1 , w 2 , . . . , w n ) W = (w_1, w_2, ..., w_n) W=(w1,w2,...,wn) 这句话会出现在语料库之中,那么通过语言模型计算得到这句话的概率越高,说明语言模型对这个语料库拟合的越好。
本例将使用尼采的一些作品,用于训练语言模型。因此,我们的要学习的语言模型将是针对尼采的写作风格和主题的模型,而不是一个通用的模型。
首先下载预料,并将其转换为小写。
import keras
import numpy as np
path = keras.utils.get_file('nietzsche.txt', origin='https://s3.amazonaws.com/text-datasets/nietzsche.txt')
text = open(path).read().lower()
print('Corpus length:', len(text))
接下来,我们要提取长度为 maxlen 的序列,这些序列之间会存在部分重叠,对它们进行 one-hot 编码,然后将其打包成形状为 ( s e q u e n c e , m a x l e n , u n i q u e _ c h a r a c t e r s ) (sequence, maxlen, unique\_characters) (sequence,maxlen,unique_characters) 的三维 Numpy 数组。与此同时,我们还需要准备一个数组 y,其中包含对应的目标,即在每一个所提取的序列之后出现的字符(已进行 one-hot 编码)。
maxlen = 60 # 提取 60 个字符组成的序列 step = 3 # 每 3 个字符采样一个新序列 sentences = [] # 保存提取的序列 next_chars = [] # 保存目标(即下一个字符) for i in range(0, len(text)-maxlen, step): sentences.append(text[i:i+maxlen]) next_chars.append(text[i+maxlen]) print("Number of sequences: ", len(sentences)) chars = sorted(list(set(text))) # 语料中唯一字符组成的列表 print("Unique characters: ", len(chars)) # 字典,将唯一字符映射为它在列表 chars 中的索引 char_indices = dict((char, chars.index(char)) for char in chars) print("Vectorization...") # 将字符 one-hot 编码为二进制数组 x = np.zeros((len(sentences), maxlen, len(chars)), dtype=np.bool) y = np.zeros((len(sentences), len(chars)), dtype=np.bool) for i, sentence in enumerate(sentences): for t, char in enumerate(sentence): x[i, t, char_indices[char]] = 1 y[i, char_indices[next_chars[i]]] = 1
这个网络是一个单层 LSTM,然后是一个 Dense 分类器和对所有可能字符的 Softmax。值得注意的是,循环神经网络并不是序列数据生成的唯一方法,最近已有研究表明一维卷积神经网络也可以用于序列数据生成。
from keras import layers
model = keras.models.Sequential()
model.add(layers.LSTM(128, input_shape=(maxlen, len(chars))))
model.add(layers.Dense(len(chars), activation='softmax'))
目标是经过 one-hot 编码的,所以训练模型需要使用 categorical_crossentropy 作为损失。
optimizer = keras.optimizers.RMSprop(lr=0.01)
model.compile(loss='categorical_crossentropy', optimizer=optimizer)
给定一个训练好的模型和一个种子文本片段,我们可以通过重复以下操作来生成新的文本。
下列代码将对模型得到的原始概率分布进行重新加权,并从中抽取一个字符索引。
def sample(preds, temperature=1.0):
preds = np.asarray(preds).astype('float64')
preds = np.log(preds) / temperature
exp_preds = np.exp(preds)
preds = exp_preds / np.sum(exp_preds)
probas = np.random.multinomial(1, preds, 1)
return np.argmax(probas)
最后,下面这个循环将反复训练并生成文本。在每轮过后都是用一系列不同的温度值来生成文本。这样我们可以看到,随着模型收敛,生成的文本如何变化,以及温度对采样的策略的影响。
import random import sys for epoch in range(1, 60): print("epoch: ", epoch) model.fit(x, y, batch_size=128, epochs=1) # 将模型在数据上拟合一次 # 随机选择一个文本种子 start_index = random.randint(0, len(text)-maxlen-1) generated_text = text[start_index: start_index+maxlen] print('---Generating with seed: "' + generated_text + '"') # 尝试一系列不同的采样温度值 for temperature in [0.2, 0.5, 1.0, 1.2]: print('------ temperature: ', temperature) sys.stdout.write(generated_text) for i in range(400): # 对目前生成的字符进行 one-hot 编码 sampled = np.zeros((1, maxlen, len(chars))) for t, char in enumerate(generated_text): sampled[0, t, char_indices[char]] = 1. # 对下一个字符进行采样 preds = model.predict(sampled, verbose=0)[0] next_index = sample(preds, temperature) next_char = chars[next_index] generated_text += next_char generated_text = generated_text[1:] sys.stdout.write(next_char)
较小的温度值会得到极端重复和可预测的文本,但局部结构是非常真实的,特别是所有单词都是真正的英文单词(单词其实就是字符的局部模式)。随着温度值越来越大,生成的文本也变得更加有趣、更出人意料,甚至更有创造性,它有时会创造出全新的单词,听起来有几分可信。
对于较大的温度值,局部模式开始分解,大部分单词看起来像是半随机的字符串。毫无疑问,在这个特定的设置下, 0.5 的温度值生成的文本最为有趣。一定要尝试多种采样策略,在学到的结构与随机性挚爱金,巧妙的平衡能够让生成的序列非常有趣。
值得注意,利用更多的数据训练一个更大的模型,并且训练时间更长,会让生成的样本看起来更连贯、更真实。但是,不要期待能够生成任何有意义的文本,除非是很偶然的情况。我们所做的一切只是从一个统计模型中对数据进行采样,这个模型是关于字符先后顺序的模型。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。