前言

本文使用 cpu 的 tensorflow 2.4 完成遷移學(xué)習(xí)和模型微調(diào)，并使用訓(xùn)練好的模型完成貓狗圖片分類任務(wù)。

預(yù)訓(xùn)練模型在 NLP 中最常見(jiàn)的可能就是 BERT 了，在 CV 中我們此次用到了 MobileNetV2 ，它也是一個(gè)輕量化預(yù)訓(xùn)練模型，它已經(jīng)經(jīng)過(guò)大量的圖片分類任務(wù)的訓(xùn)練，里面保存了一個(gè)可以通用的去捕獲圖片特征的模型網(wǎng)絡(luò)結(jié)構(gòu)，其可以通用地提取出圖片的有意義特征。這些特征捕獲功能可以輕松遷移到其他圖片任務(wù)上幫助其完成特征提取工作。

本文的工作，就是在 MobileNetV2 基礎(chǔ)上加入我們自定義的若干網(wǎng)絡(luò)層，通過(guò)使用大量的貓狗數(shù)據(jù)先對(duì)新添加的分類器層進(jìn)行遷移學(xué)習(xí)，然后結(jié)合基礎(chǔ)模型的最后幾層與我們的自定義分類器一起進(jìn)行微調(diào)訓(xùn)練，就可以輕松獲得一個(gè)效果很好的貓狗圖片分類模型，而不必基于大量數(shù)據(jù)集從頭開(kāi)始訓(xùn)練一個(gè)大型模型，這樣會(huì)很耗時(shí)間。

實(shí)現(xiàn)過(guò)程

1. 獲取數(shù)據(jù)

首先我們要使用 tensorflow 的內(nèi)置函數(shù)，從網(wǎng)絡(luò)上下載貓狗圖片集，訓(xùn)練數(shù)據(jù)中的貓、狗圖片各 1000 張，驗(yàn)證數(shù)據(jù)中的貓、狗圖片各 500 張。每張圖片的大小都是（160, 160, 3），每個(gè) batch 中有 32 張圖片。

import matplotlib.pyplot as plt
import numpy as np
import os
import tensorflow as tf
URL = 'https://storage.googleapis.com/mledu-datasets/cats_and_dogs_filtered.zip'
BATCH_SIZE = 32
IMG_SIZE = (160, 160)
path_to_zip = tf.keras.utils.get_file('cats_and_dogs.zip', origin=URL, extract=True)
PATH = os.path.join(os.path.dirname(path_to_zip), 'cats_and_dogs_filtered')
train_dir = os.path.join(PATH, 'train')
validation_dir = os.path.join(PATH, 'validation')
train_dataset = tf.keras.utils.image_dataset_from_directory(train_dir, shuffle=True, batch_size=BATCH_SIZE, image_size=IMG_SIZE)
validation_dataset = tf.keras.utils.image_dataset_from_directory(validation_dir, shuffle=True, batch_size=BATCH_SIZE, image_size=IMG_SIZE)
class_names = train_dataset.class_names

在這里我們挑選了部分圖片和標(biāo)簽進(jìn)行展示。

plt.figure(figsize=(10, 10))
for images, labels in train_dataset.take(1):
    for i in range(3):
        ax = plt.subplot(3, 3, i + 1)
        plt.imshow(images[i].numpy().astype("uint8"))
        plt.title(class_names[labels[i]])
        plt.axis("off")     

2. 數(shù)據(jù)擴(kuò)充與數(shù)據(jù)縮放

（1）由于我們沒(méi)有測(cè)試集數(shù)據(jù)，使用 tf.data.experimental.cardinality 確定驗(yàn)證集中有多少個(gè) batch 的數(shù)據(jù)，然后將其中的 20% 的 batch 編程測(cè)試集。

val_batches = tf.data.experimental.cardinality(validation_dataset)
test_dataset = validation_dataset.take(val_batches // 5)
validation_dataset = validation_dataset.skip(val_batches // 5)

（2）為了保證在加載數(shù)據(jù)的時(shí)候不會(huì)出現(xiàn) I/O 不會(huì)阻塞，我們?cè)趶拇疟P加載完數(shù)據(jù)之后，使用 cache 會(huì)將數(shù)據(jù)保存在內(nèi)存中，確保在訓(xùn)練模型過(guò)程中數(shù)據(jù)的獲取不會(huì)成為訓(xùn)練速度的瓶頸。如果說(shuō)要保存的數(shù)據(jù)量太大，可以使用 cache 創(chuàng)建磁盤緩存提高數(shù)據(jù)的讀取效率。另外我們還使用 prefetch 在訓(xùn)練過(guò)程中可以并行執(zhí)行數(shù)據(jù)的預(yù)獲取。

AUTOTUNE = tf.data.AUTOTUNE
train_dataset = train_dataset.prefetch(buffer_size=AUTOTUNE)
validation_dataset = validation_dataset.prefetch(buffer_size=AUTOTUNE)
test_dataset = test_dataset.prefetch(buffer_size=AUTOTUNE)

（3）因?yàn)槲覀儧](méi)有大量的圖片數(shù)據(jù)，我們將現(xiàn)有的圖片數(shù)據(jù)進(jìn)行旋轉(zhuǎn)或者翻轉(zhuǎn)操作可以增加樣本的多樣性，這樣也有助于減少過(guò)擬合現(xiàn)象。RandomFlip 函數(shù)將根據(jù) mode 屬性將圖片進(jìn)行水平或垂直翻轉(zhuǎn)圖像。"可選的有 "horizontal" 、 "vertical" 或 "horizontal_and_vertical" 。"horizontal" 是左右翻轉(zhuǎn)， "vertical" 是上下翻轉(zhuǎn)。RandomRotation 函數(shù)通過(guò)設(shè)置 factor 來(lái)講圖片進(jìn)行旋轉(zhuǎn)，假如 factor=0.2 會(huì)將圖片在 [-20% * 2pi, 20% * 2pi] 范圍內(nèi)隨機(jī)旋轉(zhuǎn)。

data_augmentation = tf.keras.Sequential([
  tf.keras.layers.RandomFlip('horizontal'),
  tf.keras.layers.RandomRotation(0.1),
])	

我們這里將隨便使用一張圖片，使用數(shù)據(jù)增強(qiáng)的方法對(duì)其進(jìn)行變化，第一張圖是原圖，其他的都是進(jìn)行了變化的圖片

for image, _ in train_dataset.take(1):
    plt.figure(figsize=(10, 10))
    first_image = image[0]
    for i in range(3):
        if  i==0:
            ax = plt.subplot(3, 3, i + 1)
            plt.imshow(first_image/ 255)
            plt.axis('off')
        else:
            ax = plt.subplot(3, 3, i + 1)
            augmented_image = data_augmentation(tf.expand_dims(first_image, 0))
            plt.imshow(augmented_image[0] / 255)
            plt.axis('off')

（4）由于 MobileNetV2 模型的輸入值范圍是在 [-1, 1] 范圍內(nèi)，但此時(shí)我們的圖片數(shù)據(jù)中的像素值處于 [0, 255] 范圍內(nèi)，所以要重新縮放這些像素值， Rescaling 函數(shù)可以實(shí)現(xiàn)該操作。如果將 [0, 255] 范圍的輸入縮放到 [0, 1] 的范圍，我們可以通過(guò)設(shè)置參數(shù) scale=1./255 來(lái)實(shí)現(xiàn)。如果將 [0, 255] 范圍內(nèi)的輸入縮放到 [-1, 1] 范圍內(nèi)，我們可以通過(guò)設(shè)置參數(shù) scale=1./127.5, offset=-1 來(lái)實(shí)現(xiàn)。

rescale = tf.keras.layers.Rescaling(1./127.5, offset=-1)	

3. 遷移學(xué)習(xí)

（1）我們將 MobileNet V2 模型當(dāng)做一個(gè)基礎(chǔ)模型，此模型已基于 ImageNet 數(shù)據(jù)集進(jìn)行完美的預(yù)訓(xùn)練，一般來(lái)說(shuō)最后一層就是一個(gè)分類器，我們選擇將在 MobileNet V2 的倒數(shù)第二個(gè)網(wǎng)絡(luò)層上搭建自己的分類器，與最后一層相比倒數(shù)第二層能夠保留更豐富的圖片特征。在實(shí)際操作中我們實(shí)例化 MobileNetV2 模型，通過(guò)指定 include_top=False 參數(shù)，可以加載不包括最頂層的整個(gè)預(yù)訓(xùn)練網(wǎng)絡(luò)結(jié)果。 該模型的作用就是將（160,160,3）大小的圖片轉(zhuǎn)換為（5,5,1280）的特征輸出。

（2）第一層是將我們的訓(xùn)練數(shù)據(jù)都進(jìn)行數(shù)據(jù)增強(qiáng)操作，也就是隨機(jī)的翻轉(zhuǎn)和旋轉(zhuǎn)。

（3）第二層是接收輸入為 （160,160,3）大小的圖片的輸入層。

（4）第三層是我們直接拿來(lái)用的 MobileNetV2 ，因?yàn)槲覀円苯邮褂没A(chǔ)模型的圖片特征提取能力，所以為了在訓(xùn)練過(guò)程中其權(quán)重不發(fā)生變化，我們將基礎(chǔ)模型中的權(quán)重參數(shù)都凍結(jié)。

（5）第四層是一個(gè)池化層，可以將每個(gè) batch 從 (32,5,5,1280) 壓縮為 (32,1280) 大小的輸出。

（6）第五層是 Dropout ，防止過(guò)擬合。

（7）第六層是對(duì)該圖片一個(gè)預(yù)測(cè)值，也就是 logit 。如果是正數(shù)預(yù)測(cè)標(biāo)簽 1 ，如果是負(fù)數(shù)預(yù)測(cè)標(biāo)簽 0 。

（8）我們從模型的 summary 中可以看到，此時(shí)我們的模型中的 2259265 個(gè)參數(shù)被凍結(jié)，只有 1281 個(gè)參數(shù)是可以訓(xùn)練的，它們是分屬兩個(gè)變量的可訓(xùn)練參數(shù)，即最后一個(gè)全連接層即權(quán)重 1280 個(gè)可訓(xùn)練參數(shù)和偏差 1 個(gè)可訓(xùn)練參數(shù)。

IMG_SHAPE = IMG_SIZE + (3,)
inputs = tf.keras.Input(shape=(160, 160, 3))
x = data_augmentation(inputs)
x = tf.keras.applications.mobilenet_v2.preprocess_input(x)
base_model = tf.keras.applications.MobileNetV2(input_shape=IMG_SHAPE, include_top=False, weights='imagenet')
base_model.trainable = False
x = base_model(x, training=False)
x = tf.keras.layers.GlobalAveragePooling2D()(x)
x = tf.keras.layers.Dropout(0.2)(x)
outputs = tf.keras.layers.Dense(1)(x)
model = tf.keras.Model(inputs, outputs)
model.summary()

模型結(jié)構(gòu)如下：

Model: "model_2"
_________________________________________________________________
 Layer (type)                Output Shape              Param #   
=================================================================
 input_8 (InputLayer)        [(None, 160, 160, 3)]     0         
 sequential_2 (Sequential)   (None, 160, 160, 3)       0         
 tf.math.truediv_3 (TFOpLamb  (None, 160, 160, 3)      0         
 da)                                                             
 tf.math.subtract_3 (TFOpLam  (None, 160, 160, 3)      0         
 bda)                                                            
 mobilenetv2_1.00_160 (Funct  (None, 5, 5, 1280)       2257984   
 ional)                                                          
 global_average_pooling2d_3   (None, 1280)             0         
 (GlobalAveragePooling2D)                                        
 dropout_2 (Dropout)         (None, 1280)              0         
 dense_2 (Dense)             (None, 1)                 1281      
=================================================================
Total params: 2,259,265
Trainable params: 1,281
Non-trainable params: 2,257,984

可以看到模型中的可訓(xùn)練的 2 個(gè)變量：

model.trainable_variables

結(jié)果如下：

[<tf.Variable 'dense_2/kernel:0' shape=(1280, 1) dtype=float32, numpy=
 array([[ 0.08899798],
        [-0.06681276],
        [ 0.00906871],
        ...,
        [-0.00114891],
        [-0.01134416],
        [-0.02000826]], dtype=float32)>,
 <tf.Variable 'dense_2/bias:0' shape=(1,) dtype=float32, numpy=array([0.03746362], dtype=float32)>]    

（9）選擇 Adam 優(yōu)化器，將我們的學(xué)習(xí)率設(shè)置為 0.0003 。選擇 BinaryCrossentropy 作為損失函數(shù)。選擇常規(guī)的 accuracy 作為評(píng)估指標(biāo)。此時(shí)我們先對(duì)基礎(chǔ)模型訓(xùn)練 10 個(gè) epoch 。

lr = 0.0003
initial_epochs = 10
model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=lr),
              loss=tf.keras.losses.BinaryCrossentropy(from_logits=True),
              metrics=['accuracy'])
history = model.fit(train_dataset, validation_data=test_dataset, epochs=initial_epochs)

訓(xùn)練過(guò)程如下：

Epoch 1/10
63/63 [==============================] - 24s 352ms/step - loss: 0.5517 - accuracy: 0.6835 - val_loss: 0.2720 - val_accuracy: 0.8958
Epoch 2/10
63/63 [==============================] - 21s 327ms/step - loss: 0.2792 - accuracy: 0.8865 - val_loss: 0.1499 - val_accuracy: 0.9531
...
Epoch 9/10
63/63 [==============================] - 20s 321ms/step - loss: 0.1075 - accuracy: 0.9530 - val_loss: 0.0766 - val_accuracy: 0.9740
Epoch 10/10
63/63 [==============================] - 21s 329ms/step - loss: 0.1040 - accuracy: 0.9560 - val_loss: 0.0742 - val_accuracy: 0.9740

（10）使用測(cè)試數(shù)據(jù)對(duì)模型進(jìn)行評(píng)估。

model.evaluate(validation_dataset)

結(jié)果如下：

loss: 0.0664 - accuracy: 0.9765

4. 微調(diào)

（1）在上面的操作中，我們僅僅在 MobileNetV2 基礎(chǔ)模型的頂部添加了一層池化層、一層 Dropout、一層全連接層作為我們的自定義分類器。預(yù)訓(xùn)練模型 MobileNetV2 中權(quán)重在訓(xùn)練過(guò)程中未曾發(fā)生過(guò)更新。

（2）我們還有一種被稱為微調(diào)的方法，可以在訓(xùn)練基礎(chǔ)模型權(quán)重的同時(shí)，同時(shí)訓(xùn)練我們上面自定義添加的用于分類的分類器。這個(gè)微調(diào)的訓(xùn)練過(guò)程可以將基礎(chǔ)模型的通用的圖片特征提取能力調(diào)整為專門提取本任務(wù)中貓狗數(shù)據(jù)集特征的能力。這個(gè)微調(diào)的操作只能在我們進(jìn)行了上面的遷移學(xué)習(xí)操作之后才能進(jìn)行，否則如果一開(kāi)始直接將基礎(chǔ)模型和我們自定義的若干層分類器一起進(jìn)行訓(xùn)練，則會(huì)由于隨機(jī)初始化的分類器導(dǎo)致整個(gè)模型的更新梯度太大，從而使得基礎(chǔ)模型喪失了其預(yù)訓(xùn)練的有效能力。其次在微調(diào)過(guò)程中我們應(yīng)該選擇性地去微調(diào)少量頂部的網(wǎng)絡(luò)層而不是整個(gè) MobileNet 模型，因?yàn)樵诰矸e神經(jīng)網(wǎng)絡(luò)中，低層的網(wǎng)絡(luò)層一般捕獲到的是通用的圖片特征，這個(gè)能力可以泛化應(yīng)用到幾乎所有類型的圖片，但是越往頂部的網(wǎng)絡(luò)層，越來(lái)越聚焦于捕獲訓(xùn)練時(shí)所用到的訓(xùn)練數(shù)據(jù)的特征，而微調(diào)的目標(biāo)正是是讓這個(gè)模型更加適用于所用的專門的數(shù)據(jù)集，也就是本次進(jìn)行的貓狗圖片。

（3）MobileNetV2 模型一共有 154 層結(jié)構(gòu)，我們將模型的前 100 層的參數(shù)進(jìn)行凍結(jié)，對(duì)頂部的 54 層網(wǎng)絡(luò)結(jié)構(gòu)中的參數(shù)與我們自定義的分類器的若干層一起進(jìn)行訓(xùn)練，我們通過(guò)打印模型的 summary 可以看到，此時(shí)共有 54 個(gè)可以訓(xùn)練的變量，這些變量中共有可訓(xùn)練的參數(shù) 1862721 個(gè)。

base_model.trainable = True
fine_tune_at = 100
for layer in base_model.layers[:fine_tune_at]:
    layer.trainable = False
print("%d trainable variables "%len(model.trainable_variables))
model.summary()

結(jié)果如下：

56 trainable variables 
Model: "model_2"
_________________________________________________________________
 Layer (type)                Output Shape              Param #   
=================================================================
 input_8 (InputLayer)        [(None, 160, 160, 3)]     0         
 sequential_2 (Sequential)   (None, 160, 160, 3)       0         
 tf.math.truediv_3 (TFOpLamb  (None, 160, 160, 3)      0         
 da)                                                             
 tf.math.subtract_3 (TFOpLam  (None, 160, 160, 3)      0         
 bda)                                                            
 mobilenetv2_1.00_160 (Funct  (None, 5, 5, 1280)       2257984   
 ional)                                                          
 global_average_pooling2d_3   (None, 1280)             0         
 (GlobalAveragePooling2D)                                        
 dropout_2 (Dropout)         (None, 1280)              0         
 dense_2 (Dense)             (None, 1)                 1281      
=================================================================
Total params: 2,259,265
Trainable params: 1,862,721
Non-trainable params: 396,544

（4）之前我們對(duì)基礎(chǔ)模型訓(xùn)練了 10 個(gè) epoch ，現(xiàn)在我們進(jìn)行微調(diào)過(guò)程中再對(duì)模型訓(xùn)練 10 個(gè) epoch ，且我們將從上面訓(xùn)練結(jié)束的 epoch 開(kāi)始恢復(fù)并進(jìn)行現(xiàn)在的訓(xùn)練過(guò)程。

（5）這里我們改用 RMSprop 優(yōu)化器，且因?yàn)榇藭r(shí)我們是要對(duì)整體模型進(jìn)行微調(diào)，所以設(shè)置的學(xué)習(xí)率比之前降低 10 倍，否則如果較大會(huì)很快產(chǎn)生過(guò)擬合。

fine_tune_epochs = 10
total_epochs =  initial_epochs + fine_tune_epochs
model.compile(loss=tf.keras.losses.BinaryCrossentropy(from_logits=True),
              optimizer = tf.keras.optimizers.RMSprop(learning_rate=lr/10),
              metrics=['accuracy'])
history_fine = model.fit(train_dataset,  epochs=total_epochs,  initial_epoch=history.epoch[-1],  validation_data=validation_dataset)	

訓(xùn)練結(jié)果輸出：

Epoch 10/20
63/63 [==============================] - 39s 561ms/step - loss: 0.1073 - accuracy: 0.9570 - val_loss: 0.1017 - val_accuracy: 0.9592
Epoch 11/20
63/63 [==============================] - 34s 538ms/step - loss: 0.0688 - accuracy: 0.9725 - val_loss: 0.0448 - val_accuracy: 0.9827
...
Epoch 19/20
63/63 [==============================] - 34s 537ms/step - loss: 0.0244 - accuracy: 0.9900 - val_loss: 0.0709 - val_accuracy: 0.9777
Epoch 20/20
63/63 [==============================] - 33s 528ms/step - loss: 0.0220 - accuracy: 0.9905 - val_loss: 0.0566 - val_accuracy: 0.9851

（6）使用測(cè)試數(shù)據(jù)對(duì)模型進(jìn)行評(píng)估。

model.evaluate(test_dataset)

結(jié)果輸出：

loss: 0.0544 - accuracy: 0.9792

之前模型最后的驗(yàn)證準(zhǔn)確率為 0.9740 ，測(cè)試準(zhǔn)確率為 0.9765 ，而經(jīng)過(guò)微調(diào)的模型，最后餓驗(yàn)證準(zhǔn)確率為 0.9851 ，測(cè)試準(zhǔn)確率為 0.9792 ，都有所提升。

5. 預(yù)測(cè)

我們隨機(jī)挑選一個(gè) batch 進(jìn)行預(yù)測(cè)，并將圖片與預(yù)測(cè)標(biāo)簽進(jìn)行顯示，結(jié)果表明預(yù)測(cè)全都正確。

image_batch, label_batch = test_dataset.as_numpy_iterator().next()
predictions = model.predict_on_batch(image_batch).flatten()
predictions = tf.nn.sigmoid(predictions)
predictions = tf.where(predictions < 0.5, 0, 1)
plt.figure(figsize=(10, 10))
for i in range(9):
    ax = plt.subplot(3, 3, i + 1)
    plt.imshow(image_batch[i].astype("uint8"))
    plt.title(class_names[predictions[i]])
    plt.axis("off")

以上就是深度學(xué)習(xí)Tensorflow 2.4 完成遷移學(xué)習(xí)和模型微調(diào)的詳細(xì)內(nèi)容，更多關(guān)于Tensorflow 遷移學(xué)習(xí)模型微調(diào)的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！