我有一个非常稀疏的二进制分割任务,即分割掩码包含 0 或 1,并且有 ~95% 的零和只有 ~5% 的零。我使用焦点损失来解决稀疏问题(在我的情况下相当于不平衡)。我有另一条信息想包含在损失项中。
所需的输出始终在对角线上对称。我正在寻找一种在损失中使用此信息的方法,但找不到解决方案。我该怎么做?
对于分割图中的一些对称性示例,我添加了一个箭头来显示对称轴:
我有一个非常稀疏的二进制分割任务,即分割掩码包含 0 或 1,并且有 ~95% 的零和只有 ~5% 的零。我使用焦点损失来解决稀疏问题(在我的情况下相当于不平衡)。我有另一条信息想包含在损失项中。
所需的输出始终在对角线上对称。我正在寻找一种在损失中使用此信息的方法,但找不到解决方案。我该怎么做?
对于分割图中的一些对称性示例,我添加了一个箭头来显示对称轴:
如果你知道它是对称的,那么你可以做几件事。
不要费心学习图像的两半。只需在输出矩阵的上半部分或下半部分放置一个零掩码,然后让网络回归另一半。只是不要让网络做比它需要做的更多的工作。
在您的情况下,您似乎可以创建两个添加在一起的损失函数。
这将有助于网络平等地学习两半。
这可能很愚蠢,但在和可能有助于促进对称性,但我不太喜欢它。就我个人而言,我更偏爱(1)。
您可以在训练时采用 (2) 的损失函数,但在测试时只需使用具有更好指标的一半并将其复制到下半部分。
添加一个后处理自定义层,取两半的平均值,以保证对称。
然后做你正常的焦点损失。我个人最喜欢这个,因为它始终保证对称输出并且是一个非常简单的自定义层。