Patent ID: 11907679
Assignee: KIOXIA CORPORATION
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G

Claim 0:
1. An arithmetic operation device, comprising:
a memory configured to store a first machine learning model that includes a predetermined number of parameters and is trained so as to output second data corresponding to input first data;
processing circuitry configured to execute an arithmetic operation using the first machine learning model stored in the memory; and
an interface connected to the processing circuitry, wherein the processing circuitry is further configured to
receive the first data and a required performance related to an inference through the interface, the required performance being provided from outside the arithmetic operation device and including an accuracy of inference using a machine learning model and a computational complexity based on a product of a number of multiply-accumulate computations and a number of bits of weight parameters,
remove a part of parameters of the predetermined number of parameters from the first machine learning model and decrease the number of bits of weight parameters according to the required performance to generate a second machine learning model, wherein a second accuracy of inference using the second machine learning model is lower than a first accuracy of inference using the first machine learning model but satisfies the received required performance,
input the first data into the second machine learning model to acquire third data output from the second machine learning model with a smaller computational complexity than the first machine learning model, and
output the third data through the interface.