张量流是否仅以单精度计算交叉熵?

Does tensorflow compute the cross entropy only with single precision?

我正在尝试完全理解 TensorFlow 中交叉熵的计算。在下面的一段代码中,我使用 numpy 生成双精度随机双精度数据 x,将其转换为 logits 进行二进制分类(即每个数据点只有一个 logit),通过 sigmoid 将其映射到 sig,计算交叉熵 ce,然后计算平均交叉熵 mce。下面是 TensorFlow 中的类似计算。我的问题是:

为什么我的平均交叉熵 mce(在 numpy 中用双精度计算)和 TensorFlow tf.losses.sigmoid_cross_entropy

我不知道,我在哪里忘记指定 TensorFlow 以双精度计算。此外,如果我使用 tf.nn.reduce_mean,请参阅 mcetf2 的计算,计算每个数据点的交叉熵,然后我得到我的 numpy 结果。差异从何而来?谢谢!

import numpy as np
import tensorflow as tf

#%%

# Number of data pionts nx and dimension dx
nx = 10
dx = 4

# Input data
x = np.random.rand(nx,dx)

#%% Numpy

# Transform to logits for binary classification with sigmoid
matrix = np.random.rand(dx,1)
logits = np.matmul(x,matrix)
print('Logits dimensions: %s' % str(logits.shape))

# Sigmoid
def sigmoid(x):
    return 1. / (1. + np.exp(-x))
sig = sigmoid(logits)
print('Sigmoid dimensions: %s' % str(sig.shape))

# Discrete probabilities
p = np.random.randint(2,size=nx)[:,np.newaxis]
print('Probability dimensions: %s'% str(p.shape))

# Cross entropy for each data point
ce = p*np.log(1/sig)+(1-p)*np.log(1/(1-sig))

# Mean cross entropy
mce = np.mean(ce)
print('MCE with np: %.16f' % mce)

#%% Tensorflow

xp = tf.placeholder(dtype=tf.float64,shape=[None,dx])
pp = tf.placeholder(dtype=tf.float64,shape=[None,1])

model = xp
c1 = tf.constant(matrix,dtype=tf.float64)
model = tf.matmul(xp,c1)
sigtf = tf.nn.sigmoid(model)
cetf = tf.nn.sigmoid_cross_entropy_with_logits(labels=pp,logits=model)
mcetf = tf.losses.sigmoid_cross_entropy(pp,model)
mcetf2 = tf.reduce_mean(cetf)

sess = tf.Session()
feed = {xp:x,pp:p}
print('Error in logits: %.16f' % np.max(np.abs(sess.run(model,feed)-logits)))
print('Error in sigmoid: %.16f' % np.max(np.abs(sess.run(sigtf,feed)-sig)))
print('Error in CE: %.16f' % np.max(np.abs(sess.run(cetf,feed)-ce)))
print('Error in MCE: %.16f' % np.abs(sess.run(mcetf,feed)-mce))
print('Error in MCE2: %.16f' % np.abs(sess.run(mcetf2,feed)-mce))
sess.close()

Logits dimensions: (10, 1)

Sigmoid dimensions: (10, 1)

Probability dimensions: (10, 1)

MCE with np: 0.7413128316195762

Error in logits: 0.0000000000000000

Error in sigmoid: 0.0000000000000000

Error in CE: 0.0000000000000009

Error in MCE: 0.0000000297816550

Error in MCE2: 0.0000000000000001

(32 位)floats 的使用似乎在 Tensorflowcompute_weighted_loss() function used by sigmoid_cross_entropy 中硬编码

作为一个小问题,您用于计算 ce 的 numpy 代码在数值上不是很稳定 — 但它不会影响此处的任何内容。我将其实现为:

ce = p * -np.log(sig) + (1-p) * -np.log1p(-sig)

sig 接近零时 log1p is the main change. your use of 1 - sig will lose all precision 的使用