統計學習筆記(3)——k近鄰法與kd樹

        在使用k近鄰法進行分類時,對新的實例,根據其k個最近鄰的訓練實例的類別,通過多數表決的方式進行預測。由於k近鄰模型的特徵空間一般是n維實數向量,所以距離的計算通常採用的是歐式距離。關鍵的是k值的選取,如果k值太小就意味着整體模型變得複雜,容易發生過擬合,即如果鄰近的實例點恰巧是噪聲,預測就會出錯,極端的情況是k=1,稱爲最近鄰算法,對於待預測點x,與x最近的點決定了x的類別。k值得增大意味着整體的模型變得簡單,極端的情況是k=N,那麼無論輸入實例是什麼,都簡單地預測它屬於訓練集中最多的類,這樣的模型過於簡單。經驗是,k值一般去一個比較小的值,通常採取交叉驗證的方法來選取最優的k值。

       實現k近鄰法時,主要考慮的問題是如何對訓練數據進行快速k近鄰搜索,這點在特徵空間的維數大以及訓練數據容量大時尤其重要。k近鄰法的最簡單實現是線性掃描,這時要計算輸入實例與每一個訓練實例的距離,當訓練集很大時,計算非常耗時,這種方法是不可行的。爲了提高k近鄰搜索的效率,可以考慮使用特殊的結構存儲訓練數據,以減少計算距離的次數。具體方法有很多,這裏介紹kd樹方法。

1.實例

       先以一個簡單直觀的實例來介紹k-d樹算法。假設有6個二維數據點{(2,3),(5,4),(9,6),(4,7),(8,1),(7,2)},數據點位於二維空間內(如圖2中黑點所示)。k-d樹算法就是要確定圖2中這些分割空間的分割線(多維空間即爲分割平面,一般爲超平面)。下面就要通過一步步展示k-d樹是如何確定這些分割線的。


       k-d樹算法可以分爲兩大部分,一部分是有關k-d樹本身這種數據結構建立的算法,另一部分是在建立的k-d樹上如何進行最鄰近查找的算法。

2.構造kd樹

        kd樹是一種對k維空間中的實例點進行存儲以便對其進行快速搜索的樹形數據結構。kd樹是二叉樹,表示對k維空間的一個劃分。構造kd樹相當於不斷地用垂直於座標軸的超平面將k維空間進行切分,構成一系列的k維超矩形區域。kd樹的每一個節點對應於一個k維超矩形區域。k-d樹是一個二叉樹,每個節點表示一個空間範圍。下表給出的是k-d樹每個節點中主要包含的數據結構。


         從上面對k-d樹節點的數據類型的描述可以看出構建k-d樹是一個逐級展開的遞歸過程。下面給出的是構建k-d樹的僞碼。

算法:構建k-d樹(createKDTree)
輸入:數據點集Data-set和其所在的空間Range
輸出:Kd,類型爲k-d tree
1.If Data-set爲空,則返回空的k-d tree
2.調用節點生成程序:
  (1)確定split域:對於所有描述子數據(特徵矢量),統計它們在每個維上的數據方差。假設每條數據記錄爲64維,可計算64個方差。挑選出最大值,對應的維就是split域的值。數據方差大表明沿該座標軸方向上的數據分散得比較開,在這個方向上進行數據分割有較好的分辨率;
  (2)確定Node-data域:數據點集Data-set按其第split域的值排序。位於正中間的那個數據點被選爲Node-data。此時新的Data-set' = Data-set \ Node-data(除去其中Node-data這一點)。
3.dataleft = {d屬於Data-set' && d[split] ≤ Node-data[split]}
   Left_Range = {Range && dataleft}
  dataright = {d屬於Data-set' && d[split] > Node-data[split]}
   Right_Range = {Range && dataright}
4.left = 由(dataleft,Left_Range)建立的k-d tree,即遞歸調用createKDTree(dataleft,Left_Range)。並設置left的parent域爲Kd;
   right = 由(dataright,Right_Range)建立的k-d tree,即調用createKDTree(dataleft,Left_Range)。並設置right的parent域爲Kd。

       以上述舉的實例來看,過程如下:

        由於此例簡單,數據維度只有2維,所以可以簡單地給x,y兩個方向軸編號爲0,1,也即split={0,1}。

        (1)確定split域的首先該取的值。分別計算x,y方向上數據的方差得知x方向上的方差最大,所以split域值首先取0,也就是x軸方向;

         (2)確定Node-data的域值。根據x軸方向的值2,5,9,4,8,7排序選出中值爲7,所以Node-data = (7,2)。這樣,該節點的分割超平面就是通過(7,2)並垂直於split = 0(x軸)的直線x = 7;

         (3)確定左子空間和右子空間。分割超平面x = 7將整個空間分爲兩部分,如下圖所示。x <= 7的部分爲左子空間,包含3個節點{(2,3),(5,4),(4,7)};另一部分爲右子空間,包含2個節點{(9,6),(8,1)}。


         如算法所述,k-d樹的構建是一個遞歸的過程。然後對左子空間和右子空間內的數據重複根節點的過程就可以得到下一級子節點(5,4)和(9,6)(也就是左右子空間的'根'節點),同時將空間和數據集進一步細分。如此反覆直到空間中只包含一個數據點,如下圖所示。最後生成的k-d樹如下圖所示。

3.搜索kd樹

        在k-d樹中進行數據的查找也是特徵匹配的重要環節,其目的是檢索在k-d樹中與查詢點距離最近的數據點。這裏先以一個簡單的實例來描述最鄰近查找的基本思路。

        星號表示要查詢的點(2.1,3.1)。通過二叉搜索,順着搜索路徑很快就能找到最鄰近的近似點,也就是葉子節點(2,3)。而找到的葉子節點並不一定就是最鄰近的,最鄰近肯定距離查詢點更近,應該位於以查詢點爲圓心且通過葉子節點的圓域內。爲了找到真正的最近鄰,還需要進行'回溯'操作:算法沿搜索路徑反向查找是否有距離查詢點更近的數據點。此例中先從(7,2)點開始進行二叉查找,然後到達(5,4),最後到達(2,3),此時搜索路徑中的節點爲小於(7,2)和(5,4),大於(2,3),首先以(2,3)作爲當前最近鄰點,計算其到查詢點(2.1,3.1)的距離爲0.1414,然後回溯到其父節點(5,4),並判斷在該父節點的其他子節點空間中是否有距離查詢點更近的數據點。以(2.1,3.1)爲圓心,以0.1414爲半徑畫圓,如下圖所示。發現該圓並不和超平面y = 4交割,因此不用進入(5,4)節點右子空間中去搜索。


         再回溯到(7,2),以(2.1,3.1)爲圓心,以0.1414爲半徑的圓更不會與x = 7超平面交割,因此不用進入(7,2)右子空間進行查找。至此,搜索路徑中的節點已經全部回溯完,結束整個搜索,返回最近鄰點(2,3),最近距離爲0.1414。

          一個複雜點了例子如查找點爲(2,4.5)。同樣先進行二叉查找,先從(7,2)查找到(5,4)節點,在進行查找時是由y = 4爲分割超平面的,由於查找點爲y值爲4.5,因此進入右子空間查找到(4,7),形成搜索路徑<(7,2),(5,4),(4,7)>,取(4,7)爲當前最近鄰點,計算其與目標查找點的距離爲3.202。然後回溯到(5,4),計算其與查找點之間的距離爲3.041。以(2,4.5)爲圓心,以3.041爲半徑作圓,如下圖左所示。可見該圓和y = 4超平面交割,所以需要進入(5,4)左子空間進行查找。此時需將(2,3)節點加入搜索路徑中得<(7,2),(2,3)>。回溯至(2,3)葉子節點,(2,3)距離(2,4.5)比(5,4)要近,所以最近鄰點更新爲(2,3),最近距離更新爲1.5。回溯至(7,2),以(2,4.5)爲圓心1.5爲半徑作圓,並不和x = 7分割超平面交割,如下圖右所示。至此,搜索路徑回溯完。返回最近鄰點(2,3),最近距離1.5。

       k-d樹查詢算法的僞代碼如下所示。

算法:k-d樹最鄰近查找
輸入:Kd,    //k-d tree類型
     target  //查詢數據點
輸出:nearest, //最鄰近數據點
     dist      //最鄰近數據點和查詢點間的距離
1. If Kd爲NULL,則設dist爲infinite並返回
2. //進行二叉查找,生成搜索路徑
   Kd_point = &Kd;                   //Kd-point中保存k-d tree根節點地址
   nearest = Kd_point -> Node-data;  //初始化最近鄰點
   while(Kd_point)
     push(Kd_point)到search_path中; //search_path是一個堆棧結構,存儲着搜索路徑節點指針
 /*** If Dist(nearest,target) > Dist(Kd_point -> Node-data,target)
       nearest  = Kd_point -> Node-data;    //更新最近鄰點
       Max_dist = Dist(Kd_point,target);  //更新最近鄰點與查詢點間的距離  ***/
     s = Kd_point -> split;                       //確定待分割的方向
     If target[s] <= Kd_point -> Node-data[s]     //進行二叉查找
       Kd_point = Kd_point -> left;
     else
       Kd_point = Kd_point ->right;
   nearest = search_path中最後一個葉子節點; //注意:二叉搜索時不比計算選擇搜索路徑中的最鄰近點,這部分已被註釋
   Max_dist = Dist(nearest,target);    //直接取最後葉子節點作爲回溯前的初始最近鄰點

3. //回溯查找
   while(search_path != NULL)
     back_point = 從search_path取出一個節點指針;   //從search_path堆棧彈棧
     s = back_point -> split;                   //確定分割方向
     If Dist(target[s],back_point -> Node-data[s]) < Max_dist   //判斷還需進入的子空間
       If target[s] <= back_point -> Node-data[s]
         Kd_point = back_point -> right;  //如果target位於左子空間,就應進入右子空間
       else
         Kd_point = back_point -> left;    //如果target位於右子空間,就應進入左子空間
       將Kd_point壓入search_path堆棧;
     If Dist(nearest,target) > Dist(Kd_Point -> Node-data,target)
       nearest  = Kd_point -> Node-data;                 //更新最近鄰點
       Min_dist = Dist(Kd_point -> Node-data,target);  //更新最近鄰點與查詢點間的距離

       當維數較大時,直接利用k-d樹快速檢索的性能急劇下降。假設數據集的維數爲D,一般來說要求數據的規模N滿足條件:N遠大於2的D次方,才能達到高效的搜索。


參考:

http://www.cnblogs.com/eyeszjwang/articles/2429382.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章