1. 先把KNN這件事想明白一個不訓練的算法憑什么上臺面K近鄰算法K-Nearest Neighbors簡稱KNN在機器學習里是個特別有意思的存在。你把訓練集喂給它它幾乎什么都不干——不擬合參數(shù)、不更新權(quán)重、不建梯度、不迭代。fit()一調(diào)用時間基本花在把數(shù)據(jù)存進內(nèi)存上。真正干活是在predict()那一刻來一個新樣本它把訓練集里所有點和這個新樣本逐個算一遍距離找出最近的K個鄰居看這些鄰居大多屬于哪一類就把新樣本判成哪一類。這套邏輯樸素得像小區(qū)里問路——你不知道某個地方在哪就問身邊最近的幾個人多數(shù)人說往東你大概就往東走。我在帶新人的時候常講一句話KNN的全部智能都藏在距離和K這兩個詞里。它沒有學到特征和標簽之間的映射函數(shù)它學到的其實是整個訓練集本身就是模型。這個特性決定了它的優(yōu)缺點都極其鮮明——實現(xiàn)簡單、無需訓練、天然支持多分類、對非線性邊界有一定適應(yīng)力但同時內(nèi)存占用隨數(shù)據(jù)線性增長、預(yù)測慢、對特征尺度極其敏感、在高維空間里幾乎失效。這篇文章講的是鳶尾花Iris數(shù)據(jù)集上的KNN完整實現(xiàn)。這個數(shù)據(jù)集小得可愛150條樣本4個特征花萼長度、花萼寬度、花瓣長度、花瓣寬度3個類別三個不同品種每類50條。很多人嫌它太簡單一行代碼準確率就96%但我偏偏喜歡拿它做教學載體因為它干凈到可以把每一個環(huán)節(jié)單獨拎出來反復折騰——你怎么選K、怎么縮放、怎么切分、怎么驗證、怎么解釋結(jié)果全都看得清清楚楚不會有一堆臟數(shù)據(jù)干擾你的判斷。適合誰看寫過一點Python、聽過KNN但沒真正跑通過全流程的人調(diào)包能出結(jié)果但說不清為什么準確率是96%而不是99%的人想手寫一遍KNN理解底層的人。如果你已經(jīng)能熟練做網(wǎng)格搜索和特征工程這篇可能偏基礎(chǔ)但里面關(guān)于標準化順序、維度災(zāi)難、距離度量選擇的那些坑大概率還是能幫你省點時間。1.1 距離即投票KNN的核心假設(shè)到底是什么任何算法背后都有一套對世界的假設(shè)KNN的假設(shè)直白得不像話特征空間里挨得近的樣本標簽大概率相同。這句話聽起來像廢話但它成立是需要條件的。第一個條件特征必須能構(gòu)成一個有意義的度量空間。也就是說兩個樣本之間的距離要真的能反映它們的相似程度。如果你的特征里混了用戶ID這種純編號ID1001和ID1002看起來距離是1但它們之間沒有任何語義關(guān)聯(lián)這種距離就是假的會把整個模型帶偏。第二個條件各特征的量綱要可比?；ㄝ嚅L度單位是厘米取值范圍4.3到7.9如果我還加入一個花瓣面積單位是平方厘米取值可能到20以上。那么在歐氏距離的平方和里面積這個維度會完全壓制長度維度長度特征等于白給。這就是為什么KNN幾乎必須配標準化不是建議是必須。第三個條件樣本密度要足夠。KNN是純局部方法它只看鄰居。如果某個區(qū)域訓練樣本極其稀疏那最近的K個鄰居可能其實離得很遠投票結(jié)果的可靠性就大打折扣。所以KNN在樣本分布均勻、類別邊界清晰的問題上表現(xiàn)好在長尾、稀疏、類別極不平衡的問題上容易翻車。想清楚這三點你就明白為什么很多人拿KNN跑出來的結(jié)果忽高忽低——不是算法玄學是這三個前提被破壞了。我見過有人把KNN直接用在幾千維的文本TF-IDF向量上然后抱怨KNN效果太差其實問題不在KNN在于高維空間里最近這個概念已經(jīng)失真了這個后面會專門講。1.2 它和線性模型、樹模型的根本差別在哪拿KNN和邏輯回歸對比差別不只在效果更在模型的形態(tài)。邏輯回歸訓練完得到的是一組權(quán)重模型被壓縮成幾個數(shù)字你甚至可以把它寫在紙上KNN訓練完得到的是一整個訓練集模型體積等于數(shù)據(jù)體積。這是參數(shù)化模型和非參數(shù)化模型的分水嶺。和決策樹對比更有意思。決策樹是在特征空間上切格子切完之后每個格子給一個標簽切法由訓練數(shù)據(jù)決定但一旦確定就固定了KNN不切格子它的決策邊界是隨著查詢點移動的每個新樣本都會臨時劃出一片鄰域。所以KNN的決策邊界可以非常曲折、非常非線性代價是它沒法給你一個可解釋的規(guī)則表。這個差別帶來的直接影響是預(yù)測成本。邏輯回歸預(yù)測是幾次乘加微秒級決策樹預(yù)測是幾次比較也是微秒級KNN預(yù)測要遍歷訓練集算距離10萬條樣本就是10萬次距離計算。線上服務(wù)里如果QPS高、延遲敏感KNN基本不適合直接懟在請求鏈路上得配近似最近鄰索引或者做預(yù)計算。1.3 哪些場景該用它哪些場景趁早換我自己的判斷標準很簡單看三條數(shù)據(jù)量中等偏小萬級到十萬級特征維度不高幾十維以內(nèi)且特征都是數(shù)值型、量綱可以統(tǒng)一——用KNN很合適甚至經(jīng)常能打贏一些復雜模型。樣本量極大或者維度極高幾百上千維——先別急著上KNN除非你愿意上近似檢索。需要給出為什么這么判的業(yè)務(wù)解釋——KNN能給出因為和這5個歷史樣本最像這種解釋在某些風控、推薦場景反而比權(quán)重系數(shù)更好用。實際落地里KNN出現(xiàn)頻率最高的地方往往不是主模型而是推薦系統(tǒng)的召回層找相似物品/相似用戶、圖像檢索的相似度匹配、異常檢測離所有鄰居都遠的點判為異常、缺失值填補用鄰居的均值填、以及小樣本冷啟動場景。這些場景有一個共同點——相似本身就是業(yè)務(wù)語義不需要額外解釋。2. 核心參數(shù)與原理細節(jié)K值、距離度量、決策規(guī)則2.1 K值怎么選偏差和方差的那桿秤K是KNN里唯一一個真正意義上的超參數(shù)它的作用是把局部的尺度調(diào)大調(diào)小。K1的時候模型完全信任最近的那一個點。訓練集里只要有一個噪聲點或者標錯的樣本它周圍就會被劃出一小塊錯誤的領(lǐng)地。這叫過擬合方差極大換一批訓練數(shù)據(jù)決策邊界抖得厲害。準確率在訓練集上永遠是100%因為最近的鄰居就是它自己測試集上往往明顯掉一檔。K取很大比如K100那就等于每次預(yù)測都看一大片區(qū)域。極端情況KN所有樣本都參與投票預(yù)測結(jié)果對所有輸入都一樣永遠是多數(shù)類。這叫欠擬合偏差極大。所以K的選擇本質(zhì)就是在偏差和方差之間找平衡點。經(jīng)驗做法有這么幾條從K sqrt(N)開始試N是訓練樣本數(shù)。150條樣本開方約12所以鳶尾花上可以先試K11或13這類奇數(shù)。二分類問題盡量取奇數(shù)避免平票。多分類問題平票的可能性低但sklearn內(nèi)部會按距離加權(quán)或取索引小的類行為不一定符合你的預(yù)期能避開就避開。不要用訓練集選K。必須用驗證集或者交叉驗證否則你會選出一個過擬合的極小K。K通常不會取到樣本量的很大比例一般不超過樣本數(shù)的十分之一到五分之一。在鳶尾花上做5折交叉驗證掃一遍K你會看到一個很典型的U形曲線K從1漲到5左右準確率快速上升5到15之間平臺區(qū)很平超過25之后開始緩慢下滑。平臺的平坦程度說明這個數(shù)據(jù)集類別邊界很干凈實際上K取3到15結(jié)果都差不多在0.95以上。這時候真正的決策依據(jù)不是哪個準確率高0.01而是哪個更穩(wěn)——我一般會選平臺區(qū)中間偏小的值兼顧穩(wěn)定性和局部性。2.2 距離度量不是只有歐氏距離默認的歐氏距離是L2范數(shù)公式是各維度差值的平方和再開根號。它在特征維度獨立、量綱統(tǒng)一、分布比較團狀的時候最合適。曼哈頓距離是L1范數(shù)各維度差值絕對值求和。它對異常值更不敏感因為誤差不被平方放大。如果某一維偶爾出現(xiàn)離譜的值L1會比L2穩(wěn)。余弦相似度衡量的是方向而不是長度。文本向量、embedding向量這種長度不重要、方向才重要的場景必須用余弦。你在高維稀疏向量上算歐氏距離結(jié)果基本被向量模長主導完全沒有意義。閔可夫斯基距離是把上面這些統(tǒng)一起來的通式參數(shù)p決定具體形態(tài)p1是曼哈頓p2是歐氏p→∞是切比雪夫距離取各維度差值的最大值。sklearn里metricminkowski, p2就是歐氏距離。還有幾個值得知道但少用的馬氏距離考慮了特征之間的協(xié)方差能消除特征相關(guān)性影響但對小樣本估計協(xié)方差矩陣不穩(wěn)漢明距離專門給離散/二值特征用Jaccard用于集合相似度。選距離度量的判斷順序我一般是這樣的特征有明確物理量綱且相互獨立→歐氏存在離群點或分布偏斜→曼哈頓向量方向比大小重要→余弦特征間強相關(guān)→考慮馬氏或先做PCA去相關(guān)。2.3 投票規(guī)則與距離加權(quán)標準做法是等權(quán)投票K個鄰居一人一票。但這里有個隱含問題——K5的時候最近的鄰居距離可能是1.0最遠的那個可能是3.0它們的話語權(quán)完全一樣這顯然不太合理。weightsdistance就是來解決這個的每個鄰居的票權(quán)按距離的倒數(shù)分配。距離越近權(quán)重越大距離趨近于0時權(quán)重趨于無窮大。實現(xiàn)上sklearn用的是1/d而不是嚴格的1/d2避免數(shù)值爆炸。效果上距離加權(quán)在小K時通常能提升一點準確率尤其在類別邊界附近但在噪聲多的數(shù)據(jù)上反而可能放大單個異常點的影響。我的實測經(jīng)驗是數(shù)據(jù)干凈、邊界清晰比如鳶尾花等權(quán)和距離加權(quán)差別在1個百分點以內(nèi)數(shù)據(jù)噪聲大、密度不均等權(quán)反而更穩(wěn)?；貧w任務(wù)里KNN就是取鄰居標簽的均值或加權(quán)均值邏輯完全對應(yīng)。分類投票、回歸平均這是同一套思想的兩種輸出形式。2.4 維度災(zāi)難KNN在高維空間為什么會失靈這一條很關(guān)鍵但很多教程一筆帶過。高維空間里距離這個概念會鈍化。設(shè)想在一維區(qū)間[0,1]里隨機撒點最近鄰的距離大概在1/N量級在100維的單位超立方體里撒點任意兩點之間的距離會高度集中在一個很窄的范圍內(nèi)最近鄰和最遠鄰的距離比值趨近于1。這意味著什么意味著最近的那個鄰居并不比隨便挑一個點近多少KNN的局部性假設(shè)直接崩塌。有個粗略的經(jīng)驗說法KNN在維度超過20左右就要小心了超過50基本只能靠降維或者換算法。這不是硬閾值跟樣本量有關(guān)——樣本量隨維度指數(shù)增長才夠用所以維度和樣本量的關(guān)系比單純的維度數(shù)字更重要。對策很實在先做特征選擇把明顯無關(guān)的維度砍掉。用PCA或者別的降維手段壓到低維再跑KNN很多時候比直接堆全特征效果還好。如果維度確實高比如embedding檢索不要用暴力搜索改用近似最近鄰HNSW、IVF之類的索引結(jié)構(gòu)現(xiàn)代向量檢索基本都是把這個思路工程化。2.5 特征縮放不做這一步后面全是白干標準化公式很基礎(chǔ)但值得寫清楚因為很多細節(jié)在為什么這么算上$$z \frac{x - \mu}{\sigma}$$其中 μ 是訓練集上該特征的均值σ 是訓練集上該特征的標準差。變換之后每個特征的均值是0標準差是1。標準化不改變分布形狀只改尺度和位置。這里有兩個必須記住的操作禁忌計算 μ 和 σ 只能用訓練集然后把這個 μ、σ 應(yīng)用到驗證集和測試集上。如果用全體數(shù)據(jù)算均值和方差再切分測試集的信息就泄漏進訓練過程了評估結(jié)果會虛高。正確做法是scaler.fit(X_train)后scaler.transform(X_test)或者用Pipeline把縮放和模型綁在一起。StandardScaler 對常數(shù)列標準差為0處理時會把該列置0不會報錯但那一維就廢了。如果某個特征在訓練集里只有一個值請?zhí)崆白R別出來刪掉。除了StandardScaler還有幾種常見選擇MinMaxScaler把值壓到[0,1]適合需要固定范圍或者稀疏數(shù)據(jù)RobustScaler用中位數(shù)和四分位距對離群點更穩(wěn)Normalizer是行歸一化改的是樣本向量的模長而不是列的分布別和標準化搞混。在鳶尾花上標準化的影響其實不大因為四個特征本來就在同一量級厘米歐氏距離不會嚴重失衡。但為了養(yǎng)成習慣也因為這是通用流程的一部分正式代碼里我還是會加上。這個習慣在你自己接手的真實數(shù)據(jù)上價值巨大——我踩過的最隱蔽的一次坑就是一個以元為單位的價格特征和一個以百分比為單位的比率特征混在一起KNN的準確率一直在60%左右徘徊標準化之后直接跳到88%。3. 鳶尾花數(shù)據(jù)集實操從加載到評估的完整流程3.1 加載數(shù)據(jù)與第一輪觀察先用sklearn自帶的版本省去下載和清洗的功夫。import numpy as np import pandas as pd from sklearn.datasets import load_iris iris load_iris() X pd.DataFrame(iris.data, columnsiris.feature_names) y pd.Series(iris.target, nametarget) print(X.shape) # (150, 4) print(y.value_counts()) # 每類各50 print(X.describe().round(2))跑出來你會看到各特征的范圍花萼長度4.3到7.9花萼寬度2.0到4.4花瓣長度1.0到6.9花瓣寬度0.1到2.5。四個特征量綱一致但分布寬度差了一倍多花瓣寬度的標準差比花萼寬度小不少說明花瓣相關(guān)的特征區(qū)分度可能更高。再看一下類別分布三個類各50條完全均衡。這一點很重要——均衡數(shù)據(jù)上準確率這個指標才可信如果是9:1的不均衡數(shù)據(jù)全猜多數(shù)類也有90%準確率那時候必須看混淆矩陣、看每類的precision和recall。還有一個必做的動作看特征之間的相關(guān)性。print(X.corr().round(2))在鳶尾花上花瓣長度和花瓣寬度的相關(guān)系數(shù)大約0.96花萼長度和花瓣長度大約0.87。高度相關(guān)意味著信息冗余對KNN來說不致命它不像線性回歸那樣有共線性問題但確實意味著你在重復計權(quán)同一個維度。這也是后面能不能做PCA降維的依據(jù)之一。我一般還會畫一下散點圖用花瓣長度和花瓣寬度做二維投影肉眼就能看出三類基本是分開的其中一類setosa和另外兩類完全線性可分另外兩類之間只有少量交疊。這解釋了為什么這個數(shù)據(jù)集的準確率天花板在96%到98%之間——那2%到4%的錯誤幾乎全部來自后兩類的邊界樣本無論你怎么調(diào)K都是這個數(shù)。3.2 數(shù)據(jù)劃分與標準化的正確姿勢劃分訓練測試集固定隨機種子保證可復現(xiàn)from sklearn.model_selection import train_test_split X_train, X_test, y_train, y_test train_test_split( X, y, test_size0.2, random_state42, stratifyy )stratifyy這個參數(shù)一定要加。它保證訓練集和測試集里的類別比例與原數(shù)據(jù)一致。150條數(shù)據(jù)、3個類如果不分層隨機切出30條測試集有概率某一類只分到6條甚至更少評估結(jié)果波動會很大。分層之后每類在測試集里穩(wěn)定出現(xiàn)10條。然后是標準化注意順序from sklearn.preprocessing import StandardScaler scaler StandardScaler() X_train_scaled scaler.fit_transform(X_train) # fit和transform都在訓練集 X_test_scaled scaler.transform(X_test) # 只用transform這里有個細節(jié)fit_transform(X_train)之后scaler.mean_和scaler.scale_就固定了transform(X_test)用的是訓練集的統(tǒng)計量。如果你寫成scaler.fit_transform(X_test)那就是徹頭徹尾的數(shù)據(jù)泄漏測試集的均值方差參與了變換評估結(jié)果會偏樂觀。這個坑極其常見我在代碼review里見過不止一次。更穩(wěn)妥的寫法是直接用Pipeline把縮放和模型串成一體交叉驗證的時候sklearn會自動保證每一折的縮放只在該折的訓練部分擬合from sklearn.pipeline import Pipeline from sklearn.neighbors import KNeighborsClassifier pipe Pipeline([ (scaler, StandardScaler()), (knn, KNeighborsClassifier(n_neighbors5)) ])用Pipeline最大的好處不是代碼短是從結(jié)構(gòu)上消滅了忘記順序的可能性。我現(xiàn)在的習慣是只要涉及任何需要fit的預(yù)處理一律走Pipeline不給自己犯錯的機會。3.3 建模、預(yù)測與評估from sklearn.metrics import accuracy_score, classification_report, confusion_matrix pipe.fit(X_train, y_train) y_pred pipe.predict(X_test) print(f準確率: {accuracy_score(y_test, y_pred):.4f}) print(confusion_matrix(y_test, y_pred)) print(classification_report(y_test, y_pred, target_namesiris.target_names))默認K5跑出來測試集準確率通常在0.93到0.97之間浮動取決于random_state?；煜仃嚂@示錯誤集中在那兩個容易混的類之間。光看一個測試集的結(jié)果不夠30條樣本錯一條就是3.3個百分點波動太大。養(yǎng)成習慣用交叉驗證from sklearn.model_selection import cross_val_score scores cross_val_score(pipe, X, y, cv5, scoringaccuracy) print(f5折CV: {scores.mean():.4f} ± {scores.std():.4f})5折交叉驗證用的是全部150條數(shù)據(jù)每一折都有120條訓練、30條驗證最后取5次平均。這個數(shù)字比單次train_test_split穩(wěn)定得多也是我用來判斷模型好壞的主要依據(jù)。注意這里的cross_val_score傳入的是完整的X和y內(nèi)部會自己切分——Pipeline保證了每一折的縮放只在訓練折上擬合所以沒有泄漏。還有一個指標值得看一眼predict_proba。proba pipe.predict_proba(X_test[:5]) print(proba.round(3))KNN輸出的概率就是K個鄰居里各類的占比。K5的話概率只可能是0、0.2、0.4、0.6、0.8、1.0這幾檔粒度非常粗。如果你需要比較細的概率估計比如做排序或者閾值調(diào)整KNN不是好選擇它的概率沒經(jīng)過校準。可以用CalibratedClassifierCV包一層或者干脆換邏輯回歸。3.4 手寫一遍KNN把黑箱拆開調(diào)包調(diào)通了不代表懂了。我強烈建議每個人手寫一遍代碼不超過40行import numpy as np from collections import Counter class SimpleKNN: def __init__(self, k3, p2): self.k k self.p p def fit(self, X, y): self.X_train np.asarray(X, dtypefloat) self.y_train np.asarray(y) return self def _dist(self, a, b): if self.p 1: return np.abs(a - b).sum(axis1) if self.p 2: return np.sqrt(((a - b) ** 2).sum(axis1)) return np.max(np.abs(a - b), axis1) def predict(self, X): X np.asarray(X, dtypefloat) preds [] for x in X: d self._dist(x, self.X_train) idx np.argsort(d)[:self.k] votes Counter(self.y_train[idx]) preds.append(votes.most_common(1)[0][0]) return np.array(preds)寫完自己手動標準化一下再跑mu, sigma X_train.mean(0), X_train.std(0) Xtr (X_train - mu) / sigma Xte (X_test - mu) / sigma knn SimpleKNN(k5).fit(Xtr.values, y_train.values) pred knn.predict(Xte.values) print((pred y_test.values).mean())結(jié)果應(yīng)該和sklearn基本一致。手寫的價值在于你會親眼看到預(yù)測這一步是O(N×D)的計算N是訓練樣本數(shù)D是維度。這解釋了為什么KNN預(yù)測慢、為什么大數(shù)據(jù)量下必須上索引結(jié)構(gòu)。你也會看到argsort是整個流程的瓶頸——實際上只需要前K個用argpartition會快很多工業(yè)實現(xiàn)里還會用堆或者樹結(jié)構(gòu)。順便說一個優(yōu)化點很多人不知道sklearn的KNN可以預(yù)計算距離矩陣在特征不變、只反復調(diào)K的場景下能省一半時間還有一個更實用的技巧是設(shè)置algorithmauto之外顯式指定kd_tree或ball_tree在低維中等數(shù)據(jù)量下比暴力搜索快一個數(shù)量級。4. 調(diào)參與性能優(yōu)化把準確率從96%推到98%4.1 用網(wǎng)格搜索找K和權(quán)重手工試K效率太低直接上GridSearchCV。注意要搜的參數(shù)在Pipeline里帶前綴from sklearn.model_selection import GridSearchCV param_grid { knn__n_neighbors: list(range(1, 31, 2)), knn__weights: [uniform, distance], knn__p: [1, 2] } grid GridSearchCV(pipe, param_grid, cv5, scoringaccuracy, n_jobs-1) grid.fit(X, y) print(grid.best_params_) print(f{grid.best_score_:.4f})參數(shù)空間是15×2×260組每組5折一共300次訓練加預(yù)測在鳶尾花這種規(guī)模上幾秒鐘就跑完了。一個觀察跑完之后你大概率會發(fā)現(xiàn)最優(yōu)的K落在3到11之間weights是uniform還是distance差距很小p是1還是2差距也很小。這說明在這個數(shù)據(jù)集上這些超參并不敏感。這時候別再糾結(jié)了挑一個泛化誤差曲線平臺區(qū)靠中間的值定下來把精力放到別的地方去。但有一種情況需要注意GridSearchCV用的是同一份數(shù)據(jù)做交叉驗證選參數(shù)然后再用同一個best_score_報告結(jié)果這個數(shù)字會略微偏樂觀因為參數(shù)是在這同一批數(shù)據(jù)上挑出來的最優(yōu)值。嚴格的評估應(yīng)該留一個從不參與調(diào)參的測試集最后單獨跑一次。我一般這么切先分出20%測試集鎖起來剩下80%做交叉驗證調(diào)參最后用那20%報最終結(jié)果且只報一次。4.2 縮放方式對結(jié)果的對比實驗這個實驗值得親手做一遍體會一下預(yù)處理選擇會直接影響指標。from sklearn.preprocessing import MinMaxScaler, RobustScaler from sklearn.model_selection import cross_val_score for name, sc in [(none, None), (standard, StandardScaler()), (minmax, MinMaxScaler()), (robust, RobustScaler())]: steps [(knn, KNeighborsClassifier(n_neighbors5))] if sc: steps.insert(0, (scaler, sc)) p Pipeline(steps) s cross_val_score(p, X, y, cv5) print(f{name:9s} {s.mean():.4f} ± {s.std():.4f})在鳶尾花上你會看到不縮放的結(jié)果其實只差一點點因為四個特征量綱本來就接近。但如果我把花萼寬度乘以100模擬量綱差異不縮放的準確率會立刻掉到0.6以下而標準化版本紋絲不動。這個對比實驗我每次帶人做都會做沖擊感比講十遍原理都強。順便提一個邊界情況如果數(shù)據(jù)里有明顯的極端離群點StandardScaler會被拉偏因為均值和標準差都受離群值影響這時候RobustScaler通常更穩(wěn)。判斷方法很簡單畫個箱線圖看看就清楚了。4.3 大數(shù)據(jù)量下的加速思路鳶尾花只有150條談性能優(yōu)化有點奢侈。但既然講到這把思路說清楚對將來有用。暴力搜索的復雜度預(yù)測單個樣本是O(N×D)N條查詢就是O(N×M×D)。N到十萬級、D到百級單次預(yù)測就要上千萬次運算這是不可接受的。第一條路是空間索引。KD樹在低維經(jīng)驗上D20能顯著加速本質(zhì)是把空間遞歸二分查詢時剪枝掉不可能包含近鄰的分支。Ball Tree對高維稍好一些用超球體代替超矩形切分。sklearn里algorithm參數(shù)支持這兩種設(shè)置auto會讓它自動選。但注意維度一高這些樹的剪枝效率急劇下降最后退化成接近暴力搜索。第二條路是近似最近鄰ANN。放棄精確找到最近的K個轉(zhuǎn)而以很高概率找到足夠近的K個換來數(shù)量級的速度提升。常見思路有基于圖的把樣本連成近鄰圖查詢時在圖上貪心游走、基于量化的把向量空間分桶只在附近桶里找、基于哈希的。這些在檢索場景已經(jīng)成為標配文本、圖像、向量數(shù)據(jù)庫底層基本都是這套東西。第三條路是降維。PCA降到幾十維再跑既緩解維度災(zāi)難又直接降低每次距離計算的成本雙贏。前提是降維后保留的方差足夠一般看累計解釋方差到95%以上。第四條路是工程手段預(yù)計算距離矩陣、批量向量化用矩陣運算代替循環(huán)、把訓練集常駐內(nèi)存避免重復加載、用更緊湊的數(shù)據(jù)類型float32代替float64內(nèi)存減半、速度提升。這些都是實現(xiàn)層面的小優(yōu)化累加起來也很可觀。5. 常見問題與排查實錄5.1 準確率異常低按這個順序排查我整理了一個排查順序從最常見的開始第一檢查是否做了縮放。這是頭號原因占比超過一半。特征量綱差異大而沒縮放KNN基本等于只在一個特征上做判斷。第二檢查縮放是不是做在了劃分之前。如果先對全量數(shù)據(jù)fit_transform再切分雖然準確率會虛高而不是變低但另一個常見錯誤是把fit_transform用在了測試集上這會導致訓練和測試的變換標準不一致結(jié)果可能變得很奇怪。第三檢查有沒有把ID、時間戳之類的列喂進去。這類特征的距離沒有語義會嚴重稀釋真實特征的作用。特征列一定要手動點一遍名。第四檢查類別是否極度不平衡。不平衡時KNN會傾向于預(yù)測多數(shù)類整體準確率看起來還行但少數(shù)類的召回率慘不忍睹。這時候要看每類的指標并考慮用weightsdistance或者對多數(shù)類下采樣。第五檢查K值是不是太大。K接近樣本數(shù)時模型退化所有預(yù)測趨同。第六檢查是否存在重復樣本或近重復樣本導致的泄漏。如果同一條數(shù)據(jù)或幾乎相同的樣本同時出現(xiàn)在訓練集和測試集KNN會直接背答案準確率高得離譜。這在做過數(shù)據(jù)增強或者按時間切分不當?shù)膱鼍袄锖艹Ｒ姟?.2 問題速查表現(xiàn)象最可能的原因處理方式準確率明顯低于預(yù)期未做特征縮放加StandardScaler走Pipeline準確率虛高接近100%數(shù)據(jù)泄漏或重復樣本檢查切分邏輯去重縮放只在訓練集fit訓練集100%、測試集很差K過小過擬合增大K用交叉驗證選值所有樣本預(yù)測成同一類K過大或類別極不平衡減小K處理不平衡調(diào)參后提升很小超參不敏感或特征信息不足轉(zhuǎn)去做特征工程別死磕超參維度很高時效果崩塌維度災(zāi)難特征選擇或降維換ANN索引預(yù)測速度很慢暴力搜索大樣本KD樹/Ball樹或近似檢索概率輸出只有幾檔KNN概率粒度等于1/K增大K或用概率校準5.3 幾個我踩過的坑寫下來省你時間坑一把predict和predict_proba的結(jié)果混用。前者返回類別標簽后者返回各類占比。有次我在做閾值篩選直接把predict的結(jié)果當概率去比較大小邏輯全錯排查了半天才發(fā)現(xiàn)類型不對。坑二忽略特征的相關(guān)性做特征越多越好的判斷。加了一堆高度相關(guān)的特征KNN的距離計算里同一個信息被反復計權(quán)效果反而下降。加特征之前先看相關(guān)性矩陣。坑三在時間序列數(shù)據(jù)上隨機切分。KNN天然會把未來的信息帶到過去因為相似的樣本在時間上往往相鄰。時間序列必須按時間切分否則評估結(jié)果毫無意義。坑四以為n_neighbors是唯一要調(diào)的。實際上metric的選擇在文本、embedding類任務(wù)上比K的影響大得多很多人壓根沒試過換度量。坑五在小樣本上過度相信單次評估結(jié)果。150條數(shù)據(jù)測試集30條多對一個少對一個就是3.3個百分點。一定要用交叉驗證并且報出均值±標準差而不是報一個孤零零的數(shù)字?？恿浌潭S機種子。每次跑結(jié)果都不一樣調(diào)試的時候根本不知道變化是來自你的改動還是來自切分。random_state是必備參數(shù)不是可選項。我在實際使用中的一個體會是KNN看起來是最簡單的算法之一但它對數(shù)據(jù)質(zhì)量的敏感度在所有常見模型里排前列。線性模型還能靠權(quán)重把噪聲特征壓下去樹模型還能靠分裂點忽略無用維度KNN不行——你給它什么特征它就老老實實按這些特征算距離一維都不能糊弄。所以用好KNN的關(guān)鍵從來不在算法本身而在數(shù)據(jù)準備那幾步特征選得準不準、量綱統(tǒng)一了沒有、有沒有混進偽特征。把這幾步做扎實KNN在很多中小規(guī)模問題上依然是個又便宜又穩(wěn)的基線值得在項目早期就先跑一遍看看天花板在哪。