往下拉回到首頁
ibu-boost:一個梯度提升決策樹函式庫,分割點是被絕對拒絕而非相對排序

ibu-boost:一個梯度提升決策樹函式庫,分割點是被絕對拒絕而非相對排序

ibu-boost: A GBDT Library Where Splits Are Absolutely Rejected, Not Just Relatively Ranked

我開發了一個小型梯度提升決策樹(GBDT)函式庫,基於《篩選就夠了》論文(Nakanishi 2026)中的篩選轉換方法。該論文原本是為 Transformer 設計的,但核心概念——用絕對閾值拒絕取代相對比較——自然地對應到 GBDT 的分割點選擇。這是一個獨立實現,將篩選概念應用於梯度提升樹。與其只是將候選分割點從最好排到最差,這個函式庫使用絕對閾值來直接拒絕不良分割點,可能使演算法更快更有效率。

關鍵字

gradient-boosted treessplit selectionscreening transformmachine learningopen source library