从矩阵中学习阈值

数据挖掘 机器学习 神经网络
2022-03-08 16:50:24

我有一个二维矩阵,我想从这个矩阵中学习一个阈值。矩阵中所有大于阈值的项将变为 1,所有小于阈值的项将变为 0。然后,我有一个可微的损失函数来评估生成的 0-1 矩阵。通常使用哪些方法来学习阈值?我们是把矩阵放在网络上,还是有任何非神经网络的方法?

2个回答

你有某种损失函数L(y,y^)接受某种真值,y, 和0/1矩阵中的值,y^.

隐含地,这是真值的函数,y, 的阈值,t,以及矩阵中原始的未舍入值,m.

你的目标是最小化损失,所以最小化L(y,t,m). 根据复杂性,您可以选择手动(微积分)或在计算机上通过循环可能的阈值来执行此操作。如果矩阵值都在01,例如,您可以在每个0.01增量。

这是一个相当简单的计算,因为损失函数实际上只取决于一个变量,t, 作为真值标签y和未舍入的值m是固定的。

阈值方法的层次结构是:

  1. 手动编码阈值
  2. 使用决策树学习阈值
  3. 使用一组树(例如,随机森林或 XGBoost)学习阈值
  4. 使用神经网络学习阈值