天天看點

AlphaGo 的算法和技術分析

最近我仔細看了下alphago在《自然》雜志上發表的文章,寫一些分析給大家分享。

alphago這個系統主要由幾個部分組成:

走棋網絡(policy network),給定目前局面,預測/采樣下一步的走棋。

快速走子(fast rollout),目标和1一樣,但在适當犧牲走棋品質的條件下,速度要比1快1000倍。

估值網絡(value network),給定目前局面,估計是白勝還是黑勝。

蒙特卡羅樹搜尋(monte carlo tree search,mcts),把以上這三個部分連起來,形成一個完整的系統。

我們的darkforest和alphago同樣是用4搭建的系統。darkforest較alphago而言,在訓練時加強了1,而少了2和3,然後以開源軟體pachi的預設政策 (default policy)部分替代了2的功能。以下介紹下各部分。

1.走棋網絡:

走棋網絡把目前局面作為輸入,預測/采樣下一步的走棋。它的預測不隻給出最強的一手,而是對棋盤上所有可能的下一着給一個分數。棋盤上有361個點,它就給出361個數,好招的分數比壞招要高。darkforest在這部分有創新,通過在訓練時預測三步而非一步,提高了政策輸出的品質,和他們在使用增強學習進行自我對局後得到的走棋網絡(rl network)的效果相當。當然,他們并沒有在最後的系統中使用增強學習後的網絡,而是用了直接通過訓練學習到的網絡(sl network),理由是rl network輸出的走棋缺乏變化,對搜尋不利。

AlphaGo 的算法和技術分析

有意思的是在alphago為了速度上的考慮,隻用了寬度為192的網絡,而并沒有使用最好的寬度為384的網絡(見圖2(a)),是以要是gpu更快一點(或者更多一點),alphago肯定是會變得更強的。

所謂的0.1秒走一步,就是純粹用這樣的網絡,下出有最高置信度的合法着法。這種做法一點也沒有做搜尋,但是大局觀非常強,不會陷入局部戰鬥中,說它模組化了“棋感”一點也沒有錯。我們把darkforest的走棋網絡直接放上kgs就有3d的水準,讓所有人都驚歎了下。可以說,這一波圍棋ai的突破,主要得益于走棋網絡的突破。這個在以前是不可想像的,以前用的是基于規則,或者基于局部形狀再加上簡單線性分類器訓練的走子生成法,需要慢慢調參數年,才有進步。

當然,隻用走棋網絡問題也很多,就我們在darkforest上看到的來說,會不顧大小無謂争劫,會無謂脫先,不顧局部死活,對殺出錯,等等。有點像高手不經認真思考的随手棋。因為走棋網絡沒有價值判斷功能,隻是憑“直覺”在下棋,隻有在加了搜尋之後,電腦才有價值判斷的能力。

快速走子

那有了走棋網絡,為什麼還要做快速走子呢?有兩個原因,首先走棋網絡的運作速度是比較慢的,alphago說是3毫秒,我們這裡也差不多,而快速走子能做到幾微秒級别,差了1000倍。是以在走棋網絡沒有傳回的時候讓cpu不閑着先搜尋起來是很重要的,等到網絡傳回更好的着法後,再更新對應的着法資訊。

其次,快速走子可以用來評估盤面。由于天文數字般的可能局面數,圍棋的搜尋是毫無希望走到底的,搜尋到一定程度就要對現有局面做個估分。在沒有估值網絡的時候,不像國象可以通過算棋子的分數來對盤面做比較精确的估值,圍棋盤面的估計得要通過模拟走子來進行,從目前盤面一路走到底,不考慮岔路地算出勝負,然後把勝負值作為目前盤面價值的一個估計。這裡有個需要權衡的地方:在同等時間下,模拟走子的品質高,單次估值精度高但走子速度慢;模拟走子速度快乃至使用随機走子,雖然單次估值精度低,但可以多模拟幾次算平均值,效果未必不好。是以說,如果有一個品質高又速度快的走子政策,那對于棋力的提高是非常有幫助的。

為了達到這個目标,神經網絡的模型就顯得太慢,還是要用傳統的局部特征比對(local pattern matching)加線性回歸(logistic regression)的方法,這辦法雖然不新但非常好使,幾乎所有的廣告推薦,競價排名,新聞排序,都是用的它。與更為傳統的基于規則的方案相比,它在吸納了衆多高手對局之後就具備了用梯度下降法自動調參的能力,是以性能提高起來會更快更省心。alphago用這個辦法達到了2微秒的走子速度和24.2%的走子準确率。24.2%的意思是說它的最好預測和圍棋高手的下子有0.242的機率是重合的,相比之下,走棋網絡在gpu上用2毫秒能達到57%的準确率。這裡,我們就看到了走子速度和精度的權衡。

AlphaGo 的算法和技術分析

和訓練深度學習模型不同,快速走子用到了局部特征比對,自然需要一些圍棋的領域知識來選擇局部特征。對此alphago隻提供了局部特征的數目(見extended table 4),而沒有說明特征的具體細節。我最近也實驗了他們的辦法,達到了25.1%的準确率和4-5微秒的走子速度,然而全系統整合下來并沒有複現他們的水準。我感覺上24.2%并不能完全概括他們快速走子的棋力,因為隻要走錯關鍵的一步,局面判斷就完全錯誤了;而圖2(b)更能展現他們快速走子對盤面形勢估計的精确度,要能達到他們圖2(b)這樣的水準,比簡單地比對24.2%要做更多的工作,而他們并未在文章中強調這一點。

AlphaGo 的算法和技術分析

在alphago有了快速走子之後,不需要走棋網絡和估值網絡,不借助任何深度學習和gpu的幫助,不使用增強學習,在單機上就已經達到了3d的水準(見extended table 7倒數第二行),這是相當厲害的了。任何使用傳統方法在單機上達到這個水準的圍棋程式,都需要花費數年的時間。在alphago之前,aja huang曾經自己寫過非常不錯的圍棋程式,在這方面相信是有很多的積累的。

估值網絡

AlphaGo 的算法和技術分析

alphago的估值網絡可以說是錦上添花的部分,從fig 2(b)和extended table 7來看,沒有它alphago也不會變得太弱,至少還是會在7d-8d的水準。少了估值網絡,等級分少了480分,但是少了走棋網絡,等級分就會少掉800至1000分。特别有意思的是,如果隻用估值網絡來評估局面(2177),那其效果還不及隻用快速走子(2416),隻有将兩個合起來才有更大的提高。我的猜測是,估值網絡和快速走子對盤面估計是互補的,在棋局一開始時,大家下得比較和氣,估值網絡會比較重要;但在有複雜的死活或是對殺時,通過快速走子來估計盤面就變得更重要了。考慮到估值網絡是整個系統中最難訓練的部分(需要三千萬局自我對局),我猜測它是最晚做出來并且最有可能能進一步提高的。

關于估值網絡訓練資料的生成,值得注意的是文章中的附錄小字部分。與走棋網絡不同,每一盤棋隻取一個樣本來訓練以避免過拟合,不然對同一對局而言輸入稍有不同而輸出都相同,對訓練是非常不利的。這就是為什麼需要三千萬局,而非三千萬個盤面的原因。對于每局自我對局,取樣本是很有講究的,先用sl network保證走棋的多樣性,然後随機走子,取盤面,然後用更精确的rl network走到底以得到最正确的勝負估計。當然這樣做的效果比用單一網絡相比好多少,我不好說。

一個讓我吃驚的地方是,他們完全沒有做任何局部死活/對殺分析,純粹是用暴力訓練法訓練出一個相當不錯的估值網絡。這在一定程度上說明深度卷積網絡(dcnn)有自動将問題分解成子問題,并分别解決的能力。

另外,我猜測他們在取訓練樣本時,判定最終勝負用的是中國規則。是以說三月和李世石對局的時候也要求用中國規則,不然如果換成别的規則,就需要重新訓練估值網絡(雖然我估計結果差距不會太大)。至于為什麼一開始就用的中國規則,我的猜測是程式設計非常友善(我在寫darkforest的時候也是這樣覺得的)。

蒙特卡羅樹搜尋

這部分基本用的是傳統方法,沒有太多可以評論的,他們用的是帶先驗的uct,即先考慮dcnn認為比較好的着法,然後等到每個着法探索次數多了,選擇更相信探索得來的勝率值。而darkforest則直接選了dcnn推薦的前3或是前5的着法進行搜尋。我初步試驗下來效果差不多,當然他們的辦法更靈活些,在允許使用大量搜尋次數的情況下,他們的辦法可以找到一些dcnn認為不好但卻對局面至關重要的着法。

一個有趣的地方是在每次搜尋到葉子節點時,沒有立即展開葉子節點,而是等到通路次數到達一定數目(40)才展開,這樣避免産生太多的分支,分散搜尋的注意力,也能節省gpu的寶貴資源,同時在展開時,對葉節點的盤面估值會更準确些。除此之外,他們也用了一些技巧,以在搜尋一開始時,避免多個線程同時搜尋一路變化,這部分我們在darkforest中也注意到了,并且做了改進。

總結

總的來說,這整篇文章是一個系統性的工作,而不是一兩個小點有了突破就能達到的勝利。在成功背後,是作者們,特别是兩位第一作者david silver和aja huang,在博士階段及畢業以後五年以上的積累,非一朝一夕所能完成的。他們能做出alphago并享有現在的榮譽,是實至名歸的。

從以上分析也可以看出,與之前的圍棋系統相比,alphago較少依賴圍棋的領域知識,但還遠未達到通用系統的程度。職業棋手可以在看過了寥寥幾局之後明白對手的風格并采取相應政策,一位資深遊戲玩家也可以在玩一個新遊戲幾次後很快上手,但到目前為止,人工智能系統要達到人類水準,還是需要大量樣本的訓練的。可以說,沒有千年來衆多棋手在圍棋上的積累,就沒有圍棋ai的今天。

在alphago中,增強學習(reinforcement learning)所扮演的角色并沒有想像中那麼大。在理想情況下,我們希望人工智能系統能在對局中動态地适應環境和對手的招式并且找到辦法反制之,但是在alphago中增強學習更多地是用于提供更多品質更好的樣本,給有監督學習(supervised learning)以訓練出更好的模型。在這方面增強學習還有很長的路要走。

另外,據他們的文章所言,alphago整個系統在單機上已具有了職業水準,若是谷歌願意開幾萬台機器和李世石對決(這對它來說再容易不過了,改個參數就行),相信比賽會非常精彩。

===========================

一些更新。

問題1:“alphago的mcts做rollout的時候,除了使用快速走子,還用了搜尋樹的已有部分,看起來像是amaf/rave反過來:amaf是把快速走子的資訊傳導到樹的其它無關部分,alphago是把樹的其它無關部分拿來增強快速走子。我懷疑這是不是它棋力比其它dcnn+mcts強的原因之一。"

這個辦法在解死活題的文章中出現過,會在一定程度上提高搜尋效率,但是提高多少還不知道。

問題2:“rollout的走法品質變好可能會導緻棋力下降。”

這裡要分兩種情況,tree policy和default policy。在alphago的文章裡面已經說過了,tree policy的分布不能太尖,不然在搜尋時太過重視一些看起來的好着,可能使得棋力下降。但是除了這種原因,一般來說tree policy變好棋力還是會變強的。

default policy這邊,即(半)随機走子到最後然後判分,就很複雜了,品質變好未必對局面能估得更準。default policy需要保證的是每塊棋的死活大體正确,不要把死的棋下成活的或者反之,而對大局觀的要求反而沒有那麼高。雙方完全可以配合着把每塊棋下完,然後轉戰另一塊,而不是說搶在對方前去别處占先手。

====================================分割線================================

繼續閱讀