作者：张伟辰(璟铭)

随着手机性能的不断提升，在手机端进行复杂的 AI 计算已经成为各大厂商的核心发展方向，在此之上产生了大量的端智能应用。这种端侧 AI 计算的模式，使得大量牵涉时效性、成本和隐私考虑的场景实现变成了可能。在这里，我们以广泛使用的文字识别技术（OCR）为例，介绍一下蚂蚁自研移动端 OCR 技术（xNN-OCR）。

背景介绍

文字识别技术是计算机视觉领域中历史悠久、应用广泛的一个研究方向，特别是随着深度学习技术的发展，其能力空间不断扩大。相比云端计算方式，移动端 OCR 算法可在离线情况下完成图片中文字提取，对于实时性、隐私保护和成本要求高的场景，有着很大的应用价值。另一方面，基于深度学习的 OCR 模型越来越复杂，通常具有几十 M 的参数量以及几百 GFlops 的计算量，如何在手机有限的计算资源下，完成 OCR 模型运行是一个极具挑战的任务。在支付宝中，我们结合自研的端侧推理引擎 xNN 和应用算法的深度优化，研发了又小、又快、又准的 xNN-OCR 技术产品，从 2018 年上线到银行卡号识别场景开始，陆续支撑了数十个核心业务的技术升级。本文我们将给大家完整的展开 xNN-OCR 的技术演进和能力开放情况。

xNN-OCR 技术演进

一个端侧模型研发需要经历下面的几个流程：训练数据获取和标注、网络结构设计、训练调参、端侧移植和端侧部署，各个环节相互关联也相互影响。在基础算法方面，xNN-OCR 经历了小字库、大字库到基于异构计算的三个模型研发阶段。我们将从核心的数据、网络设计和模型压缩层面分别介绍最新成果。

数据生成

数据像弹药一样，很大程度决定了模型的效果，特别是文字识别场景。中文的组合千变万化，很多场景很难获取到足够的实际数据。针对这个问题，我们探索了基于 GAN 技术的文字生成技术。在网络设计上，背景提取、文本提取和字体提取的三个编码器分别提取出对应的特征信息，通过字体迁移和背景融合，完成从源文本内容到目标字体与背景的融合。在训练过程中，除了常规的生成和对抗损失外，还加入了识别损失函数监督合成的内容是否正确。同时，我们为了将已有的真实数据使用起来，在训练链路加入 Cycle-Path，提升整体数据生成效果。采用这种方式合成的卡证数据，只采用原本 10%的数据量，就可以达到采用 100%真实数据的识别精度。

xNN-OCR 网络架构

有了数据之后，下一步就是模型设计，这里主要介绍 xNN-OCR 算法中主要的文字行检测、文字行识别和结构化三个部分。

文本检测算法

相比于通用物体检测任务，文本检测任务具有大宽高比和倾斜框的特点。针对这两个问题，传统的 Anchor-based 的检测方案需要配置非常大量的 Anchor，带来计算量的上升。因此我们设计了一个轻量级的检测网络，其中骨干网络基于 ShuffleNet 的设计思想，采用多层 Shuffle 结构，网络头部采用 Pixel-based 的密集预测方式，每个输出图的像素点都会输出类别和框位置回归，经过融合后处理之后得到最终检测结果。为了适应端侧计算环境，通常图输入的分辨率不会过大，带来小目标易丢失和长目标边界不准的问题。我们在训练时采用 Instance-balancing + OHEM 的方式解决小目标丢失问题，在预测时采用加权融合 NMS 的方式解决边框预测不准的问题，取得了性能和精度的大幅度提升。

文本识别算法

文字行检测后，对于内容的识别通常采用 CRNN 的结构，在之前工作的基础上，我们针对骨干网络和网络 head 设计做了进一步的升级。为了得到一个高性能轻量骨干网络，我们通过 NAS 面向文本识别场景设计了相应的搜索策略，搜索出目标数据集上性价比最高的网络结构参数。对于 CRNN 的结构，我们发现模型 head 部分计算量非常大，占到整体计算量 50%以上，这主要是由于 Softmax 分类的 One-hot 的稀疏编码方式导致。我们通过将稠密的 Hamming 编码方式与 CRNN 模型结合，Head 部分耗时相比原本的 Softmax 分类方案下降约 70%。

文字结构化

文字结构化指的是输出文字对应的结构信息，例如卡证场景中将 OCR 算法结果整理为 Key-Value 的输出格式。传统的结构化方法通常是基于文本位置和识别结果设计规则，调试较为复杂，并且工程上需要针对不同卡证分别开发处理逻辑，部署和维护成本高。我们从文本行检测入手，提出 Instance 检测算法进行卡证结构化。简单来讲，就是在检测网络头部增加文本框的类别信息进行学习，在结构化时直接将文本框的识别内容与类别对应起来。这种方法可以节省识别运算耗时，简化上线调试部署流程，同时因为模型学习了字段间的隐含关系，提升了整体的识别精度。

模型压缩

为了提升端侧模型研发性能和效果，xNN 在之前基于已有结构进行轻量化的基础上，自研了 xNAS 算法工具，提供模型结构搜索能力。在主流 NAS 搜索框架基础上，xNAS 扩展了端侧模型关心的计算量和硬件耗时等因素，结合超参搜索(HPO)、Multi-Trial NAS、One-Shot NAS 等算法，搜索最优的移动端模型结构。在 OCR 的场景，我们重点针对识别网络使用了 NAS 的方案，通过对每个 Channel 和卷积层数进行搜索，使模型降低了 70%的计算量并有 2%的精度提升。

在模型压缩方面，剪枝、浮点量化、定点化功能对于推理性能提升至关重要，特别是定点化能力，可有效降低模型尺寸和运行耗时。为解决在 OCR 场景中定点参数难确定导致的精度问题，xNN 结合 NAS 思想提出 qNAS 算法，有效提升了定点化精度。我们对文本检测和识别模型进行了 qNAS 量化训练，在精度下降不到 1%的情况下，模型包大小下降至原本 1/4 左右，端侧 CPU 上运算时耗下降约 50%。

xNN-OCR 性能精度

在基础模型研发的基础上，我们将能力逐步的覆盖到了大多数 OCR 应用场景，包括通用 OCR 识别以及各种卡证类识别，在保证较高精度的同时，可以在移动端计算平台上达到近似实时运算的性能，具体指标如下(耗时为高通 855 上 CPU 单线程运算耗时)：

能力开放

xNN-OCR 作为蚂蚁自研的移动端 OCR 技术，实现了 OCR 识别像扫码一样顺畅。目前在支付宝端内，已经广泛使用在安全风控、证件上传和数字金融等核心应用场景。为了让更多用户和外部业务能够使用到 xNN-OCR，我们在支付宝端内通过小程序插件的方式提供给外部开发者使用。支付宝外的用户可以用过蚂蚁 mPaaS 产品和阿里云视觉开放平台以离线 SDK 形式接入。

支付宝小程序接入可参考：https://forum.alipay.com/mini-app/post/29301014支付宝外使用可通过钉钉群（23124039）咨询 mPaaS 产品或者访问阿里云视觉开放平台接入离线 SDK。

为了方便广大开发者体验，我们将已有的插件聚合在了“小程序体验中心“，可通过支付宝扫描下面的二维码进行体验。

关注【阿里巴巴移动技术】，每周 3 篇移动技术实践 &干货给你思考！