主頁 >  其他 > 預訓練模型-從BERT原理到BERT調包和微調

預訓練模型-從BERT原理到BERT調包和微調

2023-04-16 07:52:04 其他

 一、BERT原理

BERT取名來自 Bidirectional Encoder Representations from Transformers,架構為:預訓練 + fine-tuning(對于特定的任務只需要添加一個輸出層), 1、引言        通過預訓練語言模型可顯著提高NLP下游任務,限制模型潛力的主要原因在于現有模型使用的都是單向的語言模型,無法充分了解到單詞所在的背景關系結構,        受完形填空的啟發,BERT通過使用 “Masked Language Model” 的預訓練目標來緩解單向語言模型的約束,        首先 “Masked Language Model” 會隨機屏蔽(masked)15%的token,然后讓模型根據背景關系來預測被Mask的token(被Mask的變成了標簽),具體,將masked token 位置輸出的最終隱層向量送入softmax,來預測masked token,       此外, “Next sentence prediction” 的任務預訓練文本對,將 token-level 提升到 sentence-level,以應用不同種類的下游任務,   2、BERT模型        BERT:pre-training 和 fine-tuning,兩階段的步驟,
  • pre-training 階段,BERT 在無標記的資料上進行無監督學習;
  • fine-tuning 階段,BERT利用預訓練的引數初始化模型,并利用下游任務標記好的資料進行有監督學習,并對所有引數進行微調,
所有下游任務都有單獨的 fine-tuning 模型,即使是使用同樣的預訓練引數,下圖是對 BERT 的一個概覽: 0 2.1、模型架構        BERT 是由多層雙向的 Transformer Encoder 結構組成,區別于 GPT 的單向的 Transformer Decoder 架構, 0        BERT_BASE (L=12, H=768, A=12, Total Parameters=110M) and BERTLARGE (L=24, H=1024, A=16, Total Parameters=340M),   2.2、輸入       為了能應對下游任務,BERT 給出了 sentence-level 級別的 Representation,包括句子和句子對,在 BERT 中 sequence 并不一定是一個句子,也有可能是任意的一段連續的文本;而句子對主要是因為類似 QA 問題,       BERT 的輸入: 0        分為三塊:Token Embeddings、Segment Embeddings 和 Position Embeddings
  •        Token Embeddings 采用的 WordPiece Embedding,共有 30000 個 token,每個 sequence 會以一個特殊的 classi?cation token [CLS] 開始,同時這也會作為分類任務的輸出;句子間會以 special seperator token [SEP] 進行分割,
  •        WordPiece Embedding:n-gram 字符級 Embedding,采用 BPE 雙位元組編碼,可以將單詞拆分,比如 “loved” “loving” ”loves“ 會拆分成 “lov”,“ed”,“ing”,“es”,
  •        Segment Embedding也可以用來分割句子,但主要用來區分句子對,Embedding A 和 Embedding B 分別代表左右句子,如果是普通的句子就直接用 Embedding A,
  •        Position Embedding 是用來給單詞定位的,學習出來的embedding向量,這與Transformer不同,Transformer中是預先設定好的值,
       BERT 最終的 input 是三種不同的 Embedding 直接相加,   2.3、預訓練Pre-training        BERT 采用兩種無監督任務來進行預訓練,兩個任務同時訓練,所以 BERT 的損失函式是兩個任務的損失函式相加:  1) token-level 級別的 Masked LM;   2) sentence-level 級別的 Next Sentence Prediction,   2.3.1、任務一:Masked LM        為解決雙向模型的資料泄漏的問題,Masked LM:隨機屏蔽一些token 并通過背景關系預測這些 token,在實驗中,BERT 會隨機屏蔽每個序列中的 15% 的 token,并用 [MASK] token 來代替,        這會帶來一個問題:[MASK] token 不會出現在下游任務中,為了和后續任務保持一致,采用以下三種方式來代替 [MASK] token:如:my dog is hairy
  • 80% 的 [MASK] token 會繼續保持 [MASK];—my dog is [MASK]
  • 10% 的 [MASK] token 會被隨機的一個單詞取代;my dog is apple
  • 10% 的 [MASK] token 會保持原單詞不變(但是還是要預測)my dog is hairy
       最終 Masked ML 的損失函式是只由被 [MASK] 的部分來計算:
  • 在 encoder 的輸出上添加一個前饋神經網路,將其轉換為詞匯的維度
  • softmax 計算詞匯表中每個單詞的概率
BERT 的損失函式只考慮了 mask 的預測值(CBOW),   2.3.2、Next Sentence Prediction        語言模型只能捕捉 token-level 級別的關系,為了捕捉 sentence-level 級別的關系,訓練了一個 sentence-level 的分類任務, 具體來說,訓練的輸入是句子A和B (句子級負采樣):
  • B有一半的幾率是A的下一句,即正例;
  • B有一半的幾率是隨機取一個句子作為負例,
通過 classi?cation token 連接 Softmax 輸出B是不是A的下一句,     為了幫助模型區分開訓練中的兩個句子,輸入在進入模型之前要按以下方式進行處理:在第一個句子的開頭插入 [CLS] 標記,在每個句子的末尾插入 [SEP] 標記,Segemet Embeding來表示不同的句子,        在訓練 BERT 模型時,Masked LM 和 Next Sentence Prediction 是一起訓練的,目標就是要最小化兩種策略的組合損失函式,   2.4、微調Fine-tuning        由于預訓練的 Transformer 已經完成了句子和句子對的 Representation,所以 BERT 的 Fine-tuning 非常簡單,       針對不同的下游任務,可以將具體的輸入和輸出是適配到 BERT 中,并且采用端到端的訓練去微調模型引數,BERT 可以用于各種NLP任務,只需在核心模型中添加一個層,例如:
  • 1. 在分類任務中,例如情感分析等,只需要在 Transformer 的輸出之上加一個分類層
  • 2. 在QA任務中,問答系統需要接收有關文本序列的 question,并且需要在序列中標記 answer, 可以使用 BERT 學習兩個標記 answer 開始和結尾的向量來訓練Q&A模型,
  • 3. 在命名物體識別(NER)中,系統需要接收文本序列,標記文本中的各種型別的物體(人員,組織,日期等), 可以用 BERT 將每個 token 的輸出向量送到預測 NER 標簽的分類層,

  • a、b 是 sentence-level 級別的任務,類似句子分類,情感分析等等,輸入句子或句子對,在 [CLS] 位置接入 Softmax 輸出 Label;      
  • c是token-level級別的任務,比如 QA 問題,輸入問題和段落,在 Paragraph 對應輸出的 hidden vector 后接上兩個 Softmax 層,分別訓練出 Span 的 Start index 和 End index(連續的 Span)作為 Question 的答案;      
  • d也是token-level級別的任務,比如命名物體識別問題,接上 Softmax 層即可輸出具體的分類,
0   0   0 0 0 生成 0   3. 結論        BERT采用Pre-training和Fine-tuning兩階段訓練任務(源于GPT),在Pre-training階段使用多層雙向Transformer Encoder進行訓練,并采用Masked LM 和 Next Sentence Prediction兩種訓練任務解決 token-level 和 sentence-level 的問題,為下游任務提供了一個通用的模型框架;在 Fine-tuning 階段會針對具體 NLP 任務進行微調以適應不同種類的任務需求,并通過端到端的訓練更新引數從而得到最終的模型,   BERT優點:         Transformer Encoder因為有Self-attention機制,因此BERT自帶雙向功能因為雙向功能以及多層Self-attention機制的影響,使得BERT必須使用Cloze版的語言模型Masked-LM來完成token級別的預訓練為了獲取比詞更高級別的句子級別的語意表征,        BERT加入了Next Sentence Prediction來和Masked-LM一起做聯合訓練為了適配多任務下的遷移學習,BERT設計了更通用的輸入層和輸出層微調成本小   BERT缺點:
  • [MASK]標記在實際預測中不會出現,訓練時用過多[MASK]影響模型表現;
  • 每個batch只有15%的token被預測,所以BERT收斂得比left-to-right模型要慢(它們會預測每個token);
  • BERT對硬體資源的消耗巨大,

二、BERT答疑

1、三個Embedding怎么來的 在BERT中,Token,Position,Segment Embeddings 都是通過學習來得到的,pytorch代碼中它們是這樣的: self.word_embeddings = Embedding(config.vocab_size, config.hidden_size) self.position_embeddings = Embedding(config.max_position_embeddings, config.hidden_size) self.token_type_embeddings = Embedding(config.type_vocab_size, config.hidden_size) BERT 能夠處理對輸入句子對的分類任務,這類任務就像判斷兩個文本是否是語意相似的,句子對中的兩個句子被簡單的拼接在一起后送入到模型中, 那BERT如何去區分一個句子對中的兩個句子呢?答案就是segment embeddings. 0 https://www.cnblogs.com/d0main/p/10447853.html#token-embeddings   2、不考慮多頭的原因,self-attention中詞向量不乘QKV引數矩陣,會有什么問題? Self-Attention的核心是用文本中的其它詞來增強目標詞的語意表示,從而更好的利用背景關系的資訊, self-attention中,sequence中的每個詞都會和sequence中的每個詞做點積去計算相似度,也包括這個詞本身,   3、為什么BERT選擇mask掉15%這個比例的詞,可以是其他的比例嗎? BERT采用的Masked LM,會選取語料中所有詞的15%進行隨機mask, 論文表示受到完形填空任務的啟發,但其實與CBOW也有異曲同工之妙, 從CBOW的角度,有一個比較好的解釋是:在一個大小為 的視窗中隨機選一個詞,類似CBOW中滑動視窗的中心詞,區別是這里的滑動視窗是非重疊的, 那從CBOW的滑動視窗角度,10%~20%都是還ok的比例,   4、為什么BERT在第一句前會加一個[CLS]標志?

BERT在第一句前會加一個CLS]標志,最后一層該位對應向量可以作為整句話的語意表示,從而用于下游的分類任務等,與文本中已有的其它詞相比,這個無明顯語意資訊的符號會更“公平”地融合文本中各個詞的語意資訊,從而更好的表示整句話的語意

5、Self-Attention 的時間復雜度是怎么計算的? O(n^2 d) 相加的   6、Transformer在哪里做了權重共享,為什么可以做權重共享? Transformer在兩個地方進行了權重共享: (1)Encoder和Decoder間的Embedding層權重共享; (2)Decoder中Embedding層和FC層權重共享, 解碼的詞,要有embedding,同時也可作為分類器權重,,Embedding層可以說是通過onehot去取到對應的embedding向量,FC層可以說是相反的,通過向量(定義為 x)去得到它可能是某個詞的softmax概率,取概率最大, FC層的每一行量級相同的前提下,理論上和 x 相同的那一行對應的點積和softmax概率會是最大的(內積), 通過這樣的權重共享可以減少引數的數量,加快收斂,   7、BERT非線性的來源在哪里? 前饋層的gelu激活函式和self-attention,self-attention是非線性的,感謝評論區指出,   8. BERT引數 https://blog.csdn.net/weixin_43922901/article/details/102602557 Bert采用的vocab_size=30522,hidden_size=768,max_position_embeddings=512 LN引數,gamma和beta的維度均為768,因此總引數為768 * 2 + 768 * 2 * 2 * 12(層數)   模型引數Bert模型的版本如下: BERT-Base, Uncased: 12-layer, 768-hidden, 12-heads, 110M parameters BERT-Large, Uncased: 24-layer, 1024-hidden, 16-heads, 340M parameters

 三、BERT調包和微調

BERT有兩個約束條件,
  • 1. 所有的句子必須被填充或截斷成一個固定的長度,
  • 2. 最大的句子長度是512個tokens,
填充是通過一個特殊的"[PAD]"token來完成的,它在BERT詞匯表中的索引0,下面的插圖演示了填充到8個token的 "MAX_LEN", 0 "注意力遮蓋"只是一個1和0的陣列,表示哪些標記是padding,哪些不是, 這個掩碼告訴BERT中的"自我關注"機制,不要將這些pad標記納入它對句子的解釋中, (還有點類似Decoder,不過這是為了對其句子用,而不是為了自回歸)   不過,最大長度確實會影響訓練和評估速度,例如,用特斯拉K80, MAX_LEN = 128 --> 訓練一個 epoch 需要 5:28 MAX_LEN = 64 --> 訓練一個 epoch 需要 2:57,   現在我們準備好執行真正的 tokenization 了,tokenizer.encode_plus函式為我們結合了多個步驟,
  1. 將句子分割成token,
  2. 添加特殊的[CLS]和[SEP]標記,
  3. 將這些標記映射到它們的ID上,
  4. 把所有的句子都墊上或截斷成相同的長度,
  5. 創建注意力Masl,明確區分真實 token 和[PAD]token,
  以下是HuggingFace目前提供的類串列,供微調,
  • BertModel
  • BertForPreTraining
  • BertForMaskedLM
  • BertForNextSentencePrediction(下句預測)
  • BertForSequenceClassification - 我們將使用的那個,
  • BertForTokenClassification
  • BertForQuestionAnswering
from transformers import BertForSequenceClassification, AdamW, BertConfig

# Load BertForSequenceClassification, the pretrained BERT model with a single 
# linear classification layer on top. 
model = BertForSequenceClassification.from_pretrained(
    "bert-base-uncased", # Use the 12-layer BERT model, with an uncased vocab.
    num_labels = 2, # The number of output labels--2 for binary classification.
                    # You can increase this for multi-class tasks.   
    output_attentions = False, # Whether the model returns attentions weights.
    output_hidden_states = False, # Whether the model returns all hidden-states.
)

# Tell pytorch to run this model on the GPU.
model.cuda()

參考

  • 用huggingface.transformers在文本分類任務(單任務和多任務場景下)上微調預訓練模型https://blog.csdn.net/PolarisRisingWar/article/details/127365675
  • 比賽:Datawhale零基礎入門NLP賽事 - Task5 基于深度學習的文本分類https://tianchi.aliyun.com/notebook/118258

轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/550214.html

標籤:其他

上一篇:Transformer網路-Self-attention is all your need

下一篇:再見了青春的自學極客it人變現賺金記錄

標籤雲
其他(157675) Python(38076) JavaScript(25376) Java(17977) C(15215) 區塊鏈(8255) C#(7972) AI(7469) 爪哇(7425) MySQL(7132) html(6777) 基礎類(6313) sql(6102) 熊猫(6058) PHP(5869) 数组(5741) R(5409) Linux(5327) 反应(5209) 腳本語言(PerlPython)(5129) 非技術區(4971) Android(4554) 数据框(4311) css(4259) 节点.js(4032) C語言(3288) json(3245) 列表(3129) 扑(3119) C++語言(3117) 安卓(2998) 打字稿(2995) VBA(2789) Java相關(2746) 疑難問題(2699) 细绳(2522) 單片機工控(2479) iOS(2429) ASP.NET(2402) MongoDB(2323) 麻木的(2285) 正则表达式(2254) 字典(2211) 循环(2198) 迅速(2185) 擅长(2169) 镖(2155) 功能(1967) .NET技术(1958) Web開發(1951) python-3.x(1918) HtmlCss(1915) 弹簧靴(1913) C++(1909) xml(1889) PostgreSQL(1872) .NETCore(1853) 谷歌表格(1846) Unity3D(1843) for循环(1842)

熱門瀏覽
  • 網閘典型架構簡述

    網閘架構一般分為兩種:三主機的三系統架構網閘和雙主機的2+1架構網閘。 三主機架構分別為內端機、外端機和仲裁機。三機無論從軟體和硬體上均各自獨立。首先從硬體上來看,三機都用各自獨立的主板、記憶體及存盤設備。從軟體上來看,三機有各自獨立的作業系統。這樣能達到完全的三機獨立。對于“2+1”系統,“2”分為 ......

    uj5u.com 2020-09-10 02:00:44 more
  • 如何從xshell上傳檔案到centos linux虛擬機里

    如何從xshell上傳檔案到centos linux虛擬機里及:虛擬機CentOs下執行 yum -y install lrzsz命令,出現錯誤:鏡像無法找到軟體包 前言 一、安裝lrzsz步驟 二、上傳檔案 三、遇到的問題及解決方案 總結 前言 提示:其實很簡單,往虛擬機上安裝一個上傳檔案的工具 ......

    uj5u.com 2020-09-10 02:00:47 more
  • 一、SQLMAP入門

    一、SQLMAP入門 1、判斷是否存在注入 sqlmap.py -u 網址/id=1 id=1不可缺少。當注入點后面的引數大于兩個時。需要加雙引號, sqlmap.py -u "網址/id=1&uid=1" 2、判斷文本中的請求是否存在注入 從文本中加載http請求,SQLMAP可以從一個文本檔案中 ......

    uj5u.com 2020-09-10 02:00:50 more
  • Metasploit 簡單使用教程

    metasploit 簡單使用教程 浩先生, 2020-08-28 16:18:25 分類專欄: kail 網路安全 linux 文章標簽: linux資訊安全 編輯 著作權 metasploit 使用教程 前言 一、Metasploit是什么? 二、準備作業 三、具體步驟 前言 Msfconsole ......

    uj5u.com 2020-09-10 02:00:53 more
  • 游戲逆向之驅動層與用戶層通訊

    驅動層代碼: #pragma once #include <ntifs.h> #define add_code CTL_CODE(FILE_DEVICE_UNKNOWN,0x800,METHOD_BUFFERED,FILE_ANY_ACCESS) /* 更多游戲逆向視頻www.yxfzedu.com ......

    uj5u.com 2020-09-10 02:00:56 more
  • 北斗電力時鐘(北斗授時服務器)讓網路資料更精準

    北斗電力時鐘(北斗授時服務器)讓網路資料更精準 北斗電力時鐘(北斗授時服務器)讓網路資料更精準 京準電子科技官微——ahjzsz 近幾年,資訊技術的得了快速發展,互聯網在逐漸普及,其在人們生活和生產中都得到了廣泛應用,并且取得了不錯的應用效果。計算機網路資訊在電力系統中的應用,一方面使電力系統的運行 ......

    uj5u.com 2020-09-10 02:01:03 more
  • 【CTF】CTFHub 技能樹 彩蛋 writeup

    ?碎碎念 CTFHub:https://www.ctfhub.com/ 筆者入門CTF時時剛開始刷的是bugku的舊平臺,后來才有了CTFHub。 感覺不論是網頁UI設計,還是題目質量,賽事跟蹤,工具軟體都做得很不錯。 而且因為獨到的金幣制度的確讓人有一種想去刷題賺金幣的感覺。 個人還是非常喜歡這個 ......

    uj5u.com 2020-09-10 02:04:05 more
  • 02windows基礎操作

    我學到了一下幾點 Windows系統目錄結構與滲透的作用 常見Windows的服務詳解 Windows埠詳解 常用的Windows注冊表詳解 hacker DOS命令詳解(net user / type /md /rd/ dir /cd /net use copy、批處理 等) 利用dos命令制作 ......

    uj5u.com 2020-09-10 02:04:18 more
  • 03.Linux基礎操作

    我學到了以下幾點 01Linux系統介紹02系統安裝,密碼啊破解03Linux常用命令04LAMP 01LINUX windows: win03 8 12 16 19 配置不繁瑣 Linux:redhat,centos(紅帽社區版),Ubuntu server,suse unix:金融機構,證券,銀 ......

    uj5u.com 2020-09-10 02:04:30 more
  • 05HTML

    01HTML介紹 02頭部標簽講解03基礎標簽講解04表單標簽講解 HTML前段語言 js1.了解代碼2.根據代碼 懂得挖掘漏洞 (POST注入/XSS漏洞上傳)3.黑帽seo 白帽seo 客戶網站被黑帽植入劫持代碼如何處理4.熟悉html表單 <html><head><title>TDK標題,描述 ......

    uj5u.com 2020-09-10 02:04:36 more
最新发布
  • 2023年最新微信小程式抓包教程

    01 開門見山 隔一個月發一篇文章,不過分。 首先回顧一下《微信系結手機號資料庫被脫庫事件》,我也是第一時間得知了這個訊息,然后跟蹤了整件事情的經過。下面是這起事件的相關截圖以及近日流出的一萬條資料樣本: 個人認為這件事也沒什么,還不如關注一下之前45億快遞資料查詢渠道疑似在近日復活的訊息。 訊息是 ......

    uj5u.com 2023-04-20 08:48:24 more
  • web3 產品介紹:metamask 錢包 使用最多的瀏覽器插件錢包

    Metamask錢包是一種基于區塊鏈技術的數字貨幣錢包,它允許用戶在安全、便捷的環境下管理自己的加密資產。Metamask錢包是以太坊生態系統中最流行的錢包之一,它具有易于使用、安全性高和功能強大等優點。 本文將詳細介紹Metamask錢包的功能和使用方法。 一、 Metamask錢包的功能 數字資 ......

    uj5u.com 2023-04-20 08:47:46 more
  • vulnhub_Earth

    前言 靶機地址->>>vulnhub_Earth 攻擊機ip:192.168.20.121 靶機ip:192.168.20.122 參考文章 https://www.cnblogs.com/Jing-X/archive/2022/04/03/16097695.html https://www.cnb ......

    uj5u.com 2023-04-20 07:46:20 more
  • 從4k到42k,軟體測驗工程師的漲薪史,給我看哭了

    清明節一過,盲猜大家已經無心上班,在數著日子準備過五一,但一想到銀行卡里的余額……瞬間心情就不美麗了。最近,2023年高校畢業生就業調查顯示,本科畢業月平均起薪為5825元。調查一出,便有很多同學表示自己又被平均了。看著這一資料,不免讓人想到前不久中國青年報的一項調查:近六成大學生認為畢業10年內會 ......

    uj5u.com 2023-04-20 07:44:00 more
  • 最新版本 Stable Diffusion 開源 AI 繪畫工具之中文自動提詞篇

    🎈 標簽生成器 由于輸入正向提示詞 prompt 和反向提示詞 negative prompt 都是使用英文,所以對學習母語的我們非常不友好 使用網址:https://tinygeeker.github.io/p/ai-prompt-generator 這個網址是為了讓大家在使用 AI 繪畫的時候 ......

    uj5u.com 2023-04-20 07:43:36 more
  • 漫談前端自動化測驗演進之路及測驗工具分析

    隨著前端技術的不斷發展和應用程式的日益復雜,前端自動化測驗也在不斷演進。隨著 Web 應用程式變得越來越復雜,自動化測驗的需求也越來越高。如今,自動化測驗已經成為 Web 應用程式開發程序中不可或缺的一部分,它們可以幫助開發人員更快地發現和修復錯誤,提高應用程式的性能和可靠性。 ......

    uj5u.com 2023-04-20 07:43:16 more
  • CANN開發實踐:4個DVPP記憶體問題的典型案例解讀

    摘要:由于DVPP媒體資料處理功能對存放輸入、輸出資料的記憶體有更高的要求(例如,記憶體首地址128位元組對齊),因此需呼叫專用的記憶體申請介面,那么本期就分享幾個關于DVPP記憶體問題的典型案例,并給出原因分析及解決方法。 本文分享自華為云社區《FAQ_DVPP記憶體問題案例》,作者:昇騰CANN。 DVPP ......

    uj5u.com 2023-04-20 07:43:03 more
  • msf學習

    msf學習 以kali自帶的msf為例 一、msf核心模塊與功能 msf模塊都放在/usr/share/metasploit-framework/modules目錄下 1、auxiliary 輔助模塊,輔助滲透(埠掃描、登錄密碼爆破、漏洞驗證等) 2、encoders 編碼器模塊,主要包含各種編碼 ......

    uj5u.com 2023-04-20 07:42:59 more
  • Halcon軟體安裝與界面簡介

    1. 下載Halcon17版本到到本地 2. 雙擊安裝包后 3. 步驟如下 1.2 Halcon軟體安裝 界面分為四大塊 1. Halcon的五個助手 1) 影像采集助手:與相機連接,設定相機引數,采集影像 2) 標定助手:九點標定或是其它的標定,生成標定檔案及內參外參,可以將像素單位轉換為長度單位 ......

    uj5u.com 2023-04-20 07:42:17 more
  • 在MacOS下使用Unity3D開發游戲

    第一次發博客,先發一下我的游戲開發環境吧。 去年2月份買了一臺MacBookPro2021 M1pro(以下簡稱mbp),這一年來一直在用mbp開發游戲。我大致分享一下我的開發工具以及使用體驗。 1、Unity 官網鏈接: https://unity.cn/releases 我一般使用的Apple ......

    uj5u.com 2023-04-20 07:40:19 more