2020年，移动设备上的机器学习将不再是什么热门的新事物。在移动应用中添加某种智能已经成为一种标准做法。幸运的是，这并不意味着苹果已经停止了创新。在这篇博文中，作者将总结一下Core ML的新特性以及苹果生态系统中的其他AI和ML技术。

2020年，移动设备上的机器学习将不再是什么热门的新事物。在移动应用中添加某种智能已经成为一种标准做法。

幸运的是，这并不意味着苹果已经停止了创新。

在这篇博文中，我将总结一下Core ML的新特性以及苹果生态系统中的其他AI和ML技术。

Core ML

去年Core ML有很大的更新，但今年的改进要温和得多：几个新的层类型、对加密模型的支持，以及在CloudKit上托管模型更新的能力。

版本号似乎被去掉了。去年的更新名为Core ML 3，但是现在的名称是Core ML，没有了编号。然而，coremltools确实升到了版本4。

注意：内部的mlmodel规范版本号现在是5，所以新的模型在Netron中将显示为“Core ML v5”。

Core ML中新增的层类型

新增的层类型包括：

Convolution3DLayer、Pooling3DLayer、GlobalPooling3DLayer：这些类型在处理视频数据时特别有用，现在你是用Vision框架来做这些事情。（Core ML仍然没有明确支持1D卷积，不过你可以使用常规的2D卷积层。）
OneHotLayer：对输入进行独热编码。
ClampedReLULayer：有最大值的ReLU激活函数（可用于生成ReLU6）。
ArgSortLayer：对输入张量进行排序。它将返回排好序的索引，而不是排序后的值。Core ML中没有常规的排序层，但是你可以使用GatherLayer来重新排序argsort输出的元素。
CumSumLayer：计算输入张量的累积和。
SliceBySizeLayer：Core ML已经有了几种类型的切片层。这个允许你传入一个张量，包括切片起始索引；切片的大小是固定的。

这些层类型只能用于规范版本5或更高的版本，也就是iOS 14和macOS 11.0或更高版本。

另一个有用的改进是对以下层的8位量化操作：

InnerProductLayer
BatchedMatMulLayer

在以前的Core ML版本中，你可以量化权重，但是在加载模型时，权重会被反量化为浮点数。借助新的int8DynamicQuantize特性，权重将始终为8位整数值，实际计算也使用整数进行。

INT8的计算有可能比浮点运算快很多，但我怀疑这是只存在于CPU上的优势而不是GPU，因为GPU真的很适合浮点运算。也许即将到来的Neural Engine更新将提供INT8操作原生支持。（苹果不久前确实收购了 Xnor.ai……）

在CPU上，Core ML现在也可以使用16位浮点操作，而不是32位浮点操作（A11 Bionic及以上）。正如视频“探索Swift的数值计算”中提到的，Float16现在是一种一等Swift数据类型。由于CPU原生支持16位浮点数，Core ML的速度可以达到原来的两倍多！

注意：Core ML已经在GPU和Neural Engine上使用了Float16，所以只有在使用CPU时才会有所不同。

以下是其他一些小的变化：

UpsampleLayer现在可以有一个分数比例因子。在BILINEAR模式下，网格点采样方式（“align corners”）有了新的选择。这应该能解决我在这篇博文中指出的大部分问题。
ReorganizeDataLayerParams有一个PIXEL_SHUFFLE模式。这是另一种向上采样的方法。以前你可以使用几次置换（permute）和重塑图层重组像素图，但现在它已是内置的。
SliceStaticLayer和SliceDynamicLayer现在多了一个squeezeMasks属性，切片更有趣了。
TileLayer接受第二个输入张量，这样就可以动态指定重复的次数。

设备上的训练似乎没有什么变化：仍然只支持全连接和卷积层。CoreML.framework中的MLParameterKey类现在为RMSprop优化器提供了一个配置选项，但它目前没有在NeuralNetwork.proto中列出。它可能会在以后的测试版中添加。

下面是新增的模型类型：

VisionFeaturePrint.Object：一个针对对象检测优化过的特征提取器。
SerializedModel：不确定这个是干什么用的。这是一个“私有”定义，是“在没有通知或支持的情况下做出的修改”。也许，这是苹果在mlmodel中嵌入专有模型格式的一种方式？

在CloudKit上托管模型更新

这个新的Core ML特性允许独立于应用进行模型更新。

不必发布应用更新，现有的应用只需下载一个新版本的mlmodel文件。公平地说，这并不是一个新想法，一些第三方供应商已经为此开发了SDK。自己构建这个也不是那么困难。苹果解决方案的优势是模型托管在苹果云上。

因为你的应用中可能有多个模型，模型集合这个新概念让你可以将多个模型捆绑在一起，应用将一次性更新所有模型。你可以在CloudKit控制面板上创建这些集合。

在应用中，你可以使用MLModelCollection类来下载和管理模型更新。这个WWDC的视频展示了相关的代码片段。

为了准备一个用于部署的Core ML模型，Xcode中现在有了一个Create Model Archive按钮。这会写一个.mlarchive文件。你可以将这个版本的模型上传到你的CloudKit控制面板上，然后将其放到一个模型集合中。（看起来，mlarchive只是一个包含mlmodelc文件夹内容的zip文件。）

一个不错的特性是，你可以针对不同的用户部署不同的模型集合。例如，iPhone上的摄像头和iPad上的摄像头是不同的，所以你可能想创建一个模型的两个版本，并将一个发送给iPhone用户的应用，另一个发送给iPad用户。

你可以针对设备类别（iPhone/iPad/TV/Watch）、操作系统和版本、区域代码、语言代码和应用版本创建目标规则。

似乎没有根据其他标准将用户分组的机制，例如A/B测试模型更新，或针对特定的设备类型，如“iPhone X或更低版本”。但是，你仍然可以手动实现，方法是创建具有不同名称的集合，然后在运行时显式地请求MLModelCollection按名称获取适当的集合。

部署一个新的模型版本并非总是立即生效。应用会在某个时候检测到一个新的可用模型，自动下载它并将其放入应用的沙箱中。但是你无法控制下载的时间和方式——例如，Core ML可能会在手机未使用的时候在后台执行下载。

正因为如此，在应用中提供一个内置模型作为后备选项是一个好主意，比如一个通用模型，可以同时适用于iPhone和iPad。

尽管这是一个便捷的解决方案，而且不必自己托管模型，但请记住，应用现在使用的是CloudKit。从我的理解来看，模型集合确实会计入存储配额，而下载模型会计入网络流量配额。

另见：

注意：如果你想在设备上实现模型的个性化，没有什么简单的方法可以将其与新的CloudKit更新特性结合起来，也没有直接的方法将个性化模型学到的东西迁移到新模型中，或者以某种方式组合这些模型。

模型加密

到目前为止，有人可以很容易地窃取你的Core ML模型，并将其粘贴到他们自己的应用中。在iOS 14/macOS 11.0中，Core ML可以自动加密和解密模型，这样陌生人就不会再看到你的mlmodelc文件夹了。在新的CloudKit部署中，你可以选择使用加密，也可以选择不使用。

Xcode加密编译后的模型mlmodelc，而不是原始的mlmodel文件。模型总是以加密的形式保存在用户的设备上。只有当应用实例化模型时，Core ML才会自动解密它。该解密版本仅存在于内存中，不会作为文件存储在任何地方。

首先，你需要一个加密密钥。好消息是你不需要自己管理这个密钥！现在，在Xcode的Core ML模型查看器中有一个Create Encryption Key按钮。当你点击这个按钮时，Xcode会生成一个新的加密密钥，这个密钥与你的苹果开发账户相关联。对于这个密钥，你不需要使用CSR和Keychain Access。

这个过程将创建一个新的.mlmodelkey文件。这个密钥存储在苹果的服务器上，但你也会获得一个本地副本，用于在Xcode中加密模型。你不需要在应用中嵌入这个加密密钥，也不应该这样做！

要加密Core ML模型，可以在这个模型的编译器标识中加上--encrypt YourModel.mlmodelkey 。或者，如果你打算使用CloudKit部署，那么在创建模型存档时就需要提供加密密钥。

为了在应用实例化模型时解密，Core ML需要通过网络从苹果的服务器获取加密密钥，因此这显然需要连接网络。Core ML只需要在你第一次使用该模型时执行此操作。

当然，如果网络断开导致加密密钥未能下载，应用将无法实例化Core ML模型。为此，你应该使用新增的YourModel.load()函数。它有一个完成处理程序，让你可以响应加载错误。例如，错误代码modelKeyFetch可以告诉你Core ML无法从苹果服务器加载解密密钥。

如果你担心人们会窃取你的专有技术，那么这是个非常酷的特性，而且很容易在应用中实现。

另见：

注意：根据这个帖子介绍，加密模型不能和设备个性化共用。说得有道理。

CoreML.framework

使用Core ML模型的iOS API没有太大变化。尽管如此，还是有一些有趣的事情需要指出。

唯一的新类是MLModelCollection，它是用于CloudKit部署的。

如你所知，当你向项目中添加一个mlmodel文件时，Xcode会自动生成一个Swift或Objective-C源文件，其中有一些类可以简化模型的使用。这些生成的类有以下几个变化：

默认的init()现在已不推荐使用，因此let model = YourModel()将导致编译器发出警告。取而代之，使用YourModel(configuration:)或新的YourModel.load()方法，该方法让你可以处理模型加载错误，比如加密模型无法解密的错误。
如果你的模型是作用在图像上，那么你不一定要创建CVPixelBuffer对象，你现在可以使用CGImage或本地PNG或JPG文件的URL创建一个YourModelInput对象，并对其进行预测。但是，你无法设置cropAndScale方法或cropRect，因此，如果你想更多地控制图像大小的调整方式，那么这些新方法并不是很有用。

MLModel文档中有一个新的警告：

一次在一个线程或一个分派队列上使用一个MLModel实例。要实现这一点，你可以通过序列化模型的方法调用，或者为每个线程和分派队列创建单独的模型实例。

哎，这让我觉得有罪恶感。我以前以为，MLModel内部使用一个串行队列来处理请求，但事实并非如此，或许是已经改了。无论如何，从现在开始最好听从建议。

MLMultiArray有一个新的初始化器init(concatenating:axis:dataType:) ，它通过连接几个已有的多维数组创建一个新的多维数组。除了指定的axis，它们必须具有相同的shape，并且将沿着该axis连接到一起。它看起来像是专门为视频预测而添加的，就像Create ML中新增的Action Classifier模型一样。

注意：现在，MLMultiArrayDataType enum有了静态.float和.float64属性。我不确定这些是干什么用的，因为它已经有了.float32和.double。这是测试版的小问题？

Xcode模型查看器

Xcode现在可以显示关于模型的更多信息了，比如类标签和添加到模型中的任何自定义元数据。它还显示了模型中关于层类型的统计信息。

它提供了一个方便的交互式预览，让你可以在不运行应用的情况下摆弄模型。你可以把图片、视频或文本拖到这个预览窗口中，并立即查看模型的预测结果。非常好！

此外，现在可以在操练场上使用Core ML模型了。Xcode会自动为它生成类，你可以像往常一样使用它。这是在将模型置入应用之前的另一种交互方式。

coremltools 4

对于简单的项目来说，使用Create ML构建自己的模型会很有趣，但是，更常见的是使用TensorFlow或PyTorch训练。要在Core ML中使用这样的模型，首先需要将其转换为mlmodel文件格式。这就是coremltools的作用。

好消息：文档有了极大的改进。你应该去看看。希望他们能及时更新这个用户指南，因为之前的旧文档并不总是最新的版本。

注意：遗憾的是，Jupyter notebook的例子消失不见了。这些例子现在是用户指南的一部分，但不再是Jupyter notebook的一部分。

模型转换的方式变化较大。旧有的神经网络转换器已经弃用多时，取而代之的是一种更新、更灵活的设计。

转换器现在有三种类型：

用于TensorFlow (1.x和2.x)、tf.keras和PyTorch的现代转换器。所有这些转换器都构建在同一个技术栈上，使用了一种称为MIL或模型中间语言的东西。对于这类模型，不需要再使用tfcoreml或onnx-coreml。
用于Keras 1.x、Caffe和ONNX的旧转换器。这些都是有专门用途的转换器。它们将不会再进一步开发，而只会进行Bug修复。不建议再使用ONNX转换PyTorch模型！
用于非神经网络模型（如scikit-learn和XGBoost）的转换器。

要转换一个TensorFlow 1.x或2.x、PyTorch或tf.keras模型，需要使用新增的统一转换API，如下所示：

import coremltools as ct
class_labels = [ "cat", "dog" ]
image_input = ct.ImageType(shape=(1, 224, 224, 3),
                           bias=[-1, -1, -1],
                           scale=2/255.)
model = ct.convert(
    keras_model, 
    inputs=[ image_input ],
    classifier_config=ct.ClassifierConfig(class_labels)
)
model.save("YourModel.mlmodel")

ct.convert()函数检查模型文件以确定其格式，并自动选择适当的转换器。参数与之前略有不同：预处理参数使用ImageType对象传入，分类器标签使用ClassifierConfig对象，等等。
这个新增的转换API将模型转换为称为MIL的中间表示。现在有TensorFlow 1.x到MIL、TensorFlow 2.x到MIL——包括tf.keras——以及PyTorch到MIL的转换器。如果某个时候，一个新的深度学习框架流行起来，它也可以获得自己的MIL转换器。

一旦模型转换为MIL格式，就可以根据一般规则进行优化了，比如剥离不必要的操作或将不同的层融合在一起。最后，将其从MIL转换为mlmodel格式。

我还没有详细地了解它，但这种新方法使我相信，coremltools 4生成的mlmodel文件比以前要高效得多，特别是对于TF 2.x图。

我喜欢MIL的一点是，它允许你告诉转换器如何处理它还无法识别的层。如果你的模型有一个Core ML不直接支持的层，那么你可以将其拆分为更原始的MIL操作，例如矩阵乘法或其他运算。

然后，只要遇到该类型的层，转换器就可以使用这种所谓的“复合操作”。这比使用自定义层实现不受支持的操作要容易得多，尽管仍然可以这样做。文档中有一个很好的例子，演示如何使用这种复合操作。

另见：

使用Core ML转换器在设备上获取模型（WWDC视频）
coremltools文档

其他使用ML的苹果框架

在iOS和macOS SDK中有几个高级框架可以执行机器学习相关的任务。让我们快速看看这里有什么新东西。

Vision

计算机视觉框架Vision增加了很多新功能。

Vision已经有了人脸、人脸特征和人体检测模型。这个新版本增加了以下内容：

手势检测（VNDetectHumanHandPoseRequest）
多人全身位姿检测（VNDetectHumanBodyPoseRequest）

实际上，苹果在操作系统中包含位姿估计是很酷的。有几种用于此目的的开源模型，但它们不是真有那么好，或者很慢，而商业解决方案则很贵。现在，你可以免费获得高品质的身体位姿检测了！

不仅仅是检查静态图像，现在更注重检测视频中的东西，包括离线的和来自实时摄像头的。方便起见，你可以直接在照相机的请求处理程序中使用CMSampleBuffer对象。

新增类VNStatefulRequest是VNImageBasedRequest的一个子类。对于你要找的东西，它会随着时间的推移积累证据。与典型的VNImageBasedRequest不同，你可以在多个帧上重用同一个有状态的请求。它对视频的每N帧执行一次分析操作。

一旦观察到你要找的东西，它就会使用VNObservation对象调用完成处理程序。它现在有一个timeRange属性，可以告诉你观察在视频中的开始和停止时间。

你不直接使用VNStatefulRequest ，这是一个抽象基类，目前只有用于轨迹检测的VNDetectTrajectoriesRequest这一个子类。它可以沿着抛物线路径检测形状，比如被踢或被扔的球。（这似乎是目前唯一内置的视频任务。）

要对视频执行脱机分析，可以使用VNVideoProcessor。该对象接受一个本地电影文件的URL，并每N帧或秒执行一个或多个Vision请求。

对于分析视频，光流法是一种重要的传统计算机视觉技术。Vision现在有一个VNGenerateOpticalFlowRequest类，它计算每个像素从这一帧移动到下一帧的方向（密集光流）。输出是一个VNPixelBufferObservation类，它包含一个新图像，每个像素有两个32位或16位浮点数。

还有一个新类VNDetectContoursRequest，用于检测图像中物体的轮廓。这些轮廓将作为向量路径返回。VNGeometryUtils有辅助函数对检测到的轮廓进行后处理，如将其简化为基本的几何形状。

Vision中的最后一个新特性是内置特性提取器VisionFeaturePrint的一个新变体。iOS已经自带了VisionFeaturePrint.Scene，可以用于图像分类器。现在又有了一种新的VisionFeaturePrint.Object模型，专门针对提取用于对象检测的特征做过优化。

对299×299的输入图像，它会输出两个多维数组(288、35、35)和(768、17、17)。这些还不是边框预测，只是“原始”特性。一个完整的对象检测器仍然需要添加逻辑来将这些特性转换为边框和类标签。当你使用迁移学习训练一个对象检测器时，Create ML就可以做到这一点。

另见：

自然语言处理

对于NLP任务，可以使用自然语言框架。它可以与在Create ML中训练的模型紧密协同。

今年的新特性只有几项：

NLTagger和NLModel现在可以识别出多个标签，并预测它们的可信度。之前，只能得到得分最高的标签。
句子嵌入。词嵌入以前就可以使用了，但NLEmbedding现在也支持句子。

句子嵌入使用一个内置的神经网络将整个句子编码成一个512维的向量，以便捕获单词在句子中的上下文（这是词嵌入单独无法做到的）。

另见：

利用语言处理让App更智能（WWDC视频）

语音&声响分析

这些框架似乎没什么变化。

训练模型

我们一直都是用苹果API（从iOS 11.3开始）和Metal Performance Shaders训练模型。随着时间的推移，苹果添加了一些新的训练API，今年又增加了一些——根据我的计算，我们现在有7个不同的API可以用于在iOS和/或macOS上训练神经网络！

目前，你可以在iOS和macOS上使用这些苹果API来训练机器学习模型，特别是神经网络：

使用Core ML进行设备端训练。
Create ML：你可能知道这是个App，但这也是macOS上的一个框架。
Metal Performance Shaders：用于在GPU上进行推理和训练。这实际上是两种不同的API，如果你不熟悉Metal，那么它们并不是特别容易使用。现在还有一个新的“Metal Performance Shaders Graph”框架，它似乎是这些旧API的替代品。
BNNS：Accelerate框架的一部分。之前，BNNS只有做推理的例程，但今年它增加了对训练的支持。
ML Compute：一个看起来让人兴奋的全新框架。
Turi Create：基本上是Create ML的Python版本，虽然它看起来仍在维护中，但最近似乎没有得到很多人的喜爱。

让我们具体看一下这些API都提供了什么新特性。

使用Core ML进行设备端训练

似乎没有什么变化。可能有更多的层类型变为可更新的了，但我没有看到任何相关文档。

以后的测试版可能还会增加一个新特性——RMSprop优化器，但当前的测试版还没有提供。

Create ML

Create ML一开始只是一个只能在macOS上使用的框架。你可以在Swift Playground中使用这个训练模型，只需几行代码。

去年，Create ML变成了一个有诸多限制的应用，我很高兴地看到，这个应用在今年有了很大的改进。它仍然是一个框架，所以你仍然可以在代码中使用它。实际上，该应用只是一个裹着框架的便捷的GUI。

在Create ML应用的前一个版本中，一次只能训练一个模型。如果你想调整一些东西，就必须从头开始训练，这可能会花费很长时间。

有了Xcode 12提供的新版本，你就可以暂停训练并稍后继续，保存模型检查点（快照），并预览模型的运行情况。你可以更好地控制训练过程。这个更新使Create ML应用真正变得很有用！

在CreateML.framework中，还有用于设置训练会话、处理模型检查点等的新API。我估计大多数人都只使用Create ML应用，但是看到框架暴露这个功能还是很让人高兴。

你可以用Create ML（框架和应用）做一些以前没做过的事：

图像及视频风格转换
视频中人类动作分类
用于对象检测的学习迁移
带动态词嵌入的词标记迁移学习

让我们仔细看一下新的动作分类模型。它使用了Vision的人体姿态模型。动作分类器是一种神经网络，它接受形状为(window_size, 3,18)的输入，其中第一个维度是以帧数（通常是2秒的视频）表示的窗口大小，(3,18)是位姿关键点。

神经网络不使用递归层，而是使用一维卷积。那似乎是STGCN或时空图形卷积网络的变体，是一种针对时序预测而设计的模型。在应用中使用这些模型，你不需要考虑这些细节，但我总是喜欢窥探底层的逻辑。

对于对象检测模型，你可以选择基于TinyYOLOv2训练一个完整的网络，或者新的转移学习模式，它使用了新的特征提取器VisionFeaturePrint.Object。这个模型的其他部分看起来仍然像YOLO和SSD，但是得益于转移学习，它的训练速度应该比完全基于YOLO的模型更快。

另见：

Metal Performance Shaders

Metal Performance Shaders（MPS）是一个具有便捷的Metal计算内核的框架，主要用于图像处理，但自2016年起就支持神经网络了。我已经在博客上做了大量的探讨。

如今，大多数人都使用Core ML而不是MPS。当然，在GPU上运行模型时，Core ML在底层仍然是使用MPS。但是，你也可以直接使用MPS，特别是如果你想自己进行训练的话。（不过，现在有一个新的ML Compute框架，你应该使用它作为替代，见下文。）

今年，MPSCNN中没有太多新东西，只有一些小的优化。

新增类包括用于边缘检测的MPSImageCanny和用于线段检测的MPSImageEDLines。它们对计算机视觉任务非常有用。

其他值得注意的变化有：

MPSCNNConvolutionDataSource有一个新的kernelWeightsDataType属性，这样，权重的数据类型就不必与执行卷积时的数据类型相同。有趣的是，权重不能是INT8，即使Core ML现在允许某些层使用。
如果从kernelWeightsDataType返回.float32，那么卷积和全连接层将使用32位浮点数而不是16位浮点数运行。以前，它们只能工作在16位上。
损失函数现在可以reduceAcrossBatch。

如果Metal没有让你望而却步，那么使用MPSCNN也没问题，但是，现在有一个新框架，让你可以更容易地创建和运行MPS Graph这样的图。

注意：WWDC视频还说MPSNDArray是新的，但是这个API去年就已经引入。这是一个比MPSImage更灵活的数据结构，因为模型中的张量不一定都是图像。

新的Metal Performance Shaders Graph

MPS提供MPSNNGraph API已经有一段时间了，但是这些图仅限于描述神经网络。然而，并不是所有的图都需要是神经网络，这就是Metal Performance Shaders Graph的用途。

你可以使用这个新框架来构建GPU计算操作的通用图。MPS图独立于常规的Metal Performance Shaders框架，但确实是以它为基础。

旧有的MPSNNGraph API有一个限制，就是不能将自己的自定义操作放入图中，但这个新的Graph框架更加灵活。虽然你似乎不会真的添加自己的Metal计算内核，但你必须使用现有的原语来表示所有计算。

幸运的是，MPSGraph编译器可以将这些原语融合到一个计算内核中，使其在GPU上尽可能高效地运行。但是，如果你有一个操作不可能使用所提供的原语实现，或者真的很难实现，那么就没法这样做了。我真是不明白，为什么苹果在引入这样一个新的API时总是忽略进行完全定制的能力？！。

新的MPSGraph是一个相当轻量级的结构，它描述了一组MPSGraphTensors如何通过MPSGraphTensors（包含操作结果）相互关联。你还可以定义控制依赖项，强制某些节点先于其他节点运行。把图设置好之后，运行它或将其编码到命令缓冲区中，然后等待结果。

MPSGraph有一大堆实例方法，可以向图中添加各种数学和神经网络操作。

它还可以进行训练，包括向图中添加一个损失操作，然后对所有层按相反的顺序做“梯度”操作，就像以前的MPSNNGraph一样。方便起见，它还提供了自动微分模式，使MPSGraph可以自动将梯度操作放入图中，节省了大量工作。

我很高兴他们为构建这些计算图创建了一个新的、干净的API。与旧有的图API相比，它显然更容易使用——你不再需要是Metal方面的专家。它让我想起了很多TensorFlow 1.x图，它所带来的额外的好处是，图经过了积极地优化，尽可能地降低了开销。不过，无法在图中插入任意计算核，这很糟糕。

另见：

使用Metal Performance Shaders Graph构建自定义ML模型（WWDC视频）
向Shader Graph中添加自定义函数

BNNS（Basic Neural Network Subroutines）

当Core ML在CPU上运行时，它使用BNNS，这是Accelerate框架的一部分。我在以前的一篇博文中探讨过BNNS。现在，那些BNNS函数大多已被弃用，并被一套全新的函数所取代。

以前，它只支持全连接层、卷积、池和激活函数。通过这次更新，BNNS已经支持在训练时使用n维数组、来自Core ML的几乎所有的层类型，以及这些层的旧版本——包括目前Core ML无法训练的层，如LSTM。

同样值得注意的是有一个multihead attention层，这在诸如BERT这样的Transformer模型中非常常见。嗯，张量缩并也很有趣。

你自己可能不会使用这些BNNS函数，就像你不会在GPU上使用MPS训练一样。相反，现在有一个更高级的框架ML Compute，它抽象出了正在使用的处理器。ML Compute底层使用了BNNS和MPS，但作为开发人员，你不需要考虑这些细节。

另见：BNNS框架文档。

新的ML Compute

ML Compute是一个用于在CPU或GPU（但显然不是Neural Engine）上训练神经网络的新框架。在拥有多个GPU的Mac Pro上，它可以自动使用所有GPU进行训练。

苹果又提供一个训练框架让我有点惊讶，但它确实会让事情变得更简单，因为它隐藏了BNNS和MPS的底层细节——也许将来还可以用在Neural Engine上。

ML Compute的有趣之处在于它也可以在iOS上运行，而不只是在Mac上。奇怪的是，在所有这些地方都没有提到Core ML——ML Compute似乎是一个完全独立的东西。你不能使用它来创建ML Compute模型。

据我所知，ML Compute的目标主要是促进第三方深度学习工具。你不会希望编写代码来直接使用ML Compute——我认为，其假设（希望？）是TensorFlow等工具也将开始使用它来实现Mac上的硬件加速训练。

它提供的层似乎与BNNS相同。把这些层放在一个图中，然后执行这个图。（这里没有“动态图模式”。）

要创建一个图，需要实例化一个MLCGraph对象并向其添加节点。节点是MLCLayer的一个子类，通过MLCTensor对象连接到其他节点，它们是其他层的输出。

有趣的是，拆分、连接、重塑（reshape）和转置操作不是单独的层类型，而是图本身的操作。

一个很酷的调试特性是summarizedDOTDescription，它返回图的DOT描述，然后可以用Graphviz或OmniGraffle程序来绘制。（这也是Keras生成模型图的方式。)

ML Compute区分了推理图和训练图。后者有额外的节点，比如损失层和优化器。

似乎没有方法可以创建自己的自定义层，因此，你只能使用ML Compute提供的层类型。

奇怪的是，WWDC没有关于这个新框架的环节，文档也非常少。总之，我一直在关注它，因为它似乎是可以用于在苹果设备上进行训练的API。

另见：ML Compute框架文档。

总结

Core ML有一些有用的新特性，比如自动模型更新和加密。实际上，我们并不需要太多新的层类型，因为去年添加的层基本上涵盖了所有内容。我对这个更新很满意。

coremltools 4是一个重大更新，它提供了新的转换器架构，并提供了TensorFlow 2和PyTorch原生支持。我很高兴，我们不需要再通过ONNX转换PyTorch模型了！

Vision新提供了一堆很酷的东西。我很高兴苹果添加视频分析功能。虽然可以在每个视频帧上单独运行ML，但这样做会忽略时间维度。由于移动设备的速度已经足够快，所以可以实时对视频数据执行ML。我希望，在不久的将来，我们可以看到Vision在计算机视觉研究中发挥更加突出的作用。

至于训练……我不知道为什么我们需要7种不同的API。我认为，在新API解决所有问题之前，苹果公司不会弃用旧API。关于ML Compute，我们还不太了解——但我写这篇文章时，它还在beta 1阶段，所以谁知道呢……

如果你喜欢这篇博文，可以通过Twitter（@mhollemans）或邮箱（matt@machinethink.net）联系我。

查看英文原文：

https://machinethink.net/blog/new-in-apple-machine-learning-2020/

创作场景

2020 年，苹果的 AI 还有创新吗？