Non-volatile memory with integrated artificial intelligence and updating model

A non-volatile storage apparatus, comprising:

• a memory die assembly comprising:
• a non-volatile memory structure including a first set of non-volatile memory cells configured to store host data and a second set of non-volatile memory cells storing a pre-trained model received pre-trained from a source external to the non-volatile storage apparatus, the first set of non-volatile memory cells are arranged as vertical NAND strings grouped into blocks, each of the blocks includes word lines and channels;
• and
• a control circuit connected to the non-volatile memory structure and configured to write to and read from the non-volatile memory structure, the control circuit comprises an inference engine circuit positioned in the memory die assembly, the control circuit is configured to:
• use the inference engine circuit with the pre-trained model from the second set of non-volatile memory cells with one or more metrics describing current operation of the non-volatile storage apparatus in order to attempt to predict defects in the first set of non-volatile memory cells,
• determine that a first set of the blocks have failed,
• gather a first set of one or more metrics describing the operation of the non-volatile memory structure related to the first set of multiple blocks failing, determine that a second set of multiple blocks have successfully performed memory operations,
• gather a second set of one or more metrics describing the operation of the non-volatile memory structure related to the second set of multiple blocks successfully performing memory operations, and
• use the gathered first of one or more metrics and the second set of one or more metrics to update the pre-trained model in the non-volatile storage apparatus.

A method, comprising:

• storing a pre-trained model in non-volatile memory cells of a non-volatile storage system;
• the non-volatile storage system gathering one or more metrics describing current operation of the non-volatile storage system;
• using an inference engine in the non-volatile storage system with the pre-trained model and the one or more metrics to attempt to predict a defect in the non-volatile storage system including providing a prediction about a first block of non-volatile memory cells that did not include predicting the first block failing for its intended use;
• the non-volatile storage system determining that the first block failed;
• the non-volatile storage system determining that the inference engine's prediction about the first block was incorrect;
• the non-volatile storage system gathering one or more metrics describing the operation of the first block that failed;
• and
• the non-volatile storage system using the one or more metrics describing the operation of the first block that failed to update the model on the non-volatile storage system.