NeurIPS 2024poster1 citations

On the Ability of Developers' Training Data Preservation of Learnware

Hao-Yi Lei, Zhi-Hao Tan, Zhi-Hua Zhou

Abstract

The learnware paradigm aims to enable users to leverage numerous existing well-trained models instead of building machine learning models from scratch. In this paradigm, developers worldwide can submit their well-trained models spontaneously into a learnware dock system, and the system helps developers generate specification for each model to form a learnware. As the key component, a specification should characterize the capabilities of the model, enabling it to be adequately identified and reused, while preserving the developer's original data. Recently, the RKME (Reduced Kernel Mean Embedding) specification was proposed and most commonly utilized. This paper provides a theoretical analysis of RKME specification about its preservation ability for developer's training data. By modeling it as a geometric problem on manifolds and utilizing tools from geometric analysis, we prove that the RKME specification is able to disclose none of the developer's original data and possesses robust defense against common inference attacks, while preserving sufficient information for effective learnware identification.

LearnwareModel SpecificationReduced Kernel Mean EmbeddingData PreservationSynthetic DataLearnware Dock System
BibTeX
@inproceedings{
lei2024on,
title={On the Ability of Developers' Training Data Preservation of Learnware},
author={Hao-Yi Lei and Zhi-Hao Tan and Zhi-Hua Zhou},
booktitle={The Thirty-eighth Annual Conference on Neural Information Processing Systems},
year={2024},
url={https://openreview.net/forum?id=wsqDJHPUHN}
}
On the Ability of Developers' Training Data Preservation of Learnware · NeurIPS 2024