隐藏 CoreML 模型 (.mlmodel) 文件

Hiding CoreML model (.mlmodel) files

我正在从事一个项目,该项目涉及向现有 iOS 应用程序添加 AI 对象检测功能。我能够训练自己的 DNN 模型并转换为 CoreML 的 .mlmodel 格式。

现在我需要将包含 .mlmodel 文件的工作转移给另一个开发人员进行集成。但是,我不希望他们在这个项目之外使用我训练有素的 .mlmodel 文件(根据合同)。有什么办法可以只 "hide" .mlmodel 文件,使它们只能用于这个特定的应用程序,不能简单地复制和保存以供其他用途吗?

我对 iOS 库和框架做了一些快速研究,但我仍然不确定这是否是我正在寻找的解决方案。

没有。一旦有人可以访问您的 mlmodel 文件或编译版本 mlmodelc,他们就可以在其他地方使用它。

例如,您可以从 App Store 下载应用,查看 IPA 文件,将其 mlmodelc 文件夹复制到您自己的应用中,然后立即开始使用该模型。

为了防止外人窃取您的模型,您可以加密模型(就像加密任何其他文件一样),但这只有在您可以隐藏解密密钥时才有效。也可以给模型添加一个自定义层,这样没有这个自定义层的代码就没用了。

但是,如果您雇用外部开发人员来开发您的应用程序,这些解决方案将不起作用,因为他们将——出于必要——需要访问这些解密密钥和源代码文件。

我不确定您希望其他开发人员做什么,但如果您不信任他们,那么:

  1. 找一个你信任的新开发者,
  2. 准备执行合同,或者
  3. 给他们一个版本的 mlmodel 文件,其中权重被随机数替换。该模型仍然有效,但会给出无意义的预测。一旦该开发人员完成了他们的工作,就用真实的模型替换模型。显然,如果他们需要将模型用于他们需要做的任何工作,这不是一个好的解决方案。