1
本文作者: AI研習社 | 2017-05-15 17:15 |
雷鋒網(wǎng)按:本文作者徐凱文,原文載于作者個人博客,雷鋒網(wǎng)已獲授權(quán)。
在知乎上看到一段介紹卷積神經(jīng)網(wǎng)絡(luò)的文章,感覺講的特別直觀明了,我整理了一下。首先介紹原理部分。
通過一個圖像分類問題介紹卷積神經(jīng)網(wǎng)絡(luò)是如何工作的。下面是卷積神經(jīng)網(wǎng)絡(luò)判斷一個圖片是否包含“兒童”的過程,包括四個步驟:
● 圖像輸入(InputImage)
● 卷積(Convolution)
● 最大池化(MaxPooling)
● 全連接神經(jīng)網(wǎng)絡(luò)(Fully-ConnectedNeural Network)計算。
首先將圖片分割成如下圖的重疊的獨立小塊;下圖中,這張照片被分割成了77張大小相同的小圖片。
接下來將每一個獨立小塊輸入小的神經(jīng)網(wǎng)絡(luò);這個小的神經(jīng)網(wǎng)絡(luò)已經(jīng)被訓練用來判斷一個圖片是否屬于“兒童”類別,它輸出的是一個特征數(shù)組。
標準的數(shù)碼相機有紅、綠、藍三個通道(Channels),每一種顏色的像素值在0-255之間,構(gòu)成三個堆疊的二維矩陣;灰度圖像則只有一個通道,可以用一個二維矩陣來表示。
將所有的獨立小塊輸入小的神經(jīng)網(wǎng)絡(luò)后,再將每一個輸出的特征數(shù)組按照第一步時77個獨立小塊的相對位置做排布,得到一個新數(shù)組。
第二步中,這個小的神經(jīng)網(wǎng)絡(luò)對這77張大小相同的小圖片都進行同樣的計算,也稱權(quán)重共享(SharedWeights)。這樣做是因為,第一,對圖像等數(shù)組數(shù)據(jù)來說,局部數(shù)組的值經(jīng)常是高度相關(guān)的,可以形成容易被探測到的獨特的局部特征;第二,圖像和其它信號的局部統(tǒng)計特征與其位置是不太相關(guān)的,如果特征圖能在圖片的一個部分出現(xiàn),也能出現(xiàn)在任何地方。所以不同位置的單元共享同樣的權(quán)重,并在數(shù)組的不同部分探測相同的模式。數(shù)學上,這種由一個特征圖執(zhí)行的過濾操作是一個離散的卷積,卷積神經(jīng)網(wǎng)絡(luò)由此得名。
卷積步驟完成后,再使用MaxPooling算法來縮減像素采樣數(shù)組,按照2×2來分割特征矩陣,分出的每一個網(wǎng)格中只保留最大值數(shù)組,丟棄其它數(shù)組,得到最大池化數(shù)組(Max-PooledArray)。
接下來將最大池化數(shù)組作為另一個神經(jīng)網(wǎng)絡(luò)的輸入,這個全連接神經(jīng)網(wǎng)絡(luò)會最終計算出此圖是否符合預(yù)期的判斷。
在實際應(yīng)用時,卷積、最大池化和全連接神經(jīng)網(wǎng)絡(luò)計算,這幾步中的每一步都可以多次重復(fù)進行,總思路是將大圖片不斷壓縮,直到輸出單一的值。使用更多卷積步驟,神經(jīng)網(wǎng)絡(luò)就可以處理和學習更多的特征。
下面是代碼,添加了詳細注釋:
from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)# 讀取圖片數(shù)據(jù)集
sess = tf.InteractiveSession()# 創(chuàng)建session
# 一,函數(shù)聲明部分
def weight_variable(shape):
# 正態(tài)分布,標準差為0.1,默認最大為1,最小為-1,均值為0
initial = tf.truncated_normal(shape, stddev=0.1)
return tf.Variable(initial)
def bias_variable(shape):
# 創(chuàng)建一個結(jié)構(gòu)為shape矩陣也可以說是數(shù)組shape聲明其行列,初始化所有值為0.1
initial = tf.constant(0.1, shape=shape)
return tf.Variable(initial)
def conv2d(x, W):
# 卷積遍歷各方向步數(shù)為1,SAME:邊緣外自動補0,遍歷相乘
return tf.nn.conv2d(x, W, strides=[1, 1, 1, 1], padding='SAME')
def max_pool_2x2(x):
# 池化卷積結(jié)果(conv2d)池化層采用kernel大小為2*2,步數(shù)也為2,周圍補0,取最大值。數(shù)據(jù)量縮小了4倍
return tf.nn.max_pool(x, ksize=[1, 2, 2, 1],strides=[1, 2, 2, 1], padding='SAME')
# 二,定義輸入輸出結(jié)構(gòu)
# 聲明一個占位符,None表示輸入圖片的數(shù)量不定,28*28圖片分辨率
xs = tf.placeholder(tf.float32, [None, 28*28])
# 類別是0-9總共10個類別,對應(yīng)輸出分類結(jié)果
ys = tf.placeholder(tf.float32, [None, 10])
keep_prob = tf.placeholder(tf.float32)
# x_image又把xs reshape成了28*28*1的形狀,因為是灰色圖片,所以通道是1.作為訓練時的input,-1代表圖片數(shù)量不定
x_image = tf.reshape(xs, [-1, 28, 28, 1])
# 三,搭建網(wǎng)絡(luò),定義算法公式,也就是forward時的計算
## 第一層卷積操作 ##
# 第一二參數(shù)值得卷積核尺寸大小,即patch,第三個參數(shù)是圖像通道數(shù),第四個參數(shù)是卷積核的數(shù)目,代表會出現(xiàn)多少個卷積特征圖像;
W_conv1 = weight_variable([5, 5, 1, 32])
# 對于每一個卷積核都有一個對應(yīng)的偏置量。
b_conv1 = bias_variable([32])
# 圖片乘以卷積核,并加上偏執(zhí)量,卷積結(jié)果28x28x32
h_conv1 = tf.nn.relu(conv2d(x_image, W_conv1) + b_conv1)
# 池化結(jié)果14x14x32 卷積結(jié)果乘以池化卷積核
h_pool1 = max_pool_2x2(h_conv1)
## 第二層卷積操作 ##
# 32通道卷積,卷積出64個特征
w_conv2 = weight_variable([5,5,32,64])
# 64個偏執(zhí)數(shù)據(jù)
b_conv2 = bias_variable([64])
# 注意h_pool1是上一層的池化結(jié)果,#卷積結(jié)果14x14x64
h_conv2 = tf.nn.relu(conv2d(h_pool1,w_conv2)+b_conv2)
# 池化結(jié)果7x7x64
h_pool2 = max_pool_2x2(h_conv2)
# 原圖像尺寸28*28,第一輪圖像縮小為14*14,共有32張,第二輪后圖像縮小為7*7,共有64張
## 第三層全連接操作 ##
# 二維張量,第一個參數(shù)7*7*64的patch,也可以認為是只有一行7*7*64個數(shù)據(jù)的卷積,第二個參數(shù)代表卷積個數(shù)共1024個
W_fc1 = weight_variable([7*7*64, 1024])
# 1024個偏執(zhí)數(shù)據(jù)
b_fc1 = bias_variable([1024])
# 將第二層卷積池化結(jié)果reshape成只有一行7*7*64個數(shù)據(jù)# [n_samples, 7, 7, 64] ->> [n_samples, 7*7*64]
h_pool2_flat = tf.reshape(h_pool2, [-1, 7*7*64])
# 卷積操作,結(jié)果是1*1*1024,單行乘以單列等于1*1矩陣,matmul實現(xiàn)最基本的矩陣相乘,不同于tf.nn.conv2d的遍歷相乘,自動認為是前行向量后列向量
h_fc1 = tf.nn.relu(tf.matmul(h_pool2_flat, W_fc1) + b_fc1)
# dropout操作,減少過擬合,其實就是降低上一層某些輸入的權(quán)重scale,甚至置為0,升高某些輸入的權(quán)值,甚至置為2,防止評測曲線出現(xiàn)震蕩,個人覺得樣本較少時很必要
# 使用占位符,由dropout自動確定scale,也可以自定義,比如0.5,根據(jù)tensorflow文檔可知,程序中真實使用的值為1/0.5=2,也就是某些輸入乘以2,同時某些輸入乘以0
keep_prob = tf.placeholder(tf.float32)
h_fc1_drop = tf.nn.dropout(f_fc1,keep_prob) #對卷積結(jié)果執(zhí)行dropout操作
## 第四層輸出操作 ##
# 二維張量,1*1024矩陣卷積,共10個卷積,對應(yīng)我們開始的ys長度為10
W_fc2 = weight_variable([1024, 10])
b_fc2 = bias_variable([10])
# 最后的分類,結(jié)果為1*1*10 softmax和sigmoid都是基于logistic分類算法,一個是多分類一個是二分類
y_conv=tf.nn.softmax(tf.matmul(h_fc1_drop, W_fc2) + b_fc2)
# 四,定義loss(最小誤差概率),選定優(yōu)化優(yōu)化loss,
cross_entropy = -tf.reduce_sum(ys * tf.log(y_conv)) # 定義交叉熵為loss函數(shù)
train_step = tf.train.DradientDescentOptimizer(0.5).minimize(cross_entropy) # 調(diào)用優(yōu)化器優(yōu)化,其實就是通過喂數(shù)據(jù)爭取cross_entropy最小化
# 五,開始數(shù)據(jù)訓練以及評測
correct_prediction = tf.equal(tf.argmax(y_conv,1), tf.argmax(ys,1))
accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
tf.global_variables_initializer().run()
for i in range(20000):
batch = mnist.train.next_batch(50)
if i%100 == 0:
train_accuracy = accuracy.eval(feed_dict={x:batch[0], ys: batch[1], keep_prob: 1.0})
print("step %d, training accuracy %g"%(i, train_accuracy))
train_step.run(feed_dict={x: batch[0], ys: batch[1], keep_prob: 0.5})
print("test accuracy %g"%accuracy.eval(feed_dict={x: mnist.test.images, ys: mnist.test.labels, keep_prob: 1.0}))
———————————————————————————————————————
人工智能之神經(jīng)網(wǎng)絡(luò)特訓班
20年清華大學神經(jīng)網(wǎng)絡(luò)授課導(dǎo)師,帶你系統(tǒng)學習人工智能之神經(jīng)網(wǎng)絡(luò)!
一站式深入了解深度學習的發(fā)展現(xiàn)狀、基本原理和主要方法。
課程鏈接:http://www.mooc.ai/course/65
雷鋒網(wǎng)(公眾號:雷鋒網(wǎng))相關(guān)閱讀:
CNN 在基于弱監(jiān)督學習的圖像分割中的應(yīng)用
三年來,CNN在圖像分割領(lǐng)域經(jīng)歷了怎樣的技術(shù)變革?
雷峰網(wǎng)版權(quán)文章,未經(jīng)授權(quán)禁止轉(zhuǎn)載。詳情見轉(zhuǎn)載須知。