Malware determination device, malware determination system, malware determination method, and program
A malware determination device, in which, upon input of an attribute name and an attribute value of an attribute of an executable file, a feature-selection setting unit registers the attribute with the attribute name in an attribute table as an attribute to be extracted, and registers the attribute value as an attribute value to be deleted in an attribute value table. Upon input of an executable file to be learned or to be determined, a feature extraction unit extracts an attribute value of an attribute registered as an attribute to be extracted in the attribute table from the executable file, to generate a feature vector including the extracted attribute value as a feature. A feature selection unit performs deletion of an attribute value registered as an attribute value to be deleted in the attribute value table from the feature vector.
1. A malware determination device comprising:
a memory that stores a feature selection database including an attribute table and an attribute value table;
processing circuitry configured to
prior to receiving input of an executable file, upon input of an attribute name of an attribute, register an attribute having the input attribute name in the attribute table as an attribute to be extracted, and upon input of an attribute value of an attribute, register the input attribute value in the attribute value table as an attribute value to be deleted or as an attribute value not to be deleted;
upon input of the executable file, extract a first attribute value of an attribute registered as an attribute to be extracted in the attribute table from the executable file to generate a feature vector including the extracted first attribute value as a feature;
perform deletion of a second attribute value registered as an attribute value to be deleted or deletion of an attribute value other than attribute values registered as attribute values not to be deleted in the attribute value table from the generated feature vector, to reconstruct the feature vector;
when the processing circuitry reconstructs a feature vector of an executable file to be learned, perform machine learning of the executable file to be learned based on the feature vector and information indicating whether the feature vector of the executable file to be learned is malware, and that, when the processing circuitry reconstructs a feature vector of an executable file to be determined, calculate a score of the likelihood of malware for the executable file to be determined based on a result of the machine learning and the feature vector; and
determine whether the executable file to be determined is malware based on the calculated score of the executable file to be determined.
2. A malware determination system comprising the malware determination device according to claim 1 , and further comprising a feature-selection trial device, wherein
the feature-selection trial device includes
processing circuitry configured to
repeat a process of selecting one set from sets obtained by combining one or more sets of features including an attribute name and an attribute value of an attribute of an executable file and inputting the selected set to the malware determination device,
that each time the malware detection device inputs the set, repeat a process of inputting an executable file to be determined after inputting an executable file to be learned to the malware determination device, and
each time the malware detection device inputs the set, acquire a determination result including a score of the executable file to be determined and information indicating whether the executable file to be determined has been determined to be malware, for each of executable files to be determined that have been input repeatedly to the processing circuitry, from the malware determination device, to calculate an index indicating a degree of determination accuracy of the malware determination device based on the determination result, and
select a set having a highest calculated index from the sets input to the malware detection device and input the selected set to the malware detection device.
3. The malware determination system according to claim 2 , wherein the processing circuitry of the feature-selection trial device calculates the index by using a detection rate, which is a rate of determining an executable file to be determined that is malware to be malware correctly, or an erroneous detection rate, which is a rate of erroneously determining an executable file to be determined that is not malware to be malware.
4. A malware determination system comprising the malware determination device according to claim 1 , and further comprising a user interface, wherein
the user interface is configured to
acquire an attribute name and an attribute value of an attribute of an executable file, and
display a setting screen including a list of attribute names and attribute values acquired by the user interface ,
display attribute names on the setting screen together with a first check box, and regarding an attribute name of an attribute registered as an attribute to be extracted in the attribute table, checks the first check box,
display attribute values on the setting screen together with a second check box, and regarding an attribute value other than attribute values registered as attribute values to be deleted or an attribute value registered as an attribute value not to be deleted in the attribute value table, checks the second check box, and
when a checked state of the first check box or the second check box is manually changed, input an attribute name checked in the first check box after the change to the malware determination device, and input an attribute value not checked in the second check box or a checked attribute value after the change to the malware determination device.
5. The malware determination system according to claim 4 , wherein the user interface displays an attribute value on the setting screen together with number of appearances, which is number of executable files in which the attribute value appears, and total number of executable files.
6. A malware determination method performed by a malware determination device, the malware determination method comprising:
a feature-selection setting step at which, prior to receiving input of an executable file, upon input of an attribute name of an attribute, the attribute with the input attribute name is registered in an attribute table as an attribute to be extracted, and upon input of an attribute value of an attribute, the input attribute value is registered in an attribute value table as an attribute value to be deleted or as an attribute value not to be deleted,
a feature extraction step at which, upon input of the executable file, a first attribute value of an attribute registered as an attribute to be extracted in the attribute table is extracted from the executable file and a feature vector including the extracted first attribute value as a feature is generated,
a feature selection step of performing deletion of a second attribute value registered as an attribute value to be deleted or deletion of an attribute value other than attribute values registered as attribute values not to be deleted in the attribute value table from the feature vector generated at the feature extraction step, to reconstruct the feature vector,
a classification step at which, when a feature vector of an executable file to be learned is reconstructed at the feature selection step, machine learning of the executable file to be learned is performed based on the feature vector and information indicating whether the feature vector of the executable file to be learned is malware, and at which, when a feature vector of an executable file to be determined is reconstructed at the feature selection step, a score of likelihood of malware is calculated for the executable file to be determined based on a result of the machine learning and the feature vector, and
a determination step of determining whether the executable file to be determined is malware based on the score of the executable file to be determined calculated at the classification step.
7. A non-transitory computer-readable recording medium having stored a program that causes a malware detection device to perform a method comprising:
a feature-selection setting step at which, prior to receiving input of an executable file, upon input of an attribute name of an attribute, the attribute with the input attribute name is registered in an attribute table as an attribute to be extracted, and upon input of an attribute value of an attribute, the input attribute value is registered in an attribute value table as an attribute value to be deleted or as an attribute value not to be deleted,
a feature extraction step at which, upon input of the executable file, a first attribute value of an attribute registered as an attribute to be extracted in the attribute table is extracted from the executable file and a feature vector including the extracted first attribute value as a feature is generated,
a feature selection step of performing deletion of a second attribute value registered as an attribute value to be deleted or deletion of an attribute value other than attribute values registered as attribute values not to be deleted in the attribute value table from the feature vector generated at the feature extraction step, to reconstruct the feature vector,
a classification step at which, when a feature vector of an executable file to be learned is reconstructed at the feature selection step, machine learning of the executable file to be learned is performed based on the feature vector and information indicating whether the feature vector of the executable file to be learned is malware, and at which, when a feature vector of an executable file to be determined is reconstructed at the feature selection step, a score of likelihood of malware is calculated for the executable file to be determined based on a result of the machine learning and the feature vector, and
a determination step of determining whether the executable file to be determined is malware based on the score of the executable file to be determined calculated at the classification step.