在解決一個任務(wù)時(shí)，我會選擇加載預(yù)訓(xùn)練模型并逐步fine-tune。比如，分類任務(wù)中，優(yōu)異的深度學(xué)習(xí)網(wǎng)絡(luò)有很多。

ResNet, VGG, Xception等等... 并且這些模型參數(shù)已經(jīng)在imagenet數(shù)據(jù)集中訓(xùn)練的很好了，可以直接拿過來用。

根據(jù)自己的任務(wù)，訓(xùn)練一下最后的分類層即可得到比較好的結(jié)果。此時(shí)，就需要“凍結(jié)”預(yù)訓(xùn)練模型的所有層，即這些層的權(quán)重永不會更新。

以Xception為例：

加載預(yù)訓(xùn)練模型：

from tensorflow.python.keras.applications import Xception
model = Sequential()
model.add(Xception(include_top=False, pooling='avg', weights='imagenet'))
model.add(Dense(NUM_CLASS, activation='softmax'))

include_top = False : 不包含頂層的3個全鏈接網(wǎng)絡(luò)

weights : 加載預(yù)訓(xùn)練權(quán)重

隨后，根據(jù)自己的分類任務(wù)加一層網(wǎng)絡(luò)即可。

網(wǎng)絡(luò)具體參數(shù)：

model.summary

得到兩個網(wǎng)絡(luò)層，第一層是xception層，第二層為分類層。

由于未凍結(jié)任何層，trainable params為：20, 811, 050

凍結(jié)網(wǎng)絡(luò)層：

由于第一層為xception，不想更新xception層的參數(shù)，可以加以下代碼：

model.layers[0].trainable = False

凍結(jié)預(yù)訓(xùn)練模型中的層

如果想凍結(jié)xception中的部分層，可以如下操作：

from tensorflow.python.keras.applications import Xception
model = Sequential()
model.add(Xception(include_top=False, pooling='avg', weights='imagenet'))
model.add(Dense(NUM_CLASS, activation='softmax'))
for i, layer in enumerate(model.layers[0].layers):
 if i > 115:
 layer.trainable = True
 else:
 layer.trainable = False
 print(i, layer.name, layer.trainable)

加載所有預(yù)訓(xùn)練模型的層

若想把xeption的所有層應(yīng)用在訓(xùn)練自己的數(shù)據(jù)，并改變分類數(shù)。可以如下操作：

model = Sequential()
model.add(Xception(include_top=True, weights=None, classes=NUM_CLASS))

* 如果想指定classes，有兩個條件：include_top：True， weights：None。否則無法指定classes

補(bǔ)充知識：如何利用預(yù)訓(xùn)練模型進(jìn)行模型微調(diào)（如凍結(jié)某些層，不同層設(shè)置不同學(xué)習(xí)率等）

由于預(yù)訓(xùn)練模型權(quán)重和我們要訓(xùn)練的數(shù)據(jù)集存在一定的差異，且需要訓(xùn)練的數(shù)據(jù)集有大有小，所以進(jìn)行模型微調(diào)、設(shè)置不同學(xué)習(xí)率就變得比較重要，下面主要分四種情況進(jìn)行討論，錯誤之處或者不足之處還請大佬們指正。

（1）待訓(xùn)練數(shù)據(jù)集較小，與預(yù)訓(xùn)練模型數(shù)據(jù)集相似度較高時(shí)。例如待訓(xùn)練數(shù)據(jù)集中數(shù)據(jù)存在于預(yù)訓(xùn)練模型中時(shí)，不需要重新訓(xùn)練模型，只需要修改最后一層輸出層即可。

（2）待訓(xùn)練數(shù)據(jù)集較小，與預(yù)訓(xùn)練模型數(shù)據(jù)集相似度較小時(shí)?？梢詢鼋Y(jié)模型的前k層，重新模型的后n-k層。凍結(jié)模型的前k層，用于彌補(bǔ)數(shù)據(jù)集較小的問題。

（3）待訓(xùn)練數(shù)據(jù)集較大，與預(yù)訓(xùn)練模型數(shù)據(jù)集相似度較大時(shí)。采用預(yù)訓(xùn)練模型會非常有效，保持模型結(jié)構(gòu)不變和初始權(quán)重不變，對模型重新訓(xùn)練

（4）待訓(xùn)練數(shù)據(jù)集較大，與預(yù)訓(xùn)練模型數(shù)據(jù)集相似度較小時(shí)。采用預(yù)訓(xùn)練模型不會有太大的效果，可以使用預(yù)訓(xùn)練模型或者不使用預(yù)訓(xùn)練模型，然后進(jìn)行重新訓(xùn)練。

以上這篇Keras 實(shí)現(xiàn)加載預(yù)訓(xùn)練模型并凍結(jié)網(wǎng)絡(luò)的層就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

最新評論