tl; dr即使这是一个图像分类数据集,它仍然是一项非常简单的任务,为此,可以轻松地找到从输入到预测的直接映射。
回答:
这是一个非常有趣的问题,由于逻辑回归的简单性,您实际上可以找到答案。
logistic回归所做的是为每个图像接受784输入并将它们与权重相乘以生成其预测。有趣的是,由于输入和输出之间的直接映射(即没有隐藏层),每个权重的值对应于在计算每个类别的概率时要考虑784输入中的每个输入的数量。现在,通过取每个类别的权重并将其重塑为28 × 28(即图像分辨率),我们可以知道哪些像素对每个类别的计算最重要。
再次注意,这些是权重。
现在看一看上面的图像,并专注于前两位数字(即零和一)。蓝色权重表示此像素的强度对该类别有很大贡献,而红色值表示该像素起了负面作用。
0
1个
2378
通过此操作,您可以看到逻辑回归有很好的机会获得很多正确的图像,这就是为什么它得分很高的原因。
复制上图的代码有些陈旧,但您可以执行以下操作:
import tensorflow as tf
import matplotlib.pyplot as plt
from tensorflow.examples.tutorials.mnist import input_data
# Load MNIST:
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
# Create model
x = tf.placeholder(tf.float32, shape=(None, 784))
y = tf.placeholder(tf.float32, shape=(None, 10))
W = tf.Variable(tf.zeros((784,10)))
b = tf.Variable(tf.zeros((10)))
z = tf.matmul(x, W) + b
y_hat = tf.nn.softmax(z)
cross_entropy = tf.reduce_mean(-tf.reduce_sum(y * tf.log(y_hat), reduction_indices=[1]))
optimizer = tf.train.GradientDescentOptimizer(0.5).minimize(cross_entropy) #
correct_pred = tf.equal(tf.argmax(y_hat, 1), tf.argmax(y, 1))
accuracy = tf.reduce_mean(tf.cast(correct_pred, tf.float32))
# Train model
batch_size = 64
with tf.Session() as sess:
loss_tr, acc_tr, loss_ts, acc_ts = [], [], [], []
sess.run(tf.global_variables_initializer())
for step in range(1, 1001):
x_batch, y_batch = mnist.train.next_batch(batch_size)
sess.run(optimizer, feed_dict={x: x_batch, y: y_batch})
l_tr, a_tr = sess.run([cross_entropy, accuracy], feed_dict={x: x_batch, y: y_batch})
l_ts, a_ts = sess.run([cross_entropy, accuracy], feed_dict={x: mnist.test.images, y: mnist.test.labels})
loss_tr.append(l_tr)
acc_tr.append(a_tr)
loss_ts.append(l_ts)
acc_ts.append(a_ts)
weights = sess.run(W)
print('Test Accuracy =', sess.run(accuracy, feed_dict={x: mnist.test.images, y: mnist.test.labels}))
# Plotting:
for i in range(10):
plt.subplot(2, 5, i+1)
weight = weights[:,i].reshape([28,28])
plt.title(i)
plt.imshow(weight, cmap='RdBu') # as noted by @Eric Duminil, cmap='gray' makes the numbers stand out more
frame1 = plt.gca()
frame1.axes.get_xaxis().set_visible(False)
frame1.axes.get_yaxis().set_visible(False)