Isotonic Calibration Layer in DNN

常规校准方法

  • Platt scaling

  • isotonic regression

isotonic regression layer
image.png

  • ycali =Σi=0i=kRelu(ei+wi)vi+b,vi={ step,  if i<kystepi=ky_{\text {cali }}=\Sigma_{i=0}^{i=k} \operatorname{Relu}\left(e_i+w_i\right) \cdot v_i+b, v_i=\left\{\begin{array}{ll}\text { step, } & \text { if } i<k \\ y-s t e p \cdot & \mathrm{i}=\mathrm{k}\end{array}\right.k=argmaxj(yk=\arg \max _j(y- step j>0)\cdot j>0). #card
    • 分段拟合,对预测值分桶,每个桶一个可训练的权重 wi

    • relu 保证非负

    • ei 由校准特征得到的 embedding

作者

Ryen Xiang

发布于

2025-04-18

更新于

2025-04-18

许可协议


网络回响

评论