網站首頁 編程語言 正文
貪婪搜索是在每個時間步中選擇概率最高的單詞,也是我們最常用的一種方法,Beam Search不取每個標記本身的絕對概率,而是考慮每個標記的所有可能擴展。然后根據其對數概率選擇最合適的標記序列。
例如令牌的概率如下所示:
例如,Pancakes + looks時間段1的概率等效于:
Pancakes looks so = log(0.2) + log(0.7)= -1.9
Pancakes looks fluffy = log(0.2) + log(0.3)= -2.8
所以我們需要定義一個函數來完成整句的概率計算:
import torch.nn.functional as F
def log_probability_single(logits, labels):
logp = F.log_softmax(logits, dim=-1)
logp_label = torch.gather(logp, 2, labels.unsqueeze(2)).squeeze(-1)
return logp_label
def sentence_logprob(model, labels, input_len=0):
with torch.no_grad():
result = model(labels)
log_probability = log_probability_single(result.logits[:, :-1, :],
labels[:, 1:])
sentence_log_prob = torch.sum(log_probability[:, input_len:])
return sentence_log_prob.cpu().numpy()
接下來,可以將其應用于貪婪搜索解碼方法生成的輸出,并計算生成的序列的對數概率。
在此示例中,我將在村上春木的書中簡要介紹:1Q84。
input_sentence = "A love story, a mystery, a fantasy, a novel of self-discovery, a dystopia to rival George Orwell's — 1Q84 is Haruki Murakami's most ambitious undertaking yet: an instant best seller in his native Japan, and a tremendous feat of imagination from one of our most revered contemporary writers."
max_sequence = 100
input_ids = tokenizer(input_sentence,
return_tensors='pt')['input_ids'].to(device)
output = model.generate(input_ids, max_length=max_sequence, do_sample=False)
greedy_search_output = sentence_logprob(model,
output,
input_len=len(input_ids[0]))
print(tokenizer.decode(output[0]))
我們可以看到生成的序列的對數概率為-52.31。
現在,我們將并比較通過Beam Search生成的序列的對數概率得分,得分越高潛在結果越好。
我們可以增加n-gram懲罰參數no_repeat_ngram_size,這有助于減少輸出中的重復生成的序列。
beam_search_output = model.generate(input_ids,
max_length=max_sequence,
num_beams=5,
do_sample=False,
no_repeat_ngram_size=2)
beam_search_log_prob = sentence_logprob(model,
beam_search_output,
input_len=len(input_ids[0]))
print(tokenizer.decode(beam_search_output[0]))
print(f"\nlog_prob: {beam_search_log_prob:.2f}")
輸出如下:
分時和連貫性要比貪婪的方法好很多,對吧。
原文鏈接:https://blog.csdn.net/deephub/article/details/125887084
相關推薦
- 2022-10-12 python?time時間庫詳解_python
- 2023-07-08 vs編譯運行報錯:未聲明的標識符
- 2022-05-01 C#中的三種定時計時器Timer用法介紹_C#教程
- 2022-12-24 Golang實現組合模式和裝飾模式實例詳解_Golang
- 2022-06-26 ASP.NET?Core中間件會話狀態讀寫及生命周期示例_實用技巧
- 2022-09-25 python學習筆記之循環語句
- 2022-10-16 Pytorch?linear?多維輸入的參數問題_python
- 2023-04-06 C++中的memset用法詳解_C 語言
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版(
- Linux 中運行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符,流程控制 Flo
- 1. Int 和Integer 的區別,Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤:Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串(SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列,發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支