作者 |Yuki Takahashi
编译 |VK
起源 |Towards Datas Science
自由 ImageNet 上推出 AlexNet 以来,计算机视觉的深度学习已胜利利用于各种利用。相同,NLP 在深层神经网络应用方面始终落后。许多宣称应用人工智能的应用程序通常应用某种基于规定的算法和传统的机器学习,而不是应用深层神经网络。
2018 年,在一些 NLP 工作中,一种名为 BERT 的最先进(STOA)模型的体现超过了人类的得分。在这里,我将几个模型利用于情绪剖析工作,以理解它们在我所处的金融市场中有多大用处。代码在 jupyter notebook 中,在 git repo 中可用:https://github.com/yuki678/fi…
介绍
NLP 工作能够大抵分为以下几类。
- 文本分类——过滤垃圾邮件,对文档进行分类
- 词序——词翻译,词性标记,命名实体辨认
- 文本意义——主题模型,搜寻,问答
- seq2seq——机器翻译、文本摘要、问答
- 对话零碎
不同的工作须要不同的办法,在大多数状况下是多种 NLP 技术的组合。在开发机器人时,后端逻辑通常是基于规定的搜索引擎和排名算法,以造成天然的通信。
这是有充沛理由的。语言有语法和词序,能够用基于规定的办法更好地解决,而机器学习办法能够更好地学习单词相似性。向量化技术如 word2vec、bag of word 帮忙模型以数学形式表白文本。最驰名的例子是:
King - Man + Woman = Queen
Paris - France + UK = London
第一个例子形容了性别关系,第二个例子形容了首都的概念。然而,在这些办法中,因为在任何文本中同一个词总是由同一个向量示意,因而上下文不能被捕捉,这在许多状况下是不正确的。
循环神经网络(RNN)构造利用输出序列的先验信息,解决工夫序列数据,在捕获和记忆上下文方面体现良好。LSTM 是一种典型的构造,它由输出门、输入门和忘记门组成,克服了 RNN 的梯度问题。有许多基于 LSTM 的改良模型,例如双向 LSTM,不仅能够从后面的单词中捕获上下文,而且能够从前面捕捉上下文。这些办法对于某些特定的工作是有用的,但在理论利用中却不太实用。
2017 年,咱们看到了一种新的办法来解决这个问题。BERT 是 Google 在 2018 年推出的一个多编码器堆栈的掩码语言模型,在 GLUE、SQuAD 和 SWAG 基准测试中实现了 STOA,并有了很大的改良。有很多文章和博客解释了这种架构,比方 Jay Alammar 的文章:http://jalammar.github.io/ill…
我在金融行业工作,在过来的几年里,我很难看到咱们在 NLP 上的机器学习模型在交易系统中的生产利用方面有足够的强劲体现。当初,基于 BERT 的模型正在变得成熟和易于应用,这要归功于 Huggingface 的实现和许多预训练的模型曾经公开。
我的指标是看看这个 NLP 的最新开发是否达到了在我的畛域中应用的良好程度。在这篇文章中,我比拟了不同的模型,这是一个相当简略的工作,即对金融文本的情绪剖析,以此作为基线来判断是否值得在真正的解决方案中尝试另一个研发。
此处比拟的模型有:
- 基于规定的词典办法
- 基于 Tfidf 的传统机器学习办法
- 作为一种循环神经网络构造的 LSTM
- BERT(和 ALBERT)
输出数据
在情绪剖析工作中,我采纳以下两种输出来示意行业中的不同语言。
- 财经新闻标题——正式
- 来自 Stocktwits 的 Tweets——非正式
我将为后者写另一篇文章,所以这里关注前者的数据。这是一个蕴含更正式的金融畛域特定语言的文本示例,我应用了 Malo 等人的 FinancialPhraseBank(https://www.researchgate.net/…)包含 4845 篇由 16 人手写的题目文本,并提供批准等级。我应用了 75% 的批准等级和 3448 个文本作为训练数据。
## 输出文本示例
positive "Finnish steel maker Rautaruukki Oyj (Ruukki) said on July 7 , 2008 that it won a 9.0 mln euro ($ 14.1 mln) contract to supply and install steel superstructures for Partihallsforbindelsen bridge project in Gothenburg , western Sweden."
neutral "In 2008 , the steel industry accounted for 64 percent of the cargo volumes transported , whereas the energy industry accounted for 28 percent and other industries for 8 percent."
negative "The period-end cash and cash equivalents totaled EUR6 .5 m , compared to EUR10 .5 m in the previous year."
请留神,所有数据都属于起源,用户必须恪守其版权和许可条款。
模型
上面是我比拟了四款模型的性能。
A、基于词汇的办法
创立特定于畛域的词典是一种传统的办法,在某些状况下,如果源代码来自特定的集体或媒体,则这种办法简略而弱小。Loughran 和 McDonald 情感词列表。这个列表蕴含超过 4k 个单词,这些单词呈现在带有情绪标签的财务报表上。注:此数据须要许可证能力用于商业利用。请在应用前查看他们的网站。
## 样本
negative: ABANDON
negative: ABANDONED
constraining: STRICTLY
我用了 2355 个消极单词和 354 个踊跃单词。它蕴含单词模式,因而不要对输出执行词干剖析和词干化。对于这种办法,思考否定模式是很重要的。比方 not,no,don,等等。这些词会把否定词的意思改为必定的,如果后面三个词中有否定词,这里我简略地把否定词的意思转换成必定词。
而后,情感得分定义如下。
tone_score = 100 * (pos_count — neg_count) / word_count
用默认参数训练 14 个不同的分类器,而后用网格搜寻穿插验证法对随机森林进行超参数整定。
classifiers = []
classifiers.append(("SVC", SVC(random_state=random_state)))
classifiers.append(("DecisionTree", DecisionTreeClassifier(random_state=random_state)))
classifiers.append(("AdaBoost", AdaBoostClassifier(DecisionTreeClassifier(random_state=random_state),random_state=random_state,learning_rate=0.1)))
classifiers.append(("RandomForest", RandomForestClassifier(random_state=random_state, n_estimators=100)))
classifiers.append(("ExtraTrees", ExtraTreesClassifier(random_state=random_state)))
classifiers.append(("GradientBoosting", GradientBoostingClassifier(random_state=random_state)))
classifiers.append(("MultipleLayerPerceptron", MLPClassifier(random_state=random_state)))
classifiers.append(("KNeighboors", KNeighborsClassifier(n_neighbors=3)))
classifiers.append(("LogisticRegression", LogisticRegression(random_state = random_state)))
classifiers.append(("LinearDiscriminantAnalysis", LinearDiscriminantAnalysis()))
classifiers.append(("GaussianNB", GaussianNB()))
classifiers.append(("Perceptron", Perceptron()))
classifiers.append(("LinearSVC", LinearSVC()))
classifiers.append(("SGD", SGDClassifier()))
cv_results = []
for classifier in classifiers :
cv_results.append(cross_validate(classifier[1], X_train, y=Y_train, scoring=scoring, cv=kfold, n_jobs=-1))
# 应用随机森林分类器
rf_clf = RandomForestClassifier()
# 执行网格搜寻
param_grid = {'n_estimators': np.linspace(1, 60, 10, dtype=int),
'min_samples_split': [1, 3, 5, 10],
'min_samples_leaf': [1, 2, 3, 5],
'max_features': [1, 2, 3],
'max_depth': [None],
'criterion': ['gini'],
'bootstrap': [False]}
model = GridSearchCV(rf_clf, param_grid=param_grid, cv=kfold, scoring=scoring, verbose=verbose, refit=refit, n_jobs=-1, return_train_score=True)
model.fit(X_train, Y_train)
rf_best = model.best_estimator_
B、基于 Tfidf 向量的传统机器学习
输出被 NLTK word_tokenize() 标记化,而后词干化和删除停用词。而后输出到 TfidfVectorizer,通过 Logistic 回归和随机森林分类器进行分类。
### 逻辑回归
pipeline1 = Pipeline([('vec', TfidfVectorizer(analyzer='word')),
('clf', LogisticRegression())])
pipeline1.fit(X_train, Y_train)
### 随机森林与网格搜寻
pipeline2 = Pipeline([('vec', TfidfVectorizer(analyzer='word')),
('clf', RandomForestClassifier())])
param_grid = {'clf__n_estimators': [10, 50, 100, 150, 200],
'clf__min_samples_leaf': [1, 2],
'clf__min_samples_split': [4, 6],
'clf__max_features': ['auto']
}
model = GridSearchCV(pipeline2, param_grid=param_grid, cv=kfold, scoring=scoring, verbose=verbose, refit=refit, n_jobs=-1, return_train_score=True)
model.fit(X_train, Y_train)
tfidf_best = model.best_estimator_
C、LSTM
因为 LSTM 被设计用来记忆表白上下文的长期记忆,因而应用自定义的 tokenizer 并且输出是字符而不是单词,所以不须要词干化或输入停用词。输出先到一个嵌入层,而后是两个 lstm 层。为了防止过拟合,利用 dropout,而后是全连贯层,最初采纳 log softmax。
class TextClassifier(nn.Module):
def __init__(self, vocab_size, embed_size, lstm_size, dense_size, output_size, lstm_layers=2, dropout=0.1):
"""初始化模型"""
super().__init__()
self.vocab_size = vocab_size
self.embed_size = embed_size
self.lstm_size = lstm_size
self.dense_size = dense_size
self.output_size = output_size
self.lstm_layers = lstm_layers
self.dropout = dropout
self.embedding = nn.Embedding(vocab_size, embed_size)
self.lstm = nn.LSTM(embed_size, lstm_size, lstm_layers, dropout=dropout, batch_first=False)
self.dropout = nn.Dropout(dropout)
if dense_size == 0:
self.fc = nn.Linear(lstm_size, output_size)
else:
self.fc1 = nn.Linear(lstm_size, dense_size)
self.fc2 = nn.Linear(dense_size, output_size)
self.softmax = nn.LogSoftmax(dim=1)
def init_hidden(self, batch_size):
"""初始化暗藏状态"""
weight = next(self.parameters()).data
hidden = (weight.new(self.lstm_layers, batch_size, self.lstm_size).zero_(),
weight.new(self.lstm_layers, batch_size, self.lstm_size).zero_())
return hidden
def forward(self, nn_input_text, hidden_state):
"""在 nn_input 上执行模型的前项流传"""
batch_size = nn_input_text.size(0)
nn_input_text = nn_input_text.long()
embeds = self.embedding(nn_input_text)
lstm_out, hidden_state = self.lstm(embeds, hidden_state)
# 重叠 LSTM 输入,利用 dropout
lstm_out = lstm_out[-1,:,:]
lstm_out = self.dropout(lstm_out)
# 全连贯层
if self.dense_size == 0:
out = self.fc(lstm_out)
else:
dense_out = self.fc1(lstm_out)
out = self.fc2(dense_out)
# Softmax
logps = self.softmax(out)
return logps, hidden_state
作为代替,还尝试了斯坦福大学的 GloVe 词嵌入,这是一种无监督的学习算法,用于获取单词的向量示意。在这里,用 6 百万个标识、40 万个词汇和 300 维向量对 Wikipedia 和 Gigawords 进行了预训练。在咱们的词汇表中,大概 90% 的单词都是在这个 GloVe 里找到的,其余的都是随机初始化的。
D、BERT 和 ALBERT
我应用了 Huggingface 中的 transformer 实现 BERT 模型。当初他们提供了 tokenizer 和编码器,能够生成文本 id、pad 掩码和段 id,能够间接在 BertModel 中应用,咱们应用规范训练过程。
与 LSTM 模型相似,BERT 的输入随后被传递到 dropout,全连贯层,而后利用 log softmax。如果没有足够的计算资源估算和足够的数据,从头开始训练模型不是一个抉择,所以我应用了预训练的模型并进行了微调。预训练的模型如下所示:
- BERT:bert-base-uncased
- ALBERT:albert-base-v2
预训练过的 bert 的训练过程如下所示。
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased', do_lower_case=True)
model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=3)
def train_bert(model, tokenizer)
# 挪动模型到 GUP/CPU 设施
device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
model = model.to(device)
# 将数据加载到 SimpleDataset(自定义数据集类)
train_ds = SimpleDataset(x_train, y_train)
valid_ds = SimpleDataset(x_valid, y_valid)
# 应用 DataLoader 批量加载数据集中的数据
train_loader = torch.utils.data.DataLoader(train_ds, batch_size=batch_size, shuffle=True)
valid_loader = torch.utils.data.DataLoader(valid_ds, batch_size=batch_size, shuffle=False)
# 优化器和学习率衰减
num_total_opt_steps = int(len(train_loader) * num_epochs)
optimizer = AdamW_HF(model.parameters(), lr=learning_rate, correct_bias=False)
scheduler = get_linear_schedule_with_warmup(optimizer, num_warmup_steps=num_total_opt_steps*warm_up_proportion, num_training_steps=num_total_opt_steps) # PyTorch scheduler
# 训练
model.train()
# Tokenizer 参数
param_tk = {
'return_tensors': "pt",
'padding': 'max_length',
'max_length': max_seq_length,
'add_special_tokens': True,
'truncation': True
}
# 初始化
best_f1 = 0.
early_stop = 0
train_losses = []
valid_losses = []
for epoch in tqdm(range(num_epochs), desc="Epoch"):
# print('================ epoch {} ==============='.format(epoch+1))
train_loss = 0.
for i, batch in enumerate(train_loader):
# 传输到设施
x_train_bt, y_train_bt = batch
x_train_bt = tokenizer(x_train_bt, **param_tk).to(device)
y_train_bt = torch.tensor(y_train_bt, dtype=torch.long).to(device)
# 重设梯度
optimizer.zero_grad()
# 前馈预测
loss, logits = model(**x_train_bt, labels=y_train_bt)
# 反向流传
loss.backward()
# 损失
train_loss += loss.item() / len(train_loader)
# 梯度剪切
torch.nn.utils.clip_grad_norm_(model.parameters(), max_grad_norm)
# 更新权重和学习率
optimizer.step()
scheduler.step()
train_losses.append(train_loss)
# 评估模式
model.eval()
# 初始化
val_loss = 0.
y_valid_pred = np.zeros((len(y_valid), 3))
with torch.no_grad():
for i, batch in enumerate(valid_loader):
# 传输到设施
x_valid_bt, y_valid_bt = batch
x_valid_bt = tokenizer(x_valid_bt, **param_tk).to(device)
y_valid_bt = torch.tensor(y_valid_bt, dtype=torch.long).to(device)
loss, logits = model(**x_valid_bt, labels=y_valid_bt)
val_loss += loss.item() / len(valid_loader)
valid_losses.append(val_loss)
# 计算指标
acc, f1 = metric(y_valid, np.argmax(y_valid_pred, axis=1))
# 如果改良了,保留模型。如果没有,那就提前进行
if best_f1 < f1:
early_stop = 0
best_f1 = f1
else:
early_stop += 1
print('epoch: %d, train loss: %.4f, valid loss: %.4f, acc: %.4f, f1: %.4f, best_f1: %.4f, last lr: %.6f' %
(epoch+1, train_loss, val_loss, acc, f1, best_f1, scheduler.get_last_lr()[0]))
if device == 'cuda:0':
torch.cuda.empty_cache()
# 如果达到急躁数,提前进行
if early_stop >= patience:
break
# 返回训练模式
model.train()
return model
评估
首先,输出数据以 8:2 分为训练组和测试集。测试集放弃不变,直到所有参数都固定下来,并且每个模型只应用一次。因为数据集不用于计算穿插集,因而验证集不用于计算。此外,为了克服数据集不均衡和数据集较小的问题,采纳分层 K -Fold 穿插验证进行超参数整定。
因为输出数据不均衡,因而评估以 F1 分数为根底,同时也参考了准确性。
def metric(y_true, y_pred):
acc = accuracy_score(y_true, y_pred)
f1 = f1_score(y_true, y_pred, average='macro')
return acc, f1
scoring = {'Accuracy': 'accuracy', 'F1': 'f1_macro'}
refit = 'F1'
kfold = StratifiedKFold(n_splits=5)
模型 A 和 B 应用网格搜寻穿插验证,而 C 和 D 的深层神经网络模型应用自定义穿插验证。
# 分层 KFold
skf = StratifiedKFold(n_splits=5, shuffle=True, random_state=rand_seed)
# 循环
for n_fold, (train_indices, valid_indices) in enumerate(skf.split(y_train, y_train)):
# 模型
model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=3)
# 输出数据
x_train_fold = x_train[train_indices]
y_train_fold = y_train[train_indices]
x_valid_fold = x_train[valid_indices]
y_valid_fold = y_train[valid_indices]
# 训练
train_bert(model, x_train_fold, y_train_fold, x_valid_fold, y_valid_fold)
后果
基于 BERT 的微调模型在破费了或多或少类似的超参数调整工夫之后,显著优于其余模型。
模型 A 体现不佳,因为输出过于简化为情感得分,情感分数是判断情绪的繁多值,而随机森林模型最终将大多数数据标记为中性。简略的线性模型只需对情感评分利用阈值就能够取得更好的成果,但在准确度和 f1 评分方面依然很低。
咱们没有应用欠采样 / 过采样或 SMOTE 等办法来均衡输出数据,因为它能够纠正这个问题,但会偏离存在不均衡的理论状况。如果能够证实为每个要解决的问题建设一个词典的老本是正当的,这个模型的潜在改良是建设一个自定义词典,而不是 L - M 词典。
模型 B 比前一个模型好得多,然而它以简直 100% 的准确率和 f1 分数拟合了训练集,然而没有被泛化。我试图升高模型的复杂度以防止过拟合,但最终在验证集中的得分较低。均衡数据能够帮忙解决这个问题或收集更多的数据。
模型 C 产生了与前一个模型类似的后果,但改良不大。事实上,训练数据的数量不足以从零开始训练神经网络,须要训练到多个 epoch,这往往会过拟合。预训练的 GloVe 并不能改善后果。对后一种模型的一个可能的改良是应用相似畛域的大量文本(如 10K、10Q 财务报表)来训练 GloVe,而不是应用维基百科中预训练过的模型。
模型 D 在穿插验证和最终测试中的准确率和 f1 分数均达到 90% 以上。它正确地将负面文本分类为 84%,而侧面文本正确分类为 94%,这可能是因为输出的数量,但最好仔细观察以进一步提高性能。这表明,因为迁徙学习和语言模型,预训练模型的微调在这个小数据集上体现良好。
论断
这个试验展现了基于 BERT 的模型在我的畛域中利用的后劲,以前的模型没有产生足够的性能。然而,后果不是确定性的,如果调整下超参数,后果可能会有所不同。
值得注意的是,在理论利用中,获取正确的输出数据也相当重要。没有高质量的数据(通常被称为“垃圾输出,垃圾输入”)就不能很好地训练模型。
我下次再谈这些问题。这里应用的所有代码都能够在 git repo 中找到:https://github.com/yuki678/fi…
原文链接:https://towardsdatascience.co…
欢送关注磐创 AI 博客站:
http://panchuang.net/
sklearn 机器学习中文官网文档:
http://sklearn123.com/
欢送关注磐创博客资源汇总站:
http://docs.panchuang.net/