Python tf.keras.losses.BinaryFocalCrossentropy用法及代碼示例

計算真實標簽和預測之間的焦點 cross-entropy 損失。

繼承自：Loss

用法

tf.keras.losses.BinaryFocalCrossentropy(
    gamma=2.0, from_logits=False, label_smoothing=0.0, axis=-1,
    reduction=losses_utils.ReductionV2.AUTO,
    name='binary_focal_crossentropy'
)

Binary cross-entropy loss 通常用於二元(0 或 1)分類任務。損失函數需要以下輸入：

• y_true(真實標簽)：這是 0 或 1。
• y_pred(預測值)：這是模型的預測，即單個浮點值，它或者代表一個 logit，(即，當 from_logits=True 時 [-inf, inf] 中的值)或概率(即， [0., 1.] 當 from_logits=False 時的值)。

根據 Lin 等人，2018 年的說法，它有助於將 "focal factor" 應用於 down-weight 簡單示例並更多地關注困難示例。默認情況下，焦點張量計算如下：

focal_factor = (1 - output) ** gamma 用於 1 類 focal_factor = output ** gamma 用於 0 類，其中 gamma 是聚焦參數。當 gamma=0 時，這個函數相當於二元交叉熵損失。

使用compile() API：

model.compile(
  loss=tf.keras.losses.BinaryFocalCrossentropy(gamma=2.0, from_logits=True),
  ....
)

作為獨立函數：

# Example 1:(batch_size = 1, number of samples = 4)
y_true = [0, 1, 0, 0]
y_pred = [-18.6, 0.51, 2.94, -12.8]
loss = tf.keras.losses.BinaryFocalCrossentropy(gamma=2, from_logits=True)
loss(y_true, y_pred).numpy()
0.691

# Example 2:(batch_size = 2, number of samples = 4)
y_true = [[0, 1], [0, 0]]
y_pred = [[-18.6, 0.51], [2.94, -12.8]]
# Using default 'auto'/'sum_over_batch_size' reduction type.
loss = tf.keras.losses.BinaryFocalCrossentropy(gamma=3, from_logits=True)
loss(y_true, y_pred).numpy()
0.647

# Using 'sample_weight' attribute
loss(y_true, y_pred, sample_weight=[0.8, 0.2]).numpy()
0.133

# Using 'sum' reduction` type.
loss = tf.keras.losses.BinaryFocalCrossentropy(gamma=4, from_logits=True,
    reduction=tf.keras.losses.Reduction.SUM)
loss(y_true, y_pred).numpy()
1.222

# Using 'none' reduction type.
loss = tf.keras.losses.BinaryFocalCrossentropy(gamma=5, from_logits=True,
    reduction=tf.keras.losses.Reduction.NONE)
loss(y_true, y_pred).numpy()
array([0.0017 1.1561], dtype=float32)