TensorFlow Workshop项目：基于低阶API构建深度神经网络实现MNIST分类

2025-07-05 18:42:23作者：董宙帆

前言

在TensorFlow Workshop项目中，深度神经网络(DNN)的实现是一个重要环节。本文将详细介绍如何使用TensorFlow的低阶API构建一个包含两个隐藏层的全连接神经网络，并在MNIST手写数字数据集上实现分类任务。相比简单的线性分类器，深度神经网络能够捕捉更复杂的特征关系，获得更高的分类准确率。

环境准备与数据加载

首先需要导入必要的库和模块：

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import math
import os
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data

初始化TensorFlow会话和计算图：

tf.reset_default_graph()
sess = tf.Session()

加载MNIST数据集，TensorFlow提供了便捷的接口：

mnist = input_data.read_data_sets('/tmp/data', one_hot=True)

模型参数配置

定义网络结构和训练参数：

# 隐藏层神经元数量
HIDDEN1_SIZE = 500  # 第一隐藏层
HIDDEN2_SIZE = 250  # 第二隐藏层

NUM_CLASSES = 10    # 输出类别数(0-9)
NUM_PIXELS = 28 * 28 # 输入像素数

# 训练参数
TRAIN_STEPS = 2000  # 训练步数
BATCH_SIZE = 100    # 批大小
LEARNING_RATE = 0.001 # 学习率

网络结构设计

输入层定义

使用name_scope组织输入层，便于TensorBoard可视化：

with tf.name_scope('input'):
    images = tf.placeholder(tf.float32, [None, NUM_PIXELS], name="pixels")
    labels = tf.placeholder(tf.float32, [None, NUM_CLASSES], name="labels")

全连接层函数

定义一个通用的全连接层创建函数，支持不同的激活函数：

def fc_layer(input, size_out, name="fc", activation=None):
    with tf.name_scope(name):
        size_in = int(input.shape[1])
        w = tf.Variable(tf.truncated_normal([size_in, size_out], stddev=0.1), name="weights")
        b = tf.Variable(tf.constant(0.1, shape=[size_out]), name="bias")
        wx_plus_b = tf.matmul(input, w) + b
        if activation: return activation(wx_plus_b)
        return wx_plus_b

网络架构构建

构建包含两个隐藏层的深度神经网络：

# 第一隐藏层(ReLU激活)
fc1 = fc_layer(images, HIDDEN1_SIZE, "fc1", activation=tf.nn.relu)

# 第二隐藏层(ReLU激活)
fc2 = fc_layer(fc1, HIDDEN2_SIZE, "fc2", activation=tf.nn.relu)

# Dropout层(防止过拟合)
dropped = tf.nn.dropout(fc2, keep_prob=0.9)

# 输出层(无激活函数)
y = fc_layer(dropped, NUM_CLASSES, name="output")

训练配置

损失函数

使用交叉熵损失函数：

with tf.name_scope("loss"):
    loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=y, labels=labels))
    tf.summary.scalar('loss', loss)  # 记录损失值

优化器

采用Adam优化器而非普通的梯度下降：

with tf.name_scope("optimizer"):
    train = tf.train.AdamOptimizer(LEARNING_RATE).minimize(loss)

评估指标

定义准确率计算：

with tf.name_scope("evaluation"):
    correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(labels, 1))
    accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
    tf.summary.scalar('accuracy', accuracy)  # 记录准确率

训练过程

日志记录配置

设置TensorBoard日志记录：

LOGDIR = './graphs'
train_writer = tf.summary.FileWriter(os.path.join(LOGDIR, "train"))
train_writer.add_graph(sess.graph)
test_writer = tf.summary.FileWriter(os.path.join(LOGDIR, "test"))
summary_op = tf.summary.merge_all()

模型训练

执行训练循环：

sess.run(tf.global_variables_initializer())

for step in range(TRAIN_STEPS):
    batch_xs, batch_ys = mnist.train.next_batch(BATCH_SIZE)
    summary_result, _ = sess.run([summary_op, train], 
                               feed_dict={images: batch_xs, labels: batch_ys})
    
    train_writer.add_summary(summary_result, step)
    
    # 每100步在测试集上评估一次
    if step % 100 == 0:
        summary_result, acc = sess.run([summary_op, accuracy], 
                                     feed_dict={images: mnist.test.images, 
                                                labels: mnist.test.labels})
        test_writer.add_summary(summary_result, step)
        print("test accuracy: %f at step %d" % (acc, step))

# 最终测试准确率
print("Accuracy %f" % sess.run(accuracy, 
                             feed_dict={images: mnist.test.images,
                                       labels: mnist.test.labels}))
train_writer.close()
test_writer.close()