Systems and methods for neural network pruning with accuracy preservation
AssigneeNeuralmagic Inc.
InventorDan Alistarh
Life of this patent
Estimating the term…
Abstract
Training a neural network (NN) may include training a NN N, and for S, a version of N to be sparsified (e.g. a copy of N), removing NN elements from S to create a sparsified version of S, and training S using outputs from N (e.g. “distillation”). A boosting or reintroduction phase may follow sparsification: training a NN may include for a trained NN N and S, a sparsified version of N, re-introducing NN elements previously removed from S, and training S using outputs from N. The boosting phase need not use a NN sparsified by “distillation.” Training and sparsification, or training and reintroduction, may be performed iteratively or over repetitions.
Claims28
A method for training a neural network (NN), the method comprising:
- training a NN N;
- and
- for S, a version of N to be sparsified:
- removing NN elements from S to create a sparsified version of S;
- and
- training S using outputs from N.
A method for training a neural network (NN), the method comprising:
- for a trained NN N and S, a sparsified version of N:
- re-introducing NN elements previously removed from S;
- and
- training S using outputs from N.
A system for training a neural network (NN), the system comprising:
- a memory, and
- a processor configured to:
- train a NN N;
- and
- for S, a version of N to be sparsified:
- remove NN elements from S to create a sparsified version of S;
- and
- train S using outputs from N.
A system for training a neural network (NN), the system comprising:
- a memory, and
- a processor configured to:
- for a trained NN N and S, a sparsified version of N:
- re-introduce NN elements previously removed from S;
- and
- train S using outputs from N.
Dependencies are inferred from the claim text.