为钞票认证开发一个神经网络

为新的数据集开发神经网络预测模型可能具有挑战性。

一种方法是首先检查数据集并构思可能适用的模型,然后探索简单模型在数据集上的学习动态,最后使用健壮的测试框架开发和调整模型以适应数据集。

这个过程可以用于开发有效的神经网络模型,用于分类和回归预测建模问题。

在本教程中,您将了解如何为纸币二分类数据集开发多层感知机(MLP)神经网络模型。

完成本教程后,您将了解:

  • 如何加载和汇总纸币数据集,并利用结果来建议数据准备和模型配置。
  • 如何探索简单MLP模型在数据集上的学习动态。
  • 如何开发模型性能的稳健估计,调整模型性能并对新数据进行预测。

让我们开始吧。

  • 更新于2021年10月:predict_classes() 语法已弃用
Develop a Neural Network for Banknote Authentication

为钞票认证开发一个神经网络
照片来自 Lenny K Photography,部分权利保留。

教程概述

本教程分为4个部分,它们是:

  1. 纸币分类数据集
  2. 神经网络学习动态
  3. 稳健的模型评估
  4. 最终模型及预测

纸币分类数据集

第一步是定义和探索数据集。

我们将使用“纸币”标准二分类数据集。

纸币数据集涉及根据从照片中获得的多个测量值来预测给定纸币是否为真。

该数据集包含 1,372 行,具有 5 个数值变量。这是一个具有两个类别(二分类)的分类问题。

以下是数据集中五种变量的列表。

  • 小波变换图像的方差(连续)。
  • 小波变换图像的偏度(连续)。
  • 小波变换图像的峰度(连续)。
  • 图像的熵(连续)。
  • 类别(整数)。

下面是数据集中前 5 行的样本

你可以在此处了解更多关于此数据集的信息:

我们可以直接从 URL 将数据集加载为 pandas DataFrame;例如:

运行示例将直接从 URL 加载数据集并报告数据集的形状。

在这种情况下,我们可以确认数据集有 5 个变量(4 个输入和 1 个输出),并且数据集有 1,372 行数据。

对于神经网络来说,这并不算很多行数据,这表明使用小型网络,可能带有正则化,会是比较合适的。

这也表明使用 k 折交叉验证会是一个好主意,因为它能比训练/测试分割提供更可靠的模型性能估计,并且因为单个模型可以在几秒钟内完成训练,而不是像处理大型数据集那样需要数小时或数天。

接下来,我们可以通过查看摘要统计信息和数据图来进一步了解数据集。

运行示例首先加载数据,然后打印每个变量的摘要统计信息。

我们可以看到值具有不同的均值和标准差,可能需要在建模之前进行一些归一化或标准化。

然后为每个变量创建直方图。

我们可以看到前两个变量可能具有类高斯分布的形状,而后两个输入变量可能具有偏斜高斯分布或指数分布的形状。

我们可能可以通过对每个变量使用幂变换来获得一些好处,以使概率分布的偏斜程度降低,这可能会提高模型性能。

Histograms of the Banknote Classification Dataset

纸币分类数据集的直方图

现在我们熟悉了数据集,让我们探索一下如何开发神经网络。

神经网络学习动态

我们将使用 TensorFlow 为数据集开发多层感知机 (MLP) 模型。

我们无法知道什么模型架构或学习超参数最适合这个数据集,因此我们必须进行实验和探索,找出有效的方法。

考虑到数据集很小,使用小型批次大小可能是个好主意,例如 16 或 32 行。当开始时,使用 Adam 版本的随机梯度下降是个好主意,因为它会自动调整学习率,并且在大多数数据集上效果都很好。

在我们认真评估模型之前,最好回顾学习动态并调整模型架构和学习配置,直到我们获得稳定的学习动态,然后再考虑如何充分利用模型。

我们可以通过使用简单的数据训练/测试分割并回顾 学习曲线图来做到这一点。这将帮助我们看到是否出现过拟合或欠拟合;然后我们可以相应地调整配置。

首先,我们必须确保所有输入变量都是浮点值,并将目标标签编码为整数值 0 和 1。

接下来,我们可以将数据集分割为输入和输出变量,然后分割为 67/33 的训练集和测试集。

我们可以定义一个最小的 MLP 模型。在此案例中,我们将使用一个具有 10 个节点的隐藏层和一个输出层(任意选择)。我们将在隐藏层中使用 ReLU 激活函数和“he_normal”权重初始化,因为它们结合起来是很好的实践。

模型的输出是用于二分类的 sigmoid 激活,我们将最小化二元交叉熵损失。

我们将模型训练 50 个 epoch(任意选择),批次大小为 32,因为这是一个小型数据集。

我们使用原始数据来拟合模型,我们认为这可能是个好主意,但这只是一个重要的起点。

训练结束后,我们将评估模型在测试集上的性能,并报告分类准确率作为性能指标。

最后,我们将绘制训练过程中训练集和测试集上交叉熵损失的学习曲线。

总而言之,下面列出了在纸币数据集上评估第一个 MLP 的完整示例。

运行示例首先在训练数据集上拟合模型,然后报告测试数据集上的分类准确率。

注意:由于算法或评估程序的随机性,或数值精度的差异,您的 结果可能有所不同。请考虑运行示例几次并比较平均结果。

在这种情况下,我们可以看到模型达到了 100% 的准确率。这可能表明预测问题很简单,并且/或者神经网络非常适合该问题。

然后创建训练集和测试集上损失的学习曲线图。

我们可以看到模型似乎收敛良好,并且没有显示出任何过拟合或欠拟合的迹象。

Learning Curves of Simple Multilayer Perceptron on Banknote Dataset

基于纸币数据集的简单多层感知机的学习曲线

我们第一次尝试就做得非常出色。

现在我们对简单 MLP 模型在数据集上的学习动态有了一些了解,我们可以着手开发更稳健的模型性能评估。

稳健的模型评估

k 折交叉验证过程可以提供更可靠的 MLP 性能估计,尽管它可能非常耗时。

这是因为必须拟合和评估 k 个模型。当数据集大小较小(例如纸币数据集)时,这不是问题。

我们可以使用 StratifiedKFold 类并手动枚举每个折叠,拟合模型,评估它,然后在过程结束时报告评估分数的平均值。

我们可以使用这个框架来为基础配置开发可靠的 MLP 模型性能估计,甚至可以针对一系列不同的数据准备、模型架构和学习配置进行评估。

在开始使用 k 折交叉验证 来估计性能之前,在上一节中首先了解模型在数据集上的学习动态是很重要的。如果我们直接开始调整模型,我们可能会得到不错的结果,但如果没有,我们可能不知道原因,例如模型是过拟合还是欠拟合。

如果我们再次对模型进行大的更改,最好回到并确认模型正在正确收敛。

下面列出了使用此框架评估上一节中基础 MLP 模型的完整示例。

运行示例将报告每次评估过程的模型的性能,并在运行结束时报告分类准确率的平均值和标准差。

注意:由于算法或评估程序的随机性,或数值精度的差异,您的 结果可能有所不同。请考虑运行示例几次并比较平均结果。

在这种情况下,我们可以看到 MLP 模型达到了约 99.9% 的平均准确率。

这证实了我们关于基础模型配置在此数据集上效果极佳的预期,并且该模型确实很适合这个问题,甚至可能这个问题非常容易解决。

这出乎我的意料,因为我本来预期需要进行一些数据缩放,甚至可能需要进行幂变换。

接下来,我们看看如何拟合最终模型并使用它来做出预测。

最终模型及预测

选择模型配置后,我们可以使用所有可用数据训练最终模型,并用它来对新数据进行预测。

在此案例中,我们将使用带有 dropout 和小型批次大小的模型作为我们的最终模型。

我们可以像以前一样准备数据并拟合模型,但这次是在整个数据集上,而不是在一个训练子集上。

然后,我们可以使用此模型对新数据进行预测。

首先,我们可以定义一个新数据行。

注意:我从数据集中取了第一行,期望的标签是 '0'。

然后我们可以进行预测。

然后对预测结果进行逆变换,以便我们可以使用或解释正确标签下的结果(对于此数据集,它只是一个整数)。

在这种情况下,我们将只报告预测结果。

总而言之,下面列出了拟合最终模型以用于纸币数据集并使用它对新数据进行预测的完整示例。

运行示例会将模型拟合到整个数据集,并为新数据的单行进行预测。

注意:由于算法或评估程序的随机性,或数值精度的差异,您的 结果可能有所不同。请考虑运行示例几次并比较平均结果。

在这种情况下,我们可以看到模型为输入行预测了“0”标签。

进一步阅读

如果您想深入了解,本节提供了更多关于该主题的资源。

教程

总结

在本教程中,您学习了如何为纸币二元分类数据集开发多层感知器神经网络模型。

具体来说,你学到了:

  • 如何加载和汇总纸币数据集,并利用结果来建议数据准备和模型配置。
  • 如何探索简单MLP模型在数据集上的学习动态。
  • 如何开发模型性能的稳健估计,调整模型性能并对新数据进行预测。

你有什么问题吗?
在下面的评论中提出你的问题,我会尽力回答。

16 条对开发纸币认证神经网络的回复

  1. Bilal Al Soudi 2021年3月19日 晚上7:22 #

    谢谢……

  2. Anthony The Koala 2021年4月4日 上午3:20 #

    尊敬的Jason博士,
    再次感谢您的教程。
    问题是关于测试和验证数据的学习曲线,以及模型是拟合还是过拟合。

    当测试和验证数据的学习曲线合并时,模型是否拟合?如果曲线不合并,它们是过拟合还是欠拟合?

    谢谢你,
    悉尼的Anthony

  3. Anthony The Koala 2021年4月4日 上午8:28 #

    尊敬的Jason博士,
    再次感谢。我看到了引用的文档 https://machinelearning.org.cn/learning-curves-for-diagnosing-machine-learning-model-performance/

    在欠拟合、过拟合、正确拟合和无代表性的示例中,Y 轴和 X 轴的标签是什么?X 是轮次或迭代次数,而 Y 是熵吗?

    谢谢你,
    悉尼的Anthony

    • Jason Brownlee 2021年4月5日 上午6:08 #

      在学习曲线图中,x 轴是学习迭代(通常是轮次,有时是批次),y 轴是损失。

  4. Anthony The Koala 2021年4月5日 上午6:40 #

    尊敬的Jason博士,
    谢谢你,
    悉尼的Anthony

  5. Aymeric inpong 2021年4月11日 上午5:08 #

    好帖子

    Aymeric inpong

  6. Zakir 2021年8月24日 上午4:10 #

    AttributeError: ‘Sequential’ 对象没有属性 ‘predict_classes’

    • Adrian Tam
      Adrian Tam 2021年8月24日 上午8:42 #

      这是由于 keras 的版本更改。尝试使用 predict,然后使用 numpy.argmax() 来查找类。

  7. ks 2021年10月21日 上午12:03 #

    ValueError: 分类指标无法处理二元和连续目标值的混合

    • Adrian Tam
      Adrian Tam 2021年10月22日 上午3:33 #

      这个错误从哪里来?

  8. Max Payne 2022年8月18日 上午12:44 #

    您将如何从图像中提取这 4 个特征?

留下回复

Machine Learning Mastery 是 Guiding Tech Media 的一部分,Guiding Tech Media 是一家领先的数字媒体出版商,专注于帮助人们了解技术。访问我们的公司网站以了解更多关于我们的使命和团队的信息。