Winograd 快速卷積算法原理

Terry Winograd

Winograd [1]于1980 年提出了有限脈沖響應(yīng)(finite impulse response愚战,F(xiàn)IR)濾波的最小濾波算法最小濾波算法[2] 。該算法指出,由r 拍的FIR濾波器生成m 個輸出娄柳,即F(m,r),需要的最少乘法數(shù)量μ(F(m,r)) = m + r - 1 糜工。以F(2,3) 為例捕传,最小濾波算法:


涉及到的乘法數(shù)量為μ(F(2,3))= 2 +3 -1 = 4 ,從6 次降低到了4 次债鸡。

2015 年江滨,Winograd 最小濾波算法初次被應(yīng)用在CNN 中[3],利用減少的乘法次數(shù)提升卷積算子性能厌均。如果用矩陣的形式表示一維Winograd 最小濾波算法唬滑,則可以得到:
Y = A^T[(Gg)⊙(B^Td)]
其中,g 為濾波器向量棺弊,d 為輸入數(shù)據(jù)向量晶密,Y 為輸出數(shù)據(jù)向量,G 表示濾波器變換矩陣模她,B^T 表示數(shù)據(jù)變換矩陣稻艰,⊙表示矩陣的對應(yīng)位相乘(Hadamard 積),A^T 表示輸出變換矩陣侈净。

由此擴展到二維Winograd最小濾波算法可以得到:
Y = A^T[(GgG^T)⊙(B^TdB)]A
其中尊勿,g為濾波器矩陣僧凤,d為輸入輸入數(shù)據(jù)塊。通過嵌套一維最小濾波算法F(m,r)F(n,s) 元扔,可以得到二維的最小濾波算法μ(F(m × n, r × s)) = μ(Fm,r)) μ(Fn,s)) = (m+r-1)(n+s-1)其中躯保,m×n為輸出大小,r× s為濾波器大小摇展。二維最小濾波算法所需乘法數(shù)為(m + r - 1)(n+s-1) 吻氧,而原始卷積算法需要乘法數(shù)為m × n × r × s。對于F(2 × 2,3 × 3)而言咏连,乘法計算次數(shù)從36降低到了16盯孙,減少了55.6%。

F(m × m, r × r)應(yīng)用到卷積計算祟滴,對于二維卷積算子振惰,需要先將卷積輸入劃分為相互重疊的大小為(m+r-1)×(m+r-1) 的切片,切片之間有r-1的重疊部分垄懂。對于每個通道骑晶,可以分成P = [H/m][W/m]個切片,然后通過F(m × m, r × r)對每個切片分別進行計算草慧。

將每個切片標記為(\tilde{x},\tilde{y})桶蛔,對應(yīng)第i個輸入和第k個卷積核的卷積計算結(jié)果為:

Y_{i,k,\tilde{x},\tilde{y}} = \sum_{c=1}^{C} D_{i,c,\tilde{x},\tilde{y}}*G_{k,c} \\ = \sum_{c=1}^{C} A^T[U_{k,c}⊙V_{c,i,\tilde{x},\tilde{y}}]A \\ =A^T[\sum_{c=1}^{C} U_{k,c}⊙V_{c,i,\tilde{x},\tilde{y}}]A

其中:U=GgG^TV=B^TdB漫谷。

Winograd卷積的偽代碼
Winograd 卷積的四個階段

根據(jù)二維最小濾波算法的矩陣形式仔雷,可以將Winograd卷積分為四個分離的階段:輸入變換(input transformation,ITrans)舔示、卷積核變換(kernel transformation碟婆,KTrans)、對應(yīng)位相乘(element-wise matrix multiplication惕稻,EWMM)和輸出變換(output transformation竖共,OTrans),如圖1所示俺祠。


  1. Terry Winograd, Professor Emeritus of Computer Science, Stanford University https://hci.stanford.edu/winograd/ ?

  2. WINOGRAD S. Arithmetic complexity of computations [M]. Philadelphia: SIAM, 1980. ?

  3. LAVIN A, GRAY S. Fast algorithms for convolutional neural networks[C]//Proceedings of the 2016 IEEE Conferenceon Computer Vision and Pattern Recognition, Las Vegas, Jun 27-30, 2016. Washington: IEEE Computer Society, 2016:4013-4021. ?

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末公给,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子蜘渣,更是在濱河造成了極大的恐慌妓布,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,858評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件宋梧,死亡現(xiàn)場離奇詭異,居然都是意外死亡狰挡,警方通過查閱死者的電腦和手機捂龄,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,372評論 3 395
  • 文/潘曉璐 我一進店門释涛,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人倦沧,你說我怎么就攤上這事唇撬。” “怎么了展融?”我有些...
    開封第一講書人閱讀 165,282評論 0 356
  • 文/不壞的土叔 我叫張陵窖认,是天一觀的道長。 經(jīng)常有香客問我告希,道長扑浸,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,842評論 1 295
  • 正文 為了忘掉前任燕偶,我火速辦了婚禮喝噪,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘指么。我一直安慰自己酝惧,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 67,857評論 6 392
  • 文/花漫 我一把揭開白布伯诬。 她就那樣靜靜地躺著晚唇,像睡著了一般。 火紅的嫁衣襯著肌膚如雪盗似。 梳的紋絲不亂的頭發(fā)上哩陕,一...
    開封第一講書人閱讀 51,679評論 1 305
  • 那天,我揣著相機與錄音桥言,去河邊找鬼萌踱。 笑死,一個胖子當(dāng)著我的面吹牛号阿,可吹牛的內(nèi)容都是我干的并鸵。 我是一名探鬼主播,決...
    沈念sama閱讀 40,406評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼扔涧,長吁一口氣:“原來是場噩夢啊……” “哼园担!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起枯夜,我...
    開封第一講書人閱讀 39,311評論 0 276
  • 序言:老撾萬榮一對情侶失蹤弯汰,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后湖雹,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體咏闪,經(jīng)...
    沈念sama閱讀 45,767評論 1 315
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,945評論 3 336
  • 正文 我和宋清朗相戀三年摔吏,在試婚紗的時候發(fā)現(xiàn)自己被綠了鸽嫂。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片纵装。...
    茶點故事閱讀 40,090評論 1 350
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖据某,靈堂內(nèi)的尸體忽然破棺而出橡娄,到底是詐尸還是另有隱情,我是刑警寧澤癣籽,帶...
    沈念sama閱讀 35,785評論 5 346
  • 正文 年R本政府宣布挽唉,位于F島的核電站,受9級特大地震影響筷狼,放射性物質(zhì)發(fā)生泄漏瓶籽。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,420評論 3 331
  • 文/蒙蒙 一桑逝、第九天 我趴在偏房一處隱蔽的房頂上張望棘劣。 院中可真熱鬧,春花似錦楞遏、人聲如沸茬暇。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,988評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽糙俗。三九已至,卻和暖如春预鬓,著一層夾襖步出監(jiān)牢的瞬間巧骚,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,101評論 1 271
  • 我被黑心中介騙來泰國打工格二, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留劈彪,地道東北人。 一個月前我還...
    沈念sama閱讀 48,298評論 3 372
  • 正文 我出身青樓顶猜,卻偏偏與公主長得像沧奴,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子长窄,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 45,033評論 2 355

推薦閱讀更多精彩內(nèi)容