Closed1
LambdaRankのLightGBMの実装

NN用のLambdaRankの公式実装は見つからなかったが、LightGBMの実装[1]は参考になる。
といっても、勾配更新の計算式は論文[2]の(6)式と全く同じである。
計算上の工夫として、
- 損失関数の対象性を考慮して
のケースのみ計算l_i>l_j - sigmoid関数などの関数をテーブル化
- sigmoid関数を
の区間でクリッピング[-50, 50] |\Delta_{NDCG}| = | 2^{l_i} - 2^{l_j} | \cdot | \frac{1}{\log(2 + i)} - \frac{1}{\log(2 + j)}| \cdot \frac{1}{\text{NDCG}_{\max}}
このスクラップは2022/12/29にクローズされました