arXiv:2501.00528cs.LGcs.AI2025-01被引 1

PyMilo让机器学习模型导出更安全透明,告别pickle的隐患。

PyMilo: A Python Library for ML I/O

  • 用非可执行的透明格式存储模型,避免二进制风险。
  • 支持模型在生产环境的可靠部署与反序列化。
  • 适合需要安全交换和稳定部署模型的开发者。

PyMilo是一个开源Python库,旨在解决现有机器学习(ML)模型存储格式(如pickle等二进制格式)存在的可靠性、安全性和透明性问题。该工具通过将训练好的模型以透明的非可执行格式序列化,实现模型的安全、高效传输与部署。相比传统方式,PyMilo简化了模型从开发到生产环境的全流程管理,提供端到端的预训练模型导入导出解决方案,显著提升模型在实际应用中的可维护性与安全性。

原文摘要 · Abstract (English)

PyMilo is an open-source Python package that addresses the limitations of existing Machine Learning (ML) model storage formats by providing a transparent, reliable, and safe method for exporting and deploying trained models. Current formats, such as pickle and other binary formats, have significant problems, such as reliability, safety, and transparency issues. In contrast, PyMilo serializes ML models in a transparent non-executable format, enabling straightforward and safe model exchange, while also facilitating the deserialization and deployment of exported models in production environments. This package aims to provide a seamless, end-to-end solution for the exportation and importation of pre-trained ML models, which simplifies the model development and deployment pipeline.

模型部署Python库ML I/O

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。