Convolution neural network training apparatus and method thereof
An apparatus and method of training a convolutional neural network (CNN) are provided. A method of training a CNN including a plurality of convolution layers stored in a memory involves approximating, using a processor, a convolution layer among the plurality of convolution layers using a low-rank approximation; reducing the number of output reconstruction filters of the approximated convolution layer; and modifying a structure of the CNN based on an approximation result and the reduced number of output reconstruction filters.
1. A method of training a convolutional neural network (CNN) comprising a plurality of convolution layers stored in a non-transitory memory, the method comprising:
approximating, using a processor, a convolution layer among the plurality of convolution layers using a low-rank approximation;
reducing a number of output reconstruction filters of the approximated convolution layer;
modifying a structure of the CNN based on an approximation result and the reduced number of output reconstruction filters; and
training the modified CNN,
wherein one convolution layer among the plurality of convolution layers is approximated into one or more input conversion filters, one or more convolution filters, and one or more output reconstruction filters, and
wherein the input conversion filter is configured to reduce a number of channels of input data, the convolution filter is configured to perform a convolution operation on input data having the reduced number of channels, and the output reconstruction filter is configured to restore a convolution operation result to have a same number of channels as a number of channels of the convolution layer.
2. The method of claim 1 , after the training of the modified CNN, further comprising:
performing the operations of:
sequentially approximating convolution layers that follow the approximated convolution layer;
reducing the number of output reconstruction filters of the currently approximated convolution layer and modifying the structure of the CNN; and
training the modified CNN.
3. The method of claim 1 , after the training of the modified CNN, further comprising:
classifying image data using the trained CNN; and
in response to an accuracy of the classification not satisfying a designated criteria, performing the operations of:
reducing the number of output reconstruction filters;
modifying the structure of the CNN; and
training the modified CNN.
4. The method of claim 1 , further comprising:
modifying a structure of at least one convolution layer that follows the approximated convolution layer,
wherein the structure of the CNN is modified based on a result of the modifying of the structure of said at least one convolution layer.
5. The method of claim 4 , wherein the structure of the CNN is modified by changing the number of convolution filters of the at least one convolution layer that follows the approximated convolution layer.
6. The method of claim 4 , after the training of the modified CNN, further comprising:
classifying image data using the trained CNN; and
in response to an accuracy of the classification not satisfying a designated criteria, performing the operations of:
modifying the structure of at least one convolution layer that follows the approximated convolution layer;
modifying the structure of the CNN; and
training the modified CNN.
7. A non-transitory computer-readable medium storing instructions that, when executed by a computer processor, cause the computer processor to train a convolution neural network stored in a non-transitory memory according to the method of claim 1 .
8. An apparatus for training a convolution neural network (CNN) comprising a plurality of convolution layers stored in a non-transitory memory, the apparatus comprising:
a processor configured to
approximate a convolution layer among the plurality of convolution layers using a low-rank approximation,
reduce a number of output reconstruction filters of the approximated convolution layer,
modify a structure of the CNN based on an approximation result and the reduced number of output reconstruction filters, and
train the modified CNN,
wherein the processor is further configured to approximate the convolution layer among the plurality of convolution layers into one or more input conversion filters, one or more convolution filters, and one or more output reconstruction filters, and
wherein the processor is further configured to reduce a number of channels of input data, perform a convolution operation on input data having the reduced number of channels, and restore a convolution operation result to have a same number of channels as a number of channels of the convolution layer.
9. The apparatus of claim 8 , wherein the processor is further configured to sequentially approximate another convolution layer that follows the approximated convolution layer, in response to the modified CNN being trained.
10. The apparatus of claim 8 , wherein the processor is further configured to:
classify image data using the trained CNN; and
re-change the number of output reconstruction filters in response to an accuracy of the classification not satisfying a designated criteria.
11. The apparatus of claim 8 , wherein the processor is further configured to:
modify a structure of at least one convolution layer that follows the approximated convolution layer based on a result of the modifying of the structure of the convolution layer that follows the approximated convolution layer.
12. The apparatus of claim 11 , wherein the processor is further configured to modify the structure of the at least one convolution layer by changing a respective number of convolution filters of said at least one convolution layer.
13. The apparatus of claim 11 , wherein the processor is further configured to:
classify image data using the trained CNN; and
re-modify the structure of the at least one convolution layer that follows the approximated convolution layer in response to an accuracy of the classification not satisfying a designated criteria.
14. An apparatus for training a neural network, the apparatus comprising:
a non-transitory memory storing a convolution neural network (CNN) comprising a plurality of convolution layers; and
a processor configured to
approximate a convolution layer from among the plurality of convolution layers using a low-rank approximation,
reduce a number of output reconstruction filters of the approximated convolution layer,
modify a structure of the CNN based on a result of the approximating of the convolution layer and the reduced number of output reconstruction filters, and
train the modified CNN stored in the non-transitory memory,
wherein the processor is further configured to classify input image data retrieved from a training data memory storage by using the modified CNN, and further modify the modified CNN in response to an accuracy of the classification not satisfying a designated criteria.
15. The apparatus of claim 14 , wherein the apparatus trains the modified CNN by retrieving input image data from a training data memory storage.