打开链接点亮社区Star，照亮技术的前进之路。每一个点赞，都是社区技术大佬前进的动力

Github 地址： https://github.com/secretflow/sml

我们很高兴地宣布，隐语 SML 0.1.0 版本正式发布！

SML 库目前支持约数十种核心算法，涵盖了机器学习的主要领域，如监督学习（线性模型、支持向量机等）、无监督学习（KMeans聚类算法，PCA等降维算法）、数据预处理（标准化、特征选择等）、模型评估（各种分类、回归指标）。

版本链接：https://pypi.org/project/sf-sml/#history

什么是 SML？

SML 是一个基于 Python 的机器学习模块，它利用 JAX 实现算法，并借助 SPU (SecretFlow Processing Unit) 的能力进行安全的训练和推理。

专注于隐私保护机器学习场景，致力于构建一个安全版本的 scikit-learn。

为什么需要一个“隐私版 scikit-learn”？

scikit-learn 是事实上的传统机器学习标准库，但它有一个前提假设：

• 数据集中在一处

• 运行在 NumPy / SciPy 的中心化计算环境中

• 不考虑多方协作与数据隐私

而在真实世界中，越来越多场景是：

• 数据分散在多方

• 原始数据不能共享

• 合规要求高（金融、医疗、政企等）

直接将 scikit-learn 的实现“翻译”为 MPC 版本，会遇到一些问题：

• 浮点数在 MPC 下精度和性能不可控

• 大量 CPU 友好算子 并不适合安全计算

• 简单移植会导致效率极低，无法落地

SML 的愿景是建立一个通用的隐私保护机器学习 (PPML) 库，旨在成为安全版本的 scikit-learn

支持的算法

在首个开源版本中，SML 已支持 数十种核心算法，覆盖传统机器学习的主要领域：

监督学习

• 线性模型（Logistic / Ridge / GLM 等）

• 支持向量机（SVC）

• KNN

• Gaussian Naive Bayes

• 决策树（基于 GTree 的 MPC 实现）

• 集成模型：Random Forest、AdaBoost

无监督学习与降维

• KMeans 聚类

• PCA（多种实现方式：power iteration、Jacobi、RSVD）

• NMF

• T-SNE

• ISOMAP、Spectral Embedding 等流形学习方法

数据预处理

• 标准化 / 归一化（MinMax、Standard、Robust 等）

• OneHotEncoder、KBinsDiscretizer

• QuantileTransformer

• LabelBinarizer 等

模型评估指标

• 分类指标：accuracy、precision、recall、f1、roc_auc 等

• 回归指标：MSE、R²、explained variance

• GLM 专用评估指标（Poisson / Gamma / Tweedie）

所有算法均围绕 MPC 场景下的安全性、精度与性能进行实现与权衡。

从 SPU 子模块到独立 Python 库

SML 最初是 SPU 库的一部分，随着算法数量增加和使用场景清晰，我们决定将其独立拆分为一个单独的 Python 包：

• 更清晰的定位：隐私保护机器学习算法库

• 更友好的使用方式：Python 原生 API

• 更利于社区共建与持续演进

快速开始

你可以通过 PyPI 快速体验 SML：

pip install sf-sml

从源代码安装

git clone https://github.com/secretflow/sml.git
cd sml
pip install -e .

安装完成后，您可以运行任何测试，例如：

# run single unit test
pytest tests/cluster/kmeans_test.py  # run kmeans simulation

# run all unit tests
pytest tests

# run single emulation test
python emulations/run_emulations.py --module emulations.cluster.kmeans_emul  # run kmeans emulation

# run all emulations
python emulations/run_emulations.py

# list all available emulations
python emulations/run_emulations.py --list-only

加入社区，一起共建生态

SML 是 SecretFlow 开源生态的重要组成部分。 我们期待与你一起探索 隐私计算时代的机器学习新范式：

• 贡献算法与实现

• 优化性能与数值稳定性

• 参与设计讨论与 Issue 反馈

• 分享真实应用场景

未来我们将持续推进：

• 算法覆盖与稳定性增强

• API 设计持续对齐 scikit-learn 使用习惯

• 更完善的文档、示例与最佳实践

• 与 SecretFlow / SPU / SCQL 的深度协同