tensorflow - 如何强制自动编码器中的瓶颈产生二进制值？

Question

我尝试强制自动编码器中的瓶颈层产生二进制值。我是通过使用 tensorflow.cond 来实现的在自定义损失函数中，惩罚所有不是 0 或 1 的值。但是这种方法非常慢。有没有更好的方法来执行此操作？

def custom_loss(weight):
    def loss(y_true, y_pred):
        reconstruction_loss = binary_crossentropy(y_true, y_pred)

        def binarize_loss(value):
            return tf.cond(tf.reduce_mean(value) > 0.5, lambda: tf.abs(value - 1), lambda: tf.abs(value))

        binarized_loss_value = tf.map_fn(binarize_loss, neckLayer.output)
        return reconstruction_loss + (K.mean(binarized_loss_value , axis=-1) * weight)

    return loss

Answer 1

我可能会摆脱 tf.cond语句，因为您可以使用简单的算术来做您想做的事:

def loss(y_true, y_pred):
        reconstruction_loss = binary_crossentropy(y_true, y_pred)

        binary_neck_loss = tf.abs(0.5 - tf.abs(0.5 - neckLayer.output))

        return reconstruction_loss + (K.mean(binary_neck_loss , axis=-1) * weight)

当然，我不确切知道您的数据的形状，但您应该能够从那里推断出来。