Key management for machine learning models

Abstract

Various aspects of the present disclosure relate to a wireless communications system that includes a network data analytics function (NWDAF) containing a model training logical function (MTLF), an NWDAF containing an analytics logical function (AnLF), and an analytics data repository function (ADRF). The NWDAF containing the MTLF generates a security context that protects a machine learning (ML) model that is stored in the ADRF. An NWDAF containing the AnLF obtains the protected ML model from the ADRF and obtains the security context from the NWDAF containing the MTLF. The security context is managed using a storage duration time that indicates when the ADRF is to delete the protected ML and the NWDAF containing the MTLF is to delete the security context, or a validity time that indicates when the ADRF is to delete the protected ML and the NWDAF containing the MTLF is to delete the security context.