# Patent

## Title
Incremental machine learning for data loss prevention

## Abstract

A computing device receives a document that was incorrectly classified as sensitive data based on a machine learning-based detection (MLD) profile. The computing device modifies a training data set that was used to generate the MLD profile by adding the document to the training data set as a negative example of sensitive data to generate a modified training data set. The computing device then analyzes the modified training data set using machine learning to generate an updated MLD profile.

## Background
RELATED APPLICATIONS
This patent application is a continuation in part of U.S. patent application Ser. No. 13/038,299, filed Mar. 1, 2011, which is herein incorporated by reference and which claims the benefit under 35 U.S.C. §119(e) of U.S. Provisional Application No. 61/423,053, filed Dec. 14, 2010, which is herein incorporated by reference.


TECHNICAL FIELD
Embodiments of the present invention relate to the field of data loss prevention and, more particularly, to a data loss prevention (DLP) system that generates and uses machine learning-based detection (MLD) profiles.
BACKGROUND
Many organizations implement data loss prevention (DLP) systems to identify and control access to sensitive data. Typical DLP systems protect sensitive data through deep content inspection and analysis, which includes describing technology and fingerprinting technology. Describing technology protects sensitive data by identifying matches to keywords, expressions or patterns, and file types, and by performing other signature-based detection techniques. Fingerprinting technology protects sensitive data by identifying exact matches to whole or partial files. While effective in protecting much of an organization's sensitive data, fingerprinting and describing technologies have limitations when addressing large amounts of unstructured data and intellectual property such as product formulas, source code, and sales and marketing reports.
To more accurately protect sensitive unstructured data, some DLP systems are exploring the use of vector machine learning (VML) technology. However, VML is very complex to implement. Accordingly, current DLP systems that use VML require an expert in VML to design machine learning-based detection (MLD) profiles for customers. The DLP system that is shipped to the customer then has a predefined MLD profile that the customer is unable to modify. Such DLP systems do not provide any tools (e.g., user interface or workflow) to enable users to generate their own MLD profiles.
SUMMARY
In one embodiment, a computing device receives a document that was incorrectly classified as sensitive data based on a machine learning-based detection (MLD) profile. The computing device modifies a training data set that was used to generate the MLD profile by adding the document to the training data set as a negative example of sensitive data to generate a modified training data set. The computing device then analyzes the modified training data set using machine learning to generate an updated MLD profile. In one embodiment, the computing device additionally receives a document that was incorrectly classified as non-sensitive data based on the MLD profile, and modifies the training data set that was used to generate the MLD profile by adding the document to the training data set as a positive example of sensitive data to generate the modified training data set.
In one embodiment, the computing device determines whether a threshold number of documents that were incorrectly classified as sensitive data have been received. The computing device then analyzes the modified training data set in response to determining that at least the threshold number of documents have been received. In one embodiment, the computing device performs the analysis on a periodic basis in accordance with an MLD profile retraining schedule.
In one embodiment, the computing device determines a first quality rating of the training data set and a second quality rating of the modified training data set. The computing device compares the first quality rating to the second quality rating. The computing device may then analyze the modified training data set in response to determining that the first quality rating is lower than or equal to the second quality rating. The computing device may also deploy the updated MLD profile in response to determining that the second quality rating is higher than or equal to the first quality rating. In one embodiment, the first quality rating and the second quality rating are determined based on performing at least one of latent semantic indexing or k-fold cross validation.
In one embodiment, a computer readable medium includes instructions that will cause a processor that executes the instructions to receive a document that was incorrectly classified as sensitive data based on a machine learning-based detection (MLD) profile. The instructions further cause the processor to modify a training data set that was used to generate the MLD profile by adding the document to the training data set as a negative example of sensitive data to generate a modified training data set. The instructions further cause the processor to analyze the modified training data set using machine learning to generate an updated MLD profile.
In one embodiment, a method for generating an MLD profile comprises receiving a document that was incorrectly classified as sensitive data based on a machine learning-based detection (MLD) profile, modifying a training data set that was used to generate the MLD profile by adding the document to the training data set as a negative example of sensitive data to generate a modified training data set, and analyzing the modified training data set using machine learning to generate an updated MLD profile.



BRIEF DESCRIPTION OF THE DRAWINGS
The present invention will be understood more fully from the detailed description given below and from the accompanying drawings of various embodiments of the invention.
FIG. 1 illustrates an exemplary system architecture, in accordance with one embodiment of the present invention.
FIG. 2 is a block diagram of a data loss prevention agent, in accordance with one embodiment of the present invention.
FIG. 3 is a block diagram of a machine learning manager, in accordance with one embodiment of the present invention.
FIG. 4A is a flow diagram illustrating one embodiment for a method of generating and deploying a MLD profile.
FIG. 4B is a flow diagram illustrating another embodiment for a method of generating and deploying a MLD profile.
FIGS. 5-8 illustrate various views of a user interface for generating and deploying MLD profiles, in accordance with embodiments of the present invention.
FIG. 9 is a state diagram showing different states of an ML manager during MLD profile generation, in accordance with one embodiment of the present invention.
FIG. 10A is a flow diagram illustrating one embodiment for a method of generating a MLD profile and deploying the MLD profile to an existing DLP policy.
FIG. 10B is a flow diagram illustrating another embodiment for a method of generating a MLD profile and deploying the MLD profile to an existing DLP policy.
FIG. 11 is a flow diagram illustrating one embodiment for a method of protecting a computing device from data loss using a DLP policy that includes a MLD profile.
FIG. 12 is a block diagram of an exemplary computer system that may perform one or more of the operations described herein.



DETAILED DESCRIPTION
A system and method for generating, deploying and managing machine learning-based detection (MLD) profiles for a data loss prevention (DLP) system are described. In embodiments of the present invention, the system and method provide a user interface, workflow and/or other tools that enable users who are not vector machine learning experts to generate MLD profiles. This decreases the expense of deploying MLD profiles for DLP, and improves configurability of MLD profiles. Additionally, this enables MLD profiles to continually be improved by a DLP administrator.
In the following description, numerous details are set forth. It will be apparent, however, to one of ordinary skill in the art having the benefit of this disclosure, that the present invention may be practiced without these specific details. In some instances, well-known structures and devices are shown in block diagram form, rather than in detail, in order to avoid obscuring the present invention. For example, the following description provides details for using MLD profiles in an endpoint DLP system. However, it would be clear to one of ordinary skill in the art that embodiments of the present invention also apply to network DLP systems and to discover DLP systems (DLP systems that scan storage devices to identify and/or classify sensitive data). For example, in embodiments of the present invention, MLD profiles may be generated for detecting sensitive data that moves through an enterprise network.
Some portions of the detailed description that follow are presented in terms of algorithms and symbolic representations of operations on data bits within a computer memory. These algorithmic descriptions and representations are the means used by those skilled in the data processing arts to most effectively convey the substance of their work to others skilled in the art. An algorithm is here, and generally, conceived to be a self-consistent sequence of steps leading to a desired result. The steps are those requiring physical manipulations of physical quantities. Usually, though not necessarily, these quantities take the form of electrical or magnetic signals capable of being stored, transferred, combined, compared, and otherwise manipulated. It has proven convenient at times, principally for reasons of common usage, to refer to these signals as bits, values, elements, symbols, characters, terms, numbers, or the like.
It should be borne in mind, however, that all of these and similar terms are to be associated with the appropriate physical quantities and are merely convenient labels applied to these quantities. Unless specifically stated otherwise as apparent from the following discussion, it is appreciated that throughout the description, discussions utilizing terms such as “receiving”, “analyzing”, “determining”, “enabling”, “identifying,” “modifying” or the like, refer to the actions and processes of a computer system, or similar electronic computing device, that manipulates and transforms data represented as physical (e.g., electronic) quantities within the computer system's registers and memories into other data similarly represented as physical quantities within the computer system memories or registers or other such information storage, transmission or display devices.
The present invention also relates to an apparatus for performing the operations herein. This apparatus may be specially constructed for the required purposes, or it may comprise a general purpose computer selectively activated or reconfigured by a computer program stored in the computer. Such a computer program may be stored in a computer readable storage medium, such as, but not limited to, any type of disk including floppy disks, optical disks, CD-ROMs, and magnetic-optical disks, read-only memories (ROMs), random access memories (RAMs), EPROMs, EEPROMs, magnetic or optical cards, or any type of media suitable for storing electronic instructions.
FIG. 1 illustrates an exemplary system architecture 100, in accordance with one embodiment of the present invention. The system architecture 100 includes multiple endpoint devices 102A-102C networked to an endpoint server 115, which in turn is networked to an enforcement server 120.
Each of the endpoint devices may be a personal computer (PC), a laptop, a mobile phone, a tablet computer, or any other computing device that can be accessed by a user. Each endpoint device 102A-102C has multiple different data loss vectors. Each data loss vector is a path through which data can be transferred off of the endpoint device. Examples of data loss vectors include burning files to an optical disc, copying data to a portable drive (e.g., a portable universal serial bus (USB) drive), printing data to a printer, sending data through a fax, sending an email, sending an instant message, performing a print screen operation, etc.
The endpoint devices 102A-102C each run an operating system (OS) that manages hardware and software of the endpoint devices. The OS may be, for example, Microsoft® Windows®, Linux®, Symbian®, Apple's® OS X®, Solaris®, etc. One or more applications run on top of the OS and perform various operations that involve accessing, moving, or otherwise manipulating data residing in a data store included in, attached directly to, or networked with the endpoint device. For example, the applications may include a CD or DVD burning application, an email application, a web browser, an instant messaging application, a printing application, a print screen function, etc. In one embodiment, the applications perform the operations in response to receiving user instructions.
Each endpoint device 102A-102C may be connected to a data store 135A-135C, which may be a hard disk drive, tape backup, optical drive, volatile memory (e.g., random access memory (RAM)), or other storage device. Data stores 135A-135C may be internal to endpoint devices 102A-102C or external to endpoint devices 102A-102C. In one embodiment, the data stores 135A-135C may be combined into a network storage such as a storage area network (SAN) or network attached storage (NAS). In one embodiment, data stores 135A-135C may be combined into a network database such as a relational database. Data stores 135A-135C may contain data that includes sensitive information. The data may include files (e.g., documents), tables, or other data formats. Examples of sensitive information include source code, patient health information, insurance claims, product formulas, legal documents, merger and acquisition documents, sales and marketing reports, social security numbers, credit card numbers, etc.
Each endpoint device 102A-102C includes a DLP agent 106 that monitors data loss vectors to ensure that sensitive (e.g., confidential) information does not leave the endpoint device for illegitimate purposes. The DLP agent 106 may scan data as it moves through a data loss vector and/or when a request to send the data through a data loss vector is received. When DLP agent 106 detects data moving through a data loss vector or a request to move the data through the data loss vector, the DLP agent 106 implements a DLP policy 110 to determine whether the data is sensitive data (includes sensitive information). The DLP policy 110 may specify a type of content to be monitored (e.g., messages, displayed data, stored documents, etc.), how to identify sensitive data, and/or an action to be performed when sensitive data is detected. In one embodiment, the DLP policy 110 includes a MLD profile 112. The DLP agent 106 includes a machine learning (ML) module 108 that processes the data using the MLD profile 112. By processing the data using the MLD profile 112, the ML module 108 determines whether the data is sensitive data.
For some types of DLP detection techniques, DLP agent 106 sends data to endpoint server 115, and a global DLP detection engine 122 included in the endpoint server 115 determines whether the data includes sensitive information. Once global DLP detection engine 122 determines whether a file or other data contains sensitive information, endpoint server 115 sends a message back to the DLP agent 106 stating whether or not the data is sensitive data. The DLP agent 106 may then perform one or more actions to enforce a DLP policy 110 if the data does contain confidential information. In one embodiment, global DLP detection engine 122 includes a ML module 108 and a DLP policy 126 that includes a MLD profile 128. The DLP policy 126 and MLD profile 128 may be different from DLP policy 110 and MLD profile 112.
In one embodiment, endpoint server 115 acts as an aggregator of data regarding violations of DLP policies (e.g., as an aggregator of incident reports). The endpoint server 115 may collect such data from each endpoint device 102A-102C, and report the collected data to enforcement server 120 for analysis.
Enforcement sever 120 manages DLP policies. This may include generating and modifying DLP policies (e.g., based on administrator input). The enforcement server 120 may then propagate the DLP policies to the endpoint server 115 and/or to the endpoint devices 102. Additionally, the enforcement server 120 may generate DLP response rules, which it may also propagate to the endpoint server 115 and/or to the endpoint devices 102. The DLP response rules designate actions for endpoint devices 102 and/or the endpoint server 115 to take when DLP policies are violated. Examples of actions that an endpoint device may take include sending a notification to an administrator, preventing the data from exiting the endpoint device 102A-102C through the data loss vector, locking down the endpoint device so that no data can be moved off of the endpoint device through any data loss vector, encrypting data as it is moved off the endpoint device, and so on.
In one embodiment, enforcement server 120 includes a machine learning (ML) manager 130. ML manager 130 provides a user interface and a workflow for users to generate and deploy MLD profiles. The ML manager 130 is described in greater detail below with reference to FIG. 3.
FIG. 2 is a block diagram of a data loss prevention agent 205, in accordance with one embodiment of the present invention. The DLP agent 205 may monitor different data loss vectors, applications, data, etc. to detect operations that attempt to move data off of an endpoint device. The user initiated operations may include, for example saving or accessing restricted database data on any storage device of the endpoint device, using restricted database data in an application, printing confidential data, using confidential data in a network communication protocol, etc.
The DLP agent 205 may include one or more policy violation detectors, each of which may process different DLP policies 250 and/or different profiles 255, 260, 265 within a DLP policy 250 to identify and secure sensitive data. DLP policy 250 may include criteria that may indicate an increased risk of data loss. DLP policy 250 is violated if one or more criteria included in the DLP policy 250 are satisfied. Examples of criteria include user status (e.g., whether the user has access privileges to a file), file location (e.g., whether a file to be copied is stored in a confidential database), file contents (e.g., whether a file includes sensitive information), time (e.g., whether an operation is requested during normal business hours), data loss vector, application attempting the operation, and so on.
The DLP policy 250 may include one or more profiles 255, 260, 265. Each profile may be used to identify sensitive data. In one embodiment, the DLP policy 250 includes a described content matching (DCM) profile 255. DCM profile 255 defines one or more key words and/or regular expressions to be searched for. For example, DCM profile 255 may define a social security number using a regular expression. Using DCM profile 255, DLP agent 205 determines whether any information included in scanned data match the key words and/or regular expressions. If a match is found, then it may be determined that the data includes sensitive information.
In one embodiment, the DLP policy 250 includes an exact data matching (EDM) profile and/or a indexed document matching (IDM) profile 260. Exact data matching (EDM) may be used for protecting data that is typically in structured formats, such as database records. Indexed document matching (IDM) may be used to protect unstructured data, such as Microsoft® Word or PowerPoint® documents, or CAD drawings. For both EDM and IDM, sensitive data is first identified by an organization that wishes to protect the data and then fingerprinted for precise, ongoing detection. In one embodiment, the fingerprinting process includes accessing and extracting text and data, normalizing it, and securing it using a nonreversible hash. When a file or other data is to be scanned, a fingerprint (e.g., hash) is generated of that file or contents of that file and compared to stored fingerprints. If a match is found, then the scanned file is identified as a containing sensitive data.
In one embodiment, the DLP policy 250 includes a machine learning-learning based detection (MLD) profile 265. Vector machine learning and other types of machine learning can be used to protect unstructured data, such as Microsoft® Word or PowerPoint® documents, or CAD drawings. The MLD profile 265 may include a training data set 270, a classification model 275 and a feature set 280. The training data set 270 is a collection of positive examples of sensitive data and negative examples of sensitive data. The training data set 270 is processed by a ML manager to generate the classification model 275 and the feature set 280. The classification model 275 is a machine-learning model for data classification that includes a map of support vectors that represent boundary documents. The feature set 280 is a data structure such as a list or table that includes multiple features extracted from the training data set 270. In one embodiment, each of the features is a word included in the data from the training data set 270.
One example of a policy violation detector is a machine learning module 225. The ML module 225 includes a ML engine 230 that takes as inputs a MLD profile 265 and unclassified data (e.g., a file 235), and outputs a classification for the data. The ML engine 230 processes the input data using the classification model 275 and the feature set 280. Therefore, the ML module 225 can use the MLD profile 265 to distinguish between sensitive data and non-sensitive data.
Policy violation responder 220 applies one or more DLP response rules 245 when a DLP policy violation is detected. Each DLP response rule 245 may be associated with one or more DLP policies 250. Each DLP response rule 245 includes one or more actions for policy violation responder 220 to take in response to violation of an associated DLP policy 250. Once a violation of a DLP policy 250 is discovered, policy violation responder 220 may determine which DLP response rules are associated with the violated DLP policy 250. One or more actions included in the response rule 245 can then be performed. Examples of performed actions include sending a notification to an administrator, preventing the data from exiting an endpoint device through a data loss vector, locking down the computer so that no data can be moved off of the endpoint device through any data loss vector, encrypting data as it is moved off the endpoint device, and so on.
Incident report generator 215 may generate an incident report 240 noting the violated DLP policy 250 and the circumstances surrounding the violated DLP policy 250. Incident report generator 215 maintains a record of incident reports 240 of some or all policy violations that have occurred on an endpoint device and/or that have been attempted by a particular user. The user may be identified, for example, based on a user login. In addition to identifying the DLP policy that was violated, each incident report 240 may also indicate the circumstances of the policy violation. For example, the incident report 240 may identify an application, user, data loss vector, type of sensitive data (e.g., social security number, credit card number, etc.), etc. associated with the policy violation. The incident report generator 215 may also include a time stamp showing when the policy violation occurred.
FIG. 3 is a block diagram of a machine learning (ML) manager 305, in accordance with one embodiment of the present invention. The ML manager 305 includes a MLD profile trainer 325, a MLD profile tester 320 and a MLD profile deployer 315. In one embodiment, ML manager 305 also includes a user interface 310. Note that in alternative embodiments, one or more of the MLD profile trainer 325, MLD profile tester 320, or MLD profile deployer 315 may be combined into a single module or divided into multiple modules.
MLD profile trainer 325 trains a MLD profile 365 based on a training data set 352. MLD profile training is a process in which contents are extracted from a training data set and machine-learning analysis is performed on the contents to generate a classification model and a feature set, both of which are described in greater detail below. A user (e.g., a DLP administrator) may specify the data to use in the training data set. In one embodiment, the user selects positive examples of sensitive data (positive data 345) and negative examples of sensitive data (negative data 350), and adds them to the training data set 352. This may be done via the user interface 310. Alternatively, the user may add files to a positive data folder and to a negative data folder via a standard file system interface (e.g., Microsoft® Explorer®). Data may be added to the training data set as discreet files (e.g., documents) or as components of a single compressed file (e.g., a zip file).
In one embodiment, machine learning manager 305 includes a training data separator 322. In one embodiment, the training data separator 322 is a component of MLD profile trainer 325. In some cases a user may provide a mixed corpus of data to use in the training data set without specifying positive examples of sensitive data (e.g., sensitive documents) or negative examples of sensitive data (e.g., non-sensitive documents). In such an embodiment, training data separator 322 analyzes the corpus of data to separate the corpus into two or more distinct data sets. In one embodiment, the training data separator 322 performs local weighted latent semantic indexing (LSI) or another form of LSI to separate the corpus of data into the distinct data sets. By performing local weighted LSI (or another form of LSI), the training data separator 322 identifies, for example, documents that contain similar features and groups those documents into a data set.
Local weighted LSI is a feature extraction algorithm that identifies patterns in relationships between the terms and concepts contained in an unstructured collection of text (e.g., the training data set). Local weighted LSI applies a mathematical operation called singular value decomposition (SVD) to break the training data set into multiple parts. These parts contain documents that are closely related to each other.
The inputs for local weighted LSI are a set of documents (or other textual data). The algorithm extracts out individual words (called terms) of the documents. The algorithm then creates a 2 dimensional term document matrix in which one axis includes the terms and the other axis includes the documents. Each cell of the matrix includes the frequency of a particular word in a particular document. The algorithm then performs singular value decomposition (SVD) on the matrix to reduce the matrix (which can be huge) to a much smaller matrix. Parameters on the matrix reduce down from terms and documents to concepts and documents. A concept is a statistical model that includes a group of words. Each concept may correspond to a particular type of document (e.g., source code, advertising materials, legal documents, etc.). The algorithm then performs clustering on this reduced matrix to identify which documents belong to which concept.
Once the corpus of data has been separated into multiple data sets, the user may specify one of the distinct data sets as containing positive examples of sensitive data and another one of the distinct data sets as containing negative examples of sensitive data. If the training data separator 322 generated more than two distinct data sets, then the contents of any unspecified data sets may not be used in the training data set (e.g., may be discarded). In an example, a provided corpus of data may include examples of confidential source code, examples of non-sensitive marketing materials and examples of confidential technical documentation. The training data separator 322 may separate the corpus into three distinct data sets, one for each of the types of data included in the corpus. A user may select the data set containing the non-sensitive marketing materials for the negative examples of sensitive data and may select the data set containing the confidential technical documentation for the positive examples of sensitive data.
In one embodiment, the user specifies a first document as a positive example of sensitive data and a second document as a negative example of sensitive data before local weighted LSI is performed. The training data separator 322 may then separate the data set into two data sets using the first and second documents as seed documents and identifying documents that are similar in content.
In one embodiment, data for the training data set 352 is extracted from incident reports 360. The incident reports 360 may have been generated for an existing DLP policy during enforcement of the DLP policy 385. Incident reports 360 may identify instances in which an operation was performed or requested to be performed on sensitive data. The incident reports may include genuine instances of sensitive data, and may also include false positives, in which non-sensitive data was classified as sensitive data. Other historical data, which may or may not be associated with incident reports, may also be used for the training data set. The historical data may include genuine instances of sensitive data, false positives, genuine instances of non-sensitive data and/or false negatives.
In one embodiment, MLD profile trainer 325 performs incremental training for an existing MLD profile. With incremental training, MLD profile trainer 325 may add new positive data and/or negative data to the training data set based on incident reports that have been generated since the MLD profile was last trained. The MLD profile trainer 325 may then retrain the MLD profile 365 using the updated training data set (e.g., recompute a feature set 375 and/or a classification model 380). In one embodiment, the MLD profile trainer 325 performs a full retraining of the MLD profile 365 using all of the previous contents of the training data set as well as the newly added content. In another embodiment, the MLD profile trainer 325 performs a partial retraining using only the recently added content. In still another embodiment, incremental training is used to generate an entirely new MLD profile. The new MLD profile may be based on just the new positive and/or negative data or based on a subset of the original positive and/or negative data along with the new positive and/or negative data. For example, MLD profile trainer 325 may generate a new MLD profile with the originally used positive examples of sensitive data and new negative examples of sensitive data to generate the new MLD profile.
MLD profile trainer 325 may perform incremental training automatically or in response to user input. In one embodiment, incremental training is performed according to a predetermined schedule. For example, MLD profile trainer 325 may perform training on the MLD profile on a periodic basis, such as daily, weekly, monthly, etc.
In one embodiment, the MLD profile trainer 325 will not generate a MLD profile 365 for a training data set 352 until a threshold number of documents have been added to the training data set (e.g., 20 documents, 50 documents, etc.). In one embodiment, a threshold amount of positive data 345 and a threshold amount of negative data 350 should be added. The threshold may be, for example, 50 positive documents and 50 negative documents. In one embodiment, a maximum document size (e.g., 15 MB, 30 MB, etc.) is enforced by the ML manager 305. Any documents larger than the maximum document size may be rejected for use as training data. The maximum document size may be user selectable.
In one embodiment, the MLD profile trainer 325 includes a model generator 330, a feature extractor 335 and a quality analyzer 340. The feature extractor 335 analyzes the positive examples of sensitive data and the negative examples of sensitive data in the training data set 352, and determines the frequency of occurrence of features (e.g., words) in the positive data and in the negative data. The feature extractor 335 then ranks positive features and negative features based on, for example, frequency of occurrence. In one embodiment, the feature extractor 335 filters out common word such as “the,” “it,” “and,” etc. The feature extractor 335 then selects the highest ranked features for a feature set 375.
In one embodiment, feature extractor 335 generates features from characters for character based alphabets such as Chinese characters (Kanji). Feature extractor 335 generates a feature for each character, and additionally creates a feature for each pair of adjacent characters. For example, for the characters Ξλ, the feature extractor would generate features for Θ, Ξ and ΘΞ.
The number of features added to the feature set 375 may be based on a memory allocation, which may be automatically selected by the MLD profile trainer 325 or may be chosen by a user. As the memory allocation increases, the number of features included in the feature set 375 also increases, which may increase accuracy of the MLD profile. Memory allocation may vary, for example, between about 30 MB and about 100 MB. In one embodiment, memory allocation is selectable as high, medium or low. Alternatively, specific memory allocations may be selected (e.g., 43 MB). Size of a resultant MLD profile 365 is proportional to a number of training documents and the memory allocation setting. In one embodiment, a lower memory allocation is used for a MLD profile 365 that will be implemented by a DLP agent, and a higher memory allocation is used for a MLD profile 365 that will be implemented by a global DLP detection engine.
In one embodiment, feature extractor 335 uses a term frequency-inverse document frequency (TF-IDF) algorithm to select the feature set 375. Alternatively, feature extractor 335 may use other feature extraction algorithms, such as segment-set term frequency-inverse segment-set frequency (STF-IS SF), segment-set term frequency-inverse document frequency (STF-IDF), etc. In one embodiment, the feature selection algorithm that feature extractor 335 uses is user selectable. Additionally, feature extractor 335 may perform feature extraction multiple times, each time using a different feature extraction algorithm. The feature sets generated using the different algorithms may each be used to generate different classification models, and may be tested by quality analyzer 340. The feature set that has the best quality metrics may then be saved and the others may be discarded.
After the feature extractor 335 has generated the feature set 375, model generator 330 generates a classification model 380 based on the feature set 375 and on the training data set 352. The classification model 380 is a statistical model for data classification that includes a map of support vectors that represent boundary features. The boundary features may be selected from the feature set 375, and may represent the highest ranked features in the feature set 375.
Once the feature extractor 335 generates the feature set 375 and the model generator 330 generates the classification model 380, a MLD profile 365 is complete. The MLD profile 365 may include the feature set 375, the classification model 380 and/or the training data set 370. The MLD profile 365 may also include user defined settings. In one embodiment, the user defined settings include a sensitivity threshold (also referred to as a confidence level threshold). The sensitivity threshold may be set to, for example, 75%, 90%, etc. When an ML engine uses the MLD profile 365 to classify a document as sensitive or not sensitive, the ML engine may assign a confidence value to the classification. If the confidence value for the document is 100%, then it is more likely that the decision that the document is sensitive (or not sensitive) is accurate than if the confidence value is 50%, for example. If the confidence value is less than the sensitivity threshold, then an incident may not be generated even though a document was classified as a sensitive document. This feature can help a user to further control and reduce false positives and/or false negatives. If an ML engine is trying to classify a document of a type that the training has never seen, it has a very low confidence of the document being positive and/or negative. The sensitivity threshold can be used to reduce occurrences of false positive in such cases. In one embodiment, the MLD profile trainer 325 automatically selects a sensitivity threshold for the MLD profile 365 based on the training.
In one embodiment, quality analyzer 340 analyzes a quality of the training data set 370 before a feature set 375 or classification model 380 are generated. Quality analyzer 340 may assign a quality rating to the training data set 370 based on the analysis. In one embodiment, quality analyzer 340 performs LSI (e.g., local weighted LSI) with the positive examples of sensitive data and with the negative examples of sensitive data to determine the quality rating. Local weighted LSI should identify only a single data set for the positive examples and a single data set for the negative examples. If local weighed LSI breaks either the positive examples or the negative examples into multiple parts, this indicates that an MLD profile generated from the training data set will have a suboptimal accuracy. In such cases, a DLP administrator may be advised to create multiple MLD profiles corresponding to each of the multiple parts of the positive examples. In one embodiment, the quality rating is based on the number of distinct sets of data included in the positive examples of sensitive data and in the negative examples of sensitive data. If the positive examples of sensitive data are grouped into a single data set and the negative examples of sensitive data are grouped into a separate single data set, then a highest rating may be assigned to the training data set. The rating may be reduced based on the number of additional distinct data sets included in the positive examples of sensitive data and/or in the negative examples of sensitive data.
Quality analyzer 340 may determine a quality rating for the training data set 370 before and after new documents are added to the training data set. This may be performed, for example, during incremental training. If the quality of the training data set is reduced after adding the new documents, then the new documents may be removed from the training data set and the MLD profile 365 may not be retrained. Alternatively, a portion of the new documents (e.g., those new documents that are included in a distinct data set) may be removed from the training data set before retraining is performed.
In one embodiment, quality analyzer 340 analyzes a quality of the MLD profile 365 after the MLD profile 365 has been generated and/or during the generation process, and generates one or more quality metrics for the MLD profile 365. The quality metrics may include a false positives rating (negative examples of sensitive data that were misclassified by the MLD profile 365 as sensitive data), a false negatives rating (positive examples of sensitive data that were misclassified by the MLD profile 365 as non-sensitive data), and/or a memory utilization rating (amount of memory used by the MLD profile 365).
In one embodiment, the quality analyzer 340 performs k-fold cross validation (or another type of cross validation) to determine a quality rating for the MLD profile 365. Cross validation is a technique for assessing how the results of a statistical analysis will generalize to an independent set of data. For k-fold cross validation, the training data set is divided into k separate equally sized subsets. K−1 subsets are used to generate the MLD profile, and the remaining subset is used to test the MLD profile. Testing the MLD profile includes determining the percentage of the documents in the remaining subset that are correctly classified as positive or negative examples of sensitive data. This is then repeated using a different k−1 subsets to generate the MLD profile and a different remaining subset to test the MLD profile. This process is repeated until all documents have been used for both training and testing. A mean of the testing results is then computed to determine an accuracy of the MLD profile. This mean may represent the quality rating for the MLD profile.
In one embodiment, the quality analyzer 340 can determine a document or group of documents that are reducing the quality rating for the MLD profile using k-fold cross validation (or another form of cross validation). K-fold cross validation can identify a subset of documents that is lowering the accuracy of the MLD profile. Once the subset is identified, the quality analyzer 340 may again perform k-fold cross validation on just the documents in that subset to narrow down which documents are causing problems. The problematic documents may then be removed from the training sets before the final MLD profile is created.
The quality analyzer 340 may compare the quality metrics to one or more quality thresholds. These may include a false positive threshold, a false negative threshold and/or a memory utilization threshold. In one embodiment, the false positive threshold is 5% and the false negative threshold is 5%. Alternatively, other false positive and/or false negative thresholds may be used. If the false positives rating exceeds the false positive threshold, the false negatives rating exceeds the false negatives threshold, or the memory utilization rating exceeds the memory utilization threshold, then the MLD profile 365 may not be ready for deployment. ML manager 305 may not permit the MLD profile 365 to be deployed if one or more of the quality thresholds have been exceeded by the MLD profile 365.
The MLD profile 365 may be modified by changing the training data set 352 and re-computing the feature set 375 and the classification model 380. The training data set 352 may be modified by adding new positive data 345, adding new negative data 350, removing instances of positive data 345, and/or removing instances of negative data 350. In one embodiment, quality analyzer 340 identifies particular files, documents, etc. from the negative data 350 that caused false positives and identifies particular files, documents, etc. from the positive data 345 that caused false negatives. A user may review this information to determine additional data to add to the training data set. Certain categories of documents may have been underrepresented in the training data set 352. For example, the user may wish to protect source code, and product documentation may have been cross classified as source code by the MLD profile 365. The user may correct this by adding additional examples of product documentation to the negative data set. Examples of categories of data that may be recognized and classified as sensitive or non-sensitive include source code, recipes, legal documents, product documentation, medical history documents, insurance documents, product formulas, patient health information, and so on.
In one embodiment, a user is able to specify a category for each file (e.g., document) that the user adds to the training data set. The quality analyzer 340 can then identify categories of documents that caused the most false positives and/or the most false negatives. In one embodiment, quality analyzer 340 suggests specific categories of documents for the user to add to improve the quality of the MLD profile 365.
In one embodiment, the ML manager 305 maintains a change report that includes changes made to a previously generated MLD profile. The change report may also include differences in quality metrics between the previously generated MLD profile and a recently modified MLD profile. The change report may be displayed to a user to enable the user to accept the changes or roll back the changes to revert to the previous MLD profile.
Once a MLD profile 365 is ready for deployment (e.g., the quality metrics are within the quality thresholds), MLD profile deployer 315 deploys the MLD profile 365. In one embodiment, MLD profile deployer 315 adds the MLD profile to an existing DLP policy 385. Alternatively, the MLD profile deployer 315 may generate a new DLP policy and add the MLD profile 365 to the new DLP policy 385.
In one embodiment, ML manager 305 includes a VML tester 320. VML tester tests the MLD profile against additional data. In one embodiment, the MLD profile tester performs negative testing on a predetermined set of testing data. The predetermined set of testing data may include a large amount of data (e.g., 10,000 documents) known to not include any sensitive information. The MLD profile tester 320 may also test the MLD profile on user selected data, which may include additional positive data and/or negative data.
FIG. 4A is a flow diagram illustrating one embodiment for a method 400 of generating and deploying a MLD profile. The method 400 is performed by processing logic that may comprise hardware (circuitry, dedicated logic, etc.), software (such as is run on a general purpose computer system or a dedicated machine), or a combination of both. Method 400 may be performed by a ML manager such as ML manager 130 running on enforcement server 120 of FIG. 1. Though method 400 is described below as being performed by an ML manager, method 400 may also be performed by other processing logic.
Referring to FIG. 4A, at block 405 a ML manager generates a temporary workspace for a new MLD profile or an existing MLD profile that is to be modified. In one embodiment, an empty MLD profile is generated for an existing DLP policy, which may already have other profiles such as IDM or EDM profiles running. In another embodiment, an empty MLD profile is generated for a new DLP policy that has yet to be deployed. Alternatively, an existing MLD profile is opened in the temporary workspace. In one embodiment, the temporary workspace is generated in response to a user requesting to create a new MLD profile or modify an existing MLD profile via a user interface for machine learning. In one embodiment, the new MLD profile is for protecting a specific category of data. For example, the MLD profile may be for protecting source code, for protecting patient information, for protecting sales data, etc.
FIG. 5 illustrates a first view 500 of a user interface for machine learning showing an empty temporary workspace, in accordance with one embodiment of the present invention. As shown, the temporary workspace includes a “positive” button 505 for uploading positive documents and a “negative” button 510 for uploading negative documents. In response to a user selecting the “positive” button 505 or the “negative” button 510, a ML manager may open a file browser window. A user may then navigate the file browser window to select a document or documents for uploading.
In one embodiment, the user interface includes a memory allocation button 515. In response to a user selecting the memory allocation button 515, the ML manager opens a window that presents the user with options for memory allocation. In one embodiment, the user is able to select between a high, medium and low memory allocation. Each memory allocation may be associated with a specific memory utilization threshold. Alternatively, the user may be able to select a specific memory allocation (e.g., 12 MB, 54 MB, etc.). In one embodiment, the user interface includes an additional button 520 that, when selected, opens a window in which a user can type in a profile name and/or a profile description.
Returning to FIG. 4A, at block 410 of method 400 the ML manager receives a training data set. In one embodiment, a user selects data for the training data set via the user interface. The training data set includes both positive examples of sensitive data and negative examples of sensitive data. The training data set may include multiple documents. In one embodiment, a user specifies a category (e.g., source code, sales data, medical records, etc.) for each document. In one embodiment, the ML manager checks a document size for each document in the training data set. The ML manager may reject those ML documents that exceed a maximum document size.
FIG. 6 illustrates a second view 600 of the user interface of FIG. 5 showing a training data set, in accordance with one embodiment of the present invention. Each document in the training data set may be displayed along with multiple attributes of the document. In one embodiment, displayed document attributes include a document type 605 (i.e., whether the document is a positive document or a negative document), a document name 610, a document date 615 (i.e., date that the document was uploaded), and a document author 620. Each document may also include a remove button 625. By selecting a remove button 625, a user may remove a particular document from the training data set. In one embodiment, a document category is also shown. A user may assign a document category for each document. In one embodiment, clicking a cross button present on a temporary workspace tab rolls back all changes that have been made to the profile. The profile then continues to remain in a last known functional state.
Once a threshold number of positive documents and negative documents have been added to the training data set (e.g., 20 documents of each type, 50 documents of each type, etc.), a train profile operation becomes available. In one embodiment, a “train profile” button 630 becomes active when the threshold number of positive documents and negative documents have been added. A user may select the “train profile” button 630 to train the MLD profile (e.g., to generate a feature set and a classification model for the MLD profile).
Returning to FIG. 4A, at block 415 of method 400 the ML manager receives selection of a memory allocation. At block 420, the ML manager analyzes the training data set using machine learning (e.g., vector machine learning) to train the MLD profile. In one embodiment, the ML manager performs a write lock on the MLD profile during training. In one embodiment, training the MLD profile includes performing feature extraction (block 421), generating a classification model (block 422) and determining a quality of the classification model and feature set (block 423). At block 425, the ML manager displays the results of the analysis in the user interface. The results may include one or more quality metrics, such as a false positive rating, a false negative rating, a memory utilization rating, positive documents that failed extraction, and negative documents that failed extraction. In one embodiment the user can click on the failed extraction information to find out which documents failed extraction.
FIG. 7 illustrates a third view 700 of the user interface of FIG. 5 showing the MLD profile being trained. In one embodiment, the user interface shows each step of MLD profile training, including feature extraction, accuracy calculation, model creation and final processing. A current step of the MLD profile training may be highlighted to show where the ML manager is in MLD profile generation. In one embodiment, the temporary work space is locked while the profile is being trained. Additionally, the memory allocation may not be adjusted while the profile is being trained. This ensures that accurate training results are produced. In one embodiment, a user can select a cancel training option at any time during the training to halt training.
FIG. 8 illustrates a fourth view 800 of the user interface of FIG. 5 showing the MLD profile training results. In one embodiment, MLD profile results include positive documents count 805, a negative documents count 810 and a total documents count 815. In one embodiment, the ML manager displays a list of the features (e.g., words) in the feature set. The training results also include one or more quality metrics for the MLD profile. In one embodiment, the quality metrics include a false positive rating 820, a false negative rating 825 and a memory utilization rating 830. A user may select the false positive rating 820 to view additional false positive rating information, such as specific documents that caused false positives. Additionally, the user may select the false negative rating 825 to view additional false negative rating information, such as specific documents that caused false negatives. If the quality metrics are within quality thresholds, then a “deploy training” button 835 may be active. The user may select the “deploy training” button 835 to deploy the MLD profile. The user may also select a “reject training” button 840 to reject the MLD profile.
Returning to FIG. 4A, at block 430, the ML manager determines whether the analysis results show that the MLD profile meets one or more deployment criteria. If the results meet the deployment criteria, then the method proceeds to block 435. Otherwise, the method continues to block 440.
At block 435, the ML manager enables a MLD profile deployment operation. At block 450, the ML manager receives a deploy command (e.g., based on a user pressing a deploy button in the user interface). If the MLD profile is associated with a policy, the deploy command results in deploying the profile to the detection servers. If the DLP policies are active DLP policies, then the MLD profile becomes active and may immediately be used to police documents. Note that if the MLD profile had a previously deployed version, that version remains deployed until a new version of that MLD profile is deployed. Deploying a newer version of an MLD profile may replace an older version.
At block 440, the ML manager suggests that the user make modifications to the training data set. If the user categorized the documents in the training data set, then the ML manager may identify a category of documents that should be added to the training data set. For example, if a particular category of document caused a large number of false positives, then the ML manager may suggest that the user add more documents of that category to the negative documents in the training data set.
At block 445, the ML manager determines whether any modifications were made to the training data set or to the memory allocation selection. If a change was made to the training data set or to the memory allocation, the method returns to block 420. Otherwise the method ends. The deployed profiles may be shown in a deployed Profile page. This page provides user with a view of the currently deployed profile.
FIG. 4B is a flow diagram illustrating another embodiment for a method 452 of generating and deploying a MLD profile. The method 452 is performed by processing logic that may comprise hardware (circuitry, dedicated logic, etc.), software (such as is run on a general purpose computer system or a dedicated machine), or a combination of both. Method 452 may be performed by a ML manager such as ML manager 130 running on enforcement server 120 of FIG. 1. Though method 452 is described below as being performed by an ML manager, method 452 may also be performed by other processing logic.
Referring to FIG. 4B, at block 454 a ML manager receives a training data set. At block 455, the ML manager determines whether the training data set has been divided into positive examples of sensitive data (e.g., sensitive documents) and negative examples of sensitive data (e.g., non-sensitive documents). If the training data set has not been divided into sensitive and non-sensitive documents (e.g., a user did not specify which documents are sensitive documents and which documents are non-sensitive documents), the method continues to block 456. Otherwise, the method proceeds to block 457.
At block 456, the ML manager divides the documents into sensitive documents and non-sensitive documents. In one embodiment, the ML manager performs LSI (e.g., local weighted LSI) to divide the training data set into multiple distinct sets of documents. A user may then select one or more of the distinct sets of documents for the positive examples of sensitive data and one or more additional distinct sets for the negative examples of sensitive data. Alternatively, the ML manager 456 may automatically make such selections. The method then proceeds to block 457.
At block 457, the ML manager determines a quality of the training data set. In one embodiment, at block 458 the ML manager performs local weighted LSI (or another form of LSI) to determine the quality of the training data set. Using LSI, the quality rating may be based on a number of subcategories that the training data set can be divided into. For example, the training data set may be assigned a high quality rating if sensitive documents in the training data set cannot be subdivided into multiple subcategories and if non-sensitive documents in the training data set cannot be subdivided into multiple subcategories. In another embodiment, at block 459 the ML manager performs cross validation (e.g., k-fold cross validation) to determine the quality of the training data set. Using cross validation, the quality of the training data set may be based on the accuracy of an MLD profile generated from a subset of the documents in the training data set. In one embodiment, the ML manager performs both LSI and cross validation to determine the quality of the training data set. The ML manager may then assign a quality rating to the training data set based on the determined quality.
At block 460, the ML manager determines whether the quality of the training data set is satisfactory. The quality of the training data set may be satisfactory if the training data set is assigned a quality rating that meets or exceeds a quality rating threshold. If the quality of the training data set is satisfactory, the method continues to block 462. Otherwise, the method proceeds to block 468.
At block 462, the ML manager analyzes the training data set using machine learning to create the MLD profile. This may include determining a quality of the MLD profile. In one embodiment, the ML manager performs k-fold cross validation to determine whether the quality of the MLD profile is satisfactory.
At block 466, the ML manager then determines whether a quality of the MLD profile is satisfactory. The quality of the MLD profile may be satisfactory, for example, if the MLD profile has a tested accuracy of 80% or higher. If the quality of the MLD profile is not satisfactory, the method proceeds to block 468. If the quality of the MLD profile is satisfactory, the method continues to block 472.
At block 468, the ML manager makes one or more modifications to the training data set. The modifications may be made based on user input. For example, the user may add one or more additional sensitive or non-sensitive documents to the training data set or remove documents from the training data set. Alternatively, or in addition, the ML manager may automatically remove documents that the ML manager has determined are reducing a quality of the training data set and/or of the MLD profile. This may include removing a particular subset of the sensitive documents or non-sensitive documents from the training data set, for example. The method then returns to block 457.
At block 472, the ML manager adds the MLD profile to a DLP policy. The method then ends.
FIG. 9 is a state diagram 900 showing different states of an ML manager during MLD profile generation/modification, in accordance with one embodiment of the present invention. When a user enters a command to generate a new MLD profile, the ML manager enters a “new” state 905, which causes the ML manager to generate a temporary workspace and an empty MLD profile. From the “new” state 905, the ML manager may enter a “manage profile” state 910. From the “manage profile” state 910, the ML manager can add documents to a training data set based on user input. The ML manager can also roll back the MLD profile to a previous condition and return to the “new” state 905 or proceed to a “training” state 915. If a previously generated MLD profile is being modified, the ML manager may also transition from the “manage profile” state 910 to an “accepted” state 930.
While in the “training” state 915, the ML manager trains the MLD profile. If the training is canceled or otherwise fails, the ML manager transitions to a “training failed/canceled” state 920. After user acknowledgement, the ML manger reverts to the “manage profile” state 910. If the training succeeds, the ML manager transitions to a “trained” state 925. A user may then reject the MLD profile, returning the ML manager to the “manage profile” state 910, or accept the MLD profile, transitioning the ML manager to the “accepted” state 930. From the “accepted” state, the ML manager may deploy the MLD profile.
FIG. 10A is a flow diagram illustrating one embodiment for a method 1000 of generating a MLD profile and deploying the MLD profile to an existing DLP policy. Method 1000 is performed by processing logic that may comprise hardware (circuitry, dedicated logic, etc.), software (such as is run on a general purpose computer system or a dedicated machine), or a combination of both. Method 1000 may be performed by a ML manager such as ML manager 130 running on enforcement server 120 of FIG. 1. Though method 1000 is described below as being performed by an ML manager, method 1000 may also be performed by other processing logic.
Referring to FIG. 10A, at block 1005 a ML manager collects incident reports and/or historical data generated by a DLP policy. The incident reports may include documents misclassified as non-sensitive documents and/or documents misclassified as sensitive documents. Additionally, the incident reports may include documents that were correctly classified as sensitive documents and/or documents that were correctly classified as non-sensitive documents.
At block 1010, the ML manager adds the documents from the incident reports/historical data to a training data set for a MLD profile. At block 1015, the ML manager analyzes the training data set using machine learning to train the MLD profile. This may include generating a feature set, generating a classification model and generating one or more quality metrics for the MLD profile. At block 1020, the ML manager adds the MLD profile to the DLP policy.
Method 1000 shows how incidents from an existing DLP policy may be used to generate a MLD profile. Therefore, a ML manager may perform method 1000 to improve an existing DLP policy to enable it to classify documents as sensitive or non-sensitive that it was previously unsuccessful at classifying.
FIG. 10B is a flow diagram illustrating another embodiment for a method 1025 of generating a MLD profile and deploying the MLD profile to an existing DLP policy. Method 1025 is performed by processing logic that may comprise hardware (circuitry, dedicated logic, etc.), software (such as is run on a general purpose computer system or a dedicated machine), or a combination of both. Method 1025 may be performed by a ML manager such as ML manager 130 running on enforcement server 120 of FIG. 1. Though method 1025 is described below as being performed by an ML manager, method 1025 may also be performed by other processing logic.
Referring to FIG. 10B, at block 1030 a ML manager receives documents that have been misclassified as non-sensitive documents and/or documents that have been misclassified as sensitive documents by an MLD profile. In one embodiment, the ML manager extracts these documents from incident reports.
At block 1035, the ML manager determines whether any retraining criteria are satisfied. In one embodiment, the retraining criteria include a document threshold. The document threshold may be satisfied, for example, when a received number of misclassified documents meets or exceeds the document threshold. In another embodiment, the retraining criteria include a retraining schedule. The retraining criteria may be satisfied, for example, if the retraining schedule indicates that retraining should be performed at a current time.
At block 1040, the ML manager adds the misclassified documents to a training data set for the MLD profile to create a modified training data set. At block 1045, the ML manager determines quality ratings for the training data set and for the modified training data set. At block 1050, the ML manager determines whether the modified training data set has a lower quality rating than the original training data set. If the modified training data set has a lower quality rating, then the ML manager reverts to the original training data set and the method ends. Alternatively, the ML manager may remove a portion of the misclassified documents from the modified training data set, and may repeat block 1045. If the modified training data set does not have a lower quality rating than the original training data set, the method continues to block 1055.
At block 1055, the ML manager retrains the MLD profile using machine learning based on the modified training data set. At block 1060, the ML manager determines quality ratings of the original MLD profile and of the updated MLD profile. At block 1065, the ML manager determines whether the updated MLD profile has a lower quality rating than the original MLD profile. If the updated MLD profile has a lower quality rating, then the ML manager reverts the MLD profile back to its original state, and the method ends. Alternatively, the ML manager may further modify the training data set (e.g., remove one or more misclassified documents or add additional documents) and return to block 1045 or block 1055. If the updated MLD profile does not have a lower quality rating, then the method proceeds to block 1070, and the original MLD profile is replaced in a DLP policy with the updated MLD profile. The method then ends.
FIG. 11 is a flow diagram illustrating one embodiment for a method 1100 of protecting a computing device from data loss using a DLP policy that includes a MLD profile. The method 1100 is performed by processing logic that may comprise hardware (circuitry, dedicated logic, etc.), software (such as is run on a general purpose computer system or a dedicated machine), or a combination of both. Method 1100 may be performed by a DLP agent such as DLP agent 106 running on endpoint device 102A of FIG. 1. Method 1100 may also be performed by a global DLP detection engine such as global DLP detection engine 122 running on endpoint server 115 of FIG. 1.
Referring to FIG. 11, at block 1105 processing logic receives a request to perform an operation on a document. At block 1110, a ML module analyzes the document using a MLD profile to classify the document. At block 1125, processing logic determines whether the document was classified as sensitive or non-sensitive. If the document was classified as sensitive, the method continues to block 1130, and an action specified by a DLP response rule is performed, and an incident report is generated. This may include preventing the operation, generating an incident response report, etc. If the document was classified as non-sensitive, the method proceeds to block 1135, and the operation is performed.
FIG. 12 illustrates a diagrammatic representation of a machine in the exemplary form of a computer system 1200 within which a set of instructions, for causing the machine to perform any one or more of the methodologies discussed herein, may be executed. In alternative embodiments, the machine may be connected (e.g., networked) to other machines in a LAN, an intranet, an extranet, or the Internet. The machine may operate in the capacity of a server or a client machine in client-server network environment, or as a peer machine in a peer-to-peer (or distributed) network environment. The machine may be a personal computer (PC), a tablet PC, a set-top box (STB), a Personal Digital Assistant (PDA), a cellular telephone, a web appliance, a server, a network router, switch or bridge, or any machine capable of executing a set of instructions (sequential or otherwise) that specify actions to be taken by that machine. Further, while only a single machine is illustrated, the term “machine” shall also be taken to include any collection of machines that individually or jointly execute a set (or multiple sets) of instructions to perform any one or more of the methodologies discussed herein.
The exemplary computer system 1200 includes a processing device (processor) 1202, a main memory 1204 (e.g., read-only memory (ROM), flash memory, dynamic random access memory (DRAM) such as synchronous DRAM (SDRAM) or Rambus DRAM (RDRAM), etc.), a static memory 1206 (e.g., flash memory, static random access memory (SRAM), etc.), and a data storage device 1218, which communicate with each other via a bus 1208.
Processor 1202 represents one or more general-purpose processing devices such as a microprocessor, central processing unit, or the like. More particularly, the processor 1202 may be a complex instruction set computing (CISC) microprocessor, reduced instruction set computing (RISC) microprocessor, very long instruction word (VLIW) microprocessor, or a processor implementing other instruction sets or processors implementing a combination of instruction sets. The processor 1202 may also be one or more special-purpose processing devices such as an application specific integrated circuit (ASIC), a field programmable gate array (FPGA), a digital signal processor (DSP), network processor, or the like. The processor 1202 is configured to execute instructions 1226 for performing the operations and steps discussed herein.
The computer system 1200 may further include a network interface device 1222. The computer system 1200 also may include a video display unit 1210 (e.g., a liquid crystal display (LCD) or a cathode ray tube (CRT)), an alphanumeric input device 1212 (e.g., a keyboard), a cursor control device 1214 (e.g., a mouse), and a signal generation device 1220 (e.g., a speaker).
The data storage device 1218 may include a computer-readable storage medium 1224 on which is stored one or more sets of instructions 1226 (e.g., software) embodying any one or more of the methodologies or functions described herein. The instructions 1226 may also reside, completely or at least partially, within the main memory 1204 and/or within the processor 1202 during execution thereof by the computer system 1200, the main memory 1204 and the processor 1202 also constituting computer-readable storage media. The instructions 1226 may further be transmitted or received over a network 1274 via the network interface device 1222.
In one embodiment, the instructions 1226 include instructions for a ML manager, such as ML manager 205 of FIG. 2, and or a software library containing methods that call a ML manager. While the computer-readable storage medium 1224 is shown in an exemplary embodiment to be a single medium, the term “computer-readable storage medium” should be taken to include a single medium or multiple media (e.g., a centralized or distributed database, and/or associated caches and servers) that store the one or more sets of instructions. The term “computer-readable storage medium” shall also be taken to include any medium that is capable of storing, encoding or carrying a set of instructions for execution by the machine and that cause the machine to perform any one or more of the methodologies of the present invention. The term “computer-readable storage medium” shall accordingly be taken to include, but not be limited to, solid-state memories, optical media, and magnetic media.
It is to be understood that the above description is intended to be illustrative, and not restrictive. Many other embodiments will be apparent to those of skill in the art upon reading and understanding the above description. The scope of the invention should, therefore, be determined with reference to the appended claims, along with the full scope of equivalents to which such claims are entitled.

## Claims

1. A method comprising:
receiving a plurality of first documents that were incorrectly classified as sensitive data based on a machine learning-based detection (MLD) profile;
modifying a training data set that was used to generate the MLD profile by adding the first documents to the training data set as negative examples of sensitive data to generate a modified training data set;
determining that there are at least a threshold number of the first documents; and
analyzing, by a processing device, the modified training data set using machine learning to generate an updated MLD profile in response to determining that there are at least the threshold number of the first documents.

2. The method of claim 1, wherein the analyzing is performed on a periodic basis in accordance with an MLD profile retraining schedule.

3. The method of claim 1, further comprising:
assigning a first quality rating to the training data set and a second quality rating to the modified training data set;
determining that the first quality rating is lower than or equal to the second quality rating; and
analyzing the modified training data set in response to determining that the first quality rating is lower than or equal to the second quality rating.

4. The method of claim 1, further comprising:
assigning a first quality rating to the training data set and a second quality rating to the modified training data set;
determining that the second quality rating is higher than or equal to the first quality rating; and
deploying the updated MLD profile in response to determining that the second quality rating is higher than or equal to the first quality rating.

5. The method of claim 3, wherein the first quality rating and the second quality rating are assigned based on performing at least one of latent semantic indexing or k-fold cross validation.

6. The method of claim 1, further comprising:
receiving a second document that was incorrectly classified as non-sensitive data based on the MLD profile; and
modifying the training data set that was used to generate the MLD profile by adding the second document to the training data set as a positive example of sensitive data to generate the modified training data set.

7. A non-transitory computer-readable storage medium having instructions stored therein that, when executed by a processing device, cause the processing device to perform operations comprising:
receiving a plurality of first documents that were incorrectly classified as sensitive data based on a machine learning-based detection (MLD) profile;
modifying a training data set that was used to generate the MLD profile by adding the first documents to the training data set as negative examples of sensitive data to generate a modified training data set;
determining that there are at least a threshold number of the first documents; and
analyzing, by the processing device, the modified training data set using machine learning to generate an updated MLD profile in response to determining that there are at least the threshold number of the first documents.

8. The non-transitory computer-readable storage medium of claim 7, wherein the analyzing is performed on a periodic basis in accordance with an MLD profile retraining schedule.

9. The non-transitory computer-readable storage medium of claim 7, wherein the operations further comprise:
assigning a first quality rating to the training data set and a second quality rating to the modified training data set;
determining that the first quality rating is lower than or equal to the second quality rating; and
analyzing the modified training data set in response to determining that the first quality rating is lower than or equal to the second quality rating.

10. The non-transitory computer-readable storage medium of claim 7, wherein the operations further comprise:
assigning a first quality rating to the training data set and a second quality rating to the modified training data set;
determining that the second quality rating is higher than or equal to the first quality rating; and
deploying the updated MLD profile in response to determining that the second quality rating is higher than or equal to the first quality rating.

11. The non-transitory computer-readable storage medium of claim 9, wherein the first quality rating and the second quality rating are assigned based on performing at least one of latent semantic indexing or k-fold cross validation.

12. The non-transitory computer-readable storage medium of claim 7, wherein the operations further comprise:
receiving a second document that was incorrectly classified as non-sensitive data based on the MLD profile; and
modifying the training data set that was used to generate the MLD profile by adding the second document to the training data set as a positive example of sensitive data to generate the modified training data set.

13. A system comprising:
a memory to store instructions; and
a processing device, coupled to the memory, to execute the instructions to:
receive a plurality of first documents that were incorrectly classified as sensitive data based on a machine learning-based detection (MLD) profile;
modify a training data set that was used to generate the MLD profile by adding the first documents to the training data set as negative examples of sensitive data to generate a modified training data set;
determine that there are at least a threshold number of the first documents and
analyze the modified training data set using machine learning to generate an updated MLD profile in response to the determination that there are at least the threshold number of the first documents.

14. The computing device of claim 13, wherein the processing device is further to:
assign a first quality rating to the training data set and a second quality rating to the modified training data set;
determine that the first quality rating is lower than or equal to the second quality rating; and
analyze the modified training data set in response to the determination that the first quality rating is lower than or equal to the second quality rating.

15. The system of claim 13, wherein the analyzing is performed on a periodic basis in accordance with an MLD profile retraining schedule.

16. The system of claim 13, wherein the processing device is further to:
assign a first quality rating to the training data set and a second quality rating to the modified training data set;
determine that the second quality rating is higher than or equal to the first quality rating; and
deploy the updated MLD profile in response to the determination that the second quality rating is higher than or equal to the first quality rating.

17. The system of claim 14, wherein the first quality rating and the second quality rating are assigned based on performing at least one of latent semantic indexing or k-fold cross validation.

18. The system of claim 13, wherein the processing device is further to:
receive a second document that was incorrectly classified as non-sensitive data based on the MLD profile; and
modify the training data set that was used to generate the MLD profile by adding the second document to the training data set as a positive example of sensitive data to generate the modified training data set.