基于深度CNN和极限学习机相结合的实时文档分类

摘 要 提出一种文档图像实时分类训练和测试的方法。在实际应用中，数据训练的精确性和高效性在文档 图像识别中起着关键的作用。现有的深度学习方法不能满足此要求，因为需要大量的时间用于训练和微调深层 次的网络架构。针对此问题，提出一种基于计算机视觉的新方法：第一阶段训练深度网络，作为特征提取器；第二 阶段用极限学习机（ＥＬＭ）用于分类。该方法的性能优于目前最先进的基于深度学习的相关方法，在 Ｔｏｂａｃｃｏ ３４８２数据集上的最终准确率为 ８３．４５％。与之前基于卷积神经网络（ＣＮＮ）的方法相比，相对误差降低了 ２６％。 ＥＬＭ的训练时间仅为 １．１５６秒，对 ２４８２张图像的整体预测时间是 ３．０８３秒。因此，该文档分类方法适合于大规 模实时应用。
Ｋｅｙｗｏｒｄｓ Ｄｏｃｕｍｅｎｔｉｍａｇｅｃｌａｓｓｉｆｉｃａｔｉｏｎ ＣＮＮ Ｍｉｇｒａｔｉｏｎｌｅａｒｎｉｎｇ
０ 引 言
如今，商 业 文 件 （见 图 １）通 常 由 文 档 分 析 系 统 （ＤＡＳ）进行处理，以减少工作人员的工作量。ＤＡＳ的
一项重要任务是对文档进行分类，即确定文档所指的 业务流程的类型。典型的文档类是发票、地址变更或 索赔等。文档分类方法可分为基于图像［１－６］和基于内 容的方法 ［７－８］。ＤＡＳ选 取 哪 一 种 方 法 更 合 适，通 常 取 决于用户处理的文档。像通常的字母一样，自由格式
关键词 文档图像分类 ＣＮＮ 迁移学习
中图分类号 ＴＰ３９１．４１ 文献标识码 Ａ ＤＯＩ：１０．３９６９／ｊ．ｉｓｓｎ．１０００３８６ｘ．２０１９．０３．０３２
ＲＥＡＬＴＩＭＥＤＯＣＵＭＥＮＴＣＬＡＳＳＩＦＩＣＡＴＩＯＮ ＢＡＳＥＤ ＯＮ ＤＥＥＰＣＮＮＡＮＤＥＸＴＲＥＭＥＬＥＡＲＮＩＮＧ ＭＡＣＨＩＮＥ
ＹａｎＨｅ１，２ ＷａｎｇＰｅｎｇ１ ＤｏｎｇＹｉｎｇｙａｎ１ ＬｕｏＣｈｅｎｇ１ ＬｉＨｕａｎ１
１（ＣｏｌｌｅｇｅｏｆＣｏｍｐｕｔｅｒＳｃｉｅｎｃｅ，ＣｈｏｎｇｑｉｎｇＵｎｉｖｅｒｓｉｔｙｏｆＴｅａｃｈｎｏｌｏｇｙ，Ｃｈｏｎｇｑｉｎｇ４０１３２０，Ｃｈｉｎａ） ２（ＡｒｔｉｆｉｃｉａｌＩｎｔｅｌｌｉｇｅｎｃｅＣｏｌｌｅｇｅ，ＣｈｏｎｇｑｉｎｇＵｎｉｖｅｒｓｉｔｙｏｆＴｅａｃｈｎｏｌｏｇｙ，Ｃｈｏｎｇｑｉｎｇ４０１１４７，Ｃｈｉｎａ）
的文档通常需要基于内容的分类，而在不同布局中包 含相同 文 本 的 表 单 则 可 以 通 过 基 于 图 像 的 方 法 来 区分。
１７５
图 １ 来自不同类别的 Ｔｏｂａｃｃｏ３４８２数据集的样本图像
然而，并不总是事先知道文档属于什么类别，这就 是为什么在基于图像的方法和基于内容的方法之间很 难选择的原因。一般来说，基于图像的方法是大多数 学者首选的方法，因为它直接工作在数字图像上。由 于文档图像类的多样性，存在高类内方差和低类间方 差的类，分别如图 ２和图 ３所示。因此，很难找到用于 文档图像分类的人工特征提取方法。
Ａｂｓｔｒａｃｔ Ｔｈｉｓｐａｐｅｒｐｒｅｓｅｎｔｅｄａｒｅａｌｔｉｍｅｔｒａｉｎｉｎｇａｎｄｔｅｓｔｉｎｇｍｅｔｈｏｄｆｏｒｄｏｃｕｍｅｎｔｉｍａｇｅｃｌａｓｓｉｆｉｃａｔｉｏｎ．Ｉｎｐｒａｃｔｉｃａｌ ａｐｐｌｉｃａｔｉｏｎｓ，ｔｈｅａｃｃｕｒａｃｙａｎｄｅｆｆｉｃｉｅｎｃｙｏｆｄａｔａｔｒａｉｎｉｎｇｐｌａｙａｋｅｙｒｏｌｅｉｎｄｏｃｕｍｅｎｔｉｍａｇｅｒｅｃｏｇｎｉｔｉｏｎ．Ｔｈｅｅｘｉｓｔｉｎｇ ｄｅｅｐｌｅａｒｎｉｎｇｍｅｔｈｏｄｓｃａｎｎｏｔｍｅｅｔｔｈｉｓｒｅｑｕｉｒｅｍｅｎｔ，ｂｅｃａｕｓｅｔｈｅｙｎｅｅｄａｌｏｔｏｆｔｉｍｅｔｏｔｒａｉｎａｎｄｆｉｎｅｔｕｎｅｔｈｅｄｅｅｐ ｎｅｔｗｏｒｋａｒｃｈｉｔｅｃｔｕｒｅ．Ｔｏｓｏｌｖｅｔｈｉｓｐｒｏｂｌｅｍ，ｗｅｐｒｏｐｏｓｅｄａｎｅｗ ｍｅｔｈｏｄｂａｓｅｄｏｎｃｏｍｐｕｔｅｒｖｉｓｉｏｎ．Ｔｈｅｍｅｔｈｏｄｗａｓ ｄｉｖｉｄｅｄｉｎｔｏｔｗｏｓｔｅｐｓ：ｔｈｅｄｅｐｔｈｎｅｔｗｏｒｋｗａｓｔｒａｉｎｅｄａｓａｆｅａｔｕｒｅｅｘｔｒａｃｔｏｒ；ｗｅｕｓｅｄｔｈｅｅｘｔｒｅｍｅｌｅａｒｎｉｎｇｍａｃｈｉｎｅ （ＥＬＭ）ｆｏｒｃｌａｓｓｉｆｉｃａｔｉｏｎ．Ｔｈｅｐｅｒｆｏｒｍａｎｃｅｏｆｔｈｉｓｍｅｔｈｏｄｉｓｓｕｐｅｒｉｏｒｔｏｔｈｅａｄｖａｎｃｅｄｍｅｔｈｏｄｓｂａｓｅｄｏｎｄｅｅｐｌｅａｒｎｉｎｇ． ＴｈｅｆｉｎａｌａｃｃｕｒａｃｙｏｆｔｈｉｓｍｅｔｈｏｄｏｎＴｏｂａｃｃｏ３４８２ｄａｔａｓｅｔｉｓ８３．４５％．Ｃｏｍｐａｒｅｄｗｉｔｈｔｈｅｍｅｔｈｏｄｂａｓｅｄｏｎｃｏｎｖｏｌｕｔｉｏｎ ｎｅｕｒａｌｎｅｔｗｏｒｋ，ｔｈｅｒｅｌａｔｉｖｅｅｒｒｏｒｉｓｒｅｄｕｃｅｄｂｙ２６％．ＴｈｅｔｒａｉｎｉｎｇｔｉｍｅｏｆＥＬＭ ｉｓｏｎｌｙ１．１５６ｓ，ａｎｄｔｈｅｏｖｅｒａｌｌ ｐｒｅｄｉｃｔｉｏｎｔｉｍｅｏｆ２４８２ｉｍａｇｅｓｉｓ３．０８３ｓ．Ｔｈｅｒｅｆｏｒｅ，ｔｈｅｍｅｔｈｏｄｉｓｓｕｉｔａｂｌｅｆｏｒｌａｒｇｅｓｃａｌｅｒｅａｌｔｉｍｅａｐｐｌｉｃａｔｉｏｎｓ．
第 ３６卷第 ３期 ２０１９年 ３月
计算机应用与软件 ＣｏｍｐｕｔｅｒＡｐｐｌｉｃａｔｉｏｎｓａｎｄＳｏｆｔｗａｒｅ
Ｖｏｌ３６ Ｎｏ．３ Ｍａｒ．２０１９
基于深度 ＣＮＮ和极限学习机相结合的实时文档分类
闫 河１，２ 王 鹏１ 董莺艳１ 罗 成１ 李 焕１
１（重庆理工大学计算机科学与工程学院 重庆 ４０１３２０） ２（重庆理工大学两江人工智能学院 重庆 ４０１１４７）
收稿日期：２０１８－１０－０９。国家自然科学基金面上项目（６１１７３１８４）；重庆市自然科学基金项目（ｃｓｔｃ２０１８ｊｃｙｊＡＸ０６９４）。闫河，教 授，主研领域：深度学习，图像识别。王鹏，硕士生。董莺艳，硕士生。罗成，硕士生。李焕，学习机相结合的实时文档分类
图 ２ Ｔｏｂａｃｃｏ３４８２数据集广告类的文档， 显示了较高的类内差异