# Patent

## Title
Confidence level threshold selection assistance for a data loss prevention system using machine learning

## Abstract

Machine-learning based detection (MLD) profiles can be used to identify sensitive information in documents. The MLD profile can be used to generate a confidence value for the document that expresses the degree of confidence with which the MLD profile can classify the document as sensitive or not. In one embodiment, a data loss prevention system provides or suggests a confidence level threshold to a user of the data loss prevention system by providing a confidence level threshold for the MLD profile to the user, the confidence level threshold to be used as the boundary between sensitive data and non-sensitive data. In one embodiment the provided confidence level threshold is determined by scanning a random data set using the MLD profile.

## Background
RELATED APPLICATIONS
This patent application 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. This patent application is also related to U.S. patent application Ser. No. 13/038,299, filed Mar. 1, 2011.


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 provides an interface that enables users to generate and deploy 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 typically 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 user interface or workflow to enable users to generate their own MLD profiles. However, adjusting a feature set of a user-generated MLD profile can be a challenge for both machine learning experts and non-experts.
SUMMARY
In one embodiment, a computer-implemented method is performed by a data loss prevention (DLP) system. The method includes training a machine learning-based detection (MLD) profile, the MLD profile to be used to classify new data as sensitive data or as non-sensitive data, and providing a confidence level threshold for the MLD profile to a user, the confidence level threshold to be used as the boundary between sensitive data and non-sensitive data. The MLD profile can be used to classify new data by using the MLD profile to assign a confidence value to the new data, and classifying the new data as sensitive if the confidence value of the new data is above the confidence level threshold. Each confidence value is from a range of confidence values, wherein a higher value in the range of confidence values indicates a higher likelihood of sensitive data than a lower value in the range of confidence values.
In one embodiment, providing the confidence level threshold includes scanning a random data set using the MLD profile and determining a confidence value for each datum in the random data set using the MLD profile. Each datum in the random data set can be a document from a random document set. For example, each datum in the random data set can be a Web page selected randomly from a subset of public Web pages. In one embodiment, providing the confidence level threshold means selecting the confidence value of the random datum with the highest confidence value as a suggested confidence level threshold.
In one embodiment, providing the confidence level threshold includes setting a default value of a confidence level threshold user interface element to the provided confidence value, wherein the user can change the provided confidence level threshold using the confidence level threshold user interface element.
In one embodiment, the method also includes testing the provided confidence level threshold using a test document set that is different from the random document set, and selecting a second random data set if the testing of the suggested confidence level shown that the provided confidence level threshold is not acceptable.
In one embodiment, a computer readable storage medium including instructions is provided, that, when executed by a processing device, cause the processing device to perform a method that includes training a machine learning-based detection (MLD) profile, the MLD profile to be used to classify new data as sensitive data or as non-sensitive data, and providing a confidence level threshold for the MLD profile to a user, the confidence level threshold to be used as the boundary between sensitive data and non-sensitive data.
In one embodiment, a computing device is provided that includes a memory to store instructions for a machine learning manager, and a processing device to execute the instructions, the instructions causing the processing device to train a machine learning-based detection (MLD) profile, the MLD profile to be used to classify new data as sensitive data or as non-sensitive data, and provide a confidence level threshold for the MLD profile to a user, the confidence level threshold to be used as the boundary between sensitive data and non-sensitive data.



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. 4 is a flow diagram illustrating one 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 histogram representation of a distribution of confidence values for a good and a bad training data set according to one embodiment.
FIG. 10 is a flow diagram illustrating one embodiment for a method of suggesting a confidence level threshold for an MLD profile.
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, providing, or suggesting a confidence level threshold to be used in machine learning-based detection (MLD) profiles for a data loss prevention (DLP) system are described. 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”, “displaying”, “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 reside in 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 128 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. In one embodiment, one feature of an MLD profile is the confidence threshold to use to separate the sensitive data from the non-sensitive date. The ML manager 130 can also provide the interface to provide and configure the confidence threshold. 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 user initiated 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 an 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-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 statistical model for data classification that includes a map of support vectors that represent boundary features. 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.
In one embodiment, the ML module 225 applies the MLD profile 265 to unclassified data—such as file 235—and generates a confidence value for the file 235. If the confidence value meets or exceeds a confidence level threshold, then the file 235 is classified as sensitive, otherwise the file 235 is classified as non-sensitive. In one embodiment, this confidence value is user-selected and configurable. In one embodiment, the ML manager 305 assists the user by providing and/or suggesting a confidence level threshold to use with the MLD profile.
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 MLR profile trainer 325 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 statistical 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, 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 adds 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. 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 325 for a training data set 352 until a threshold number of documents have been added to the training data set. 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 EX, 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-ISSF), 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), which will be discussed in greater detail below
In one embodiment, quality analyzer 340 analyzes a quality of the MLD profile 365 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). 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.
In one embodiment, the ML manager 305 also includes a confidence level threshold module 385. The confidence level threshold module 385 can determine and provide the suggested confidence level threshold by using the MLD profile tester 320 to scan a random document set 390. In one embodiment, the highest confidence value of the random document set 390 is selected as the suggested confidence level threshold. Some embodiments of providing confidence level threshold suggestions or settings to a user are described in more detail with reference to FIG. 9 and FIG. 10 further below.
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 315. 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. 4 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. 4, 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 515 that, when selected, opens a window in which a user can type in a profile name and/or a profile description.
Returning to FIG. 4, 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 has 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. 4, 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. 4, 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 432. Otherwise, the method continues to block 440. At block 432, a confidence level threshold for the MLD profile is determined. The confidence level threshold can be used to scan documents using the MLD profile as explained further below. One embodiment of a process for determining the confidence level threshold is described below with reference to FIG. 10.
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.
Once an MLD profile has been deployed, it can be used to scan documents as part of a DLP policy, as described with reference to FIG. 2. In operation, for each scanned document, the DLP agent 205 uses the feature set 280 and the classification model 275 of the MDL profile 265 to generate a confidence value for the scanned document. The confidence value is a numeric representation of the match between the MDL profile and the scanned document. In one embodiment, the confidence value is a represented as an integer value between zero and one hundred (0-100).
In one embodiment, a confidence value of 100 indicates an absolute certainty that the scanned document matches the training data consisting of sensitive documents (the positive training data). Similarly, a confidence value of 0 indicates an absolute certainty that the scanned document matches the training data consisting of non-sensitive documents (the negative training data). A confidence value of 90 would thus indicate a high confidence that the scanned document is sensitive, while a confidence value of 10 would indicate a similarly high confidence that the scanned document is non-sensitive. A confidence value of 50 indicates an equal likelihood that the scanned document is sensitive or non-sensitive.
In one embodiment, the DLP agents uses a confidence level threshold to trigger a rule or policy violation. That is, a certain confidence value is designated as the confidence level threshold. Thus, if the computed confidence value for a document is equal to or greater than the confidence level threshold, then the scanned document will be classified as a sensitive document. Conversely, if the confidence value of the scanned document is less than the confidence level threshold, then the scanned document will be classified as a non-sensitive document.
In one embodiment, the confidence level threshold is a user-selected value. For example a slider-bar UI element can be used to select the confidence level threshold, or simply a data entry box. In other embodiments, the confidence level threshold can be calculated and set automatically by the ML manager 305. In one embodiment, even where the user selects the confidence threshold value, the ML manager 305 recommends or suggests a confidence level threshold to the user to save the user time over a trial and error process.
Determining a suggested confidence level threshold is complicated by fact that, for a good training data set, there is a wide range of possible options. Furthermore, in one embodiment it can be desirable to set the confidence level threshold as low as possible, to capture all of the potentially sensitive documents while simultaneously minimizing false positive results. This is now illustrated with reference to FIG. 9.
If the quality analyzer and the processing described with reference to FIG. 4 (eg. Block 420-445) has been successful, the training data set is “good” and contributes to a useful profile. FIG. 9 illustrates the relationship between confidence values and the quality of the training data set. The good training set histogram 902 illustrates that when the training data set is scanned by the MLD profile created using the training data set, the confidence values between the positive documents and the negative documents are clearly separated. The positive training documents are distributed between the high marker 907 (approximately 80) and 100 and the negative training documents are distributed between 0 and the low marker 905 (approximately 20), according to an approximate distribution shown in the histogram 902, leaving a gap 910 between the distributions.
In contrast, the bad training set histogram 920 shows the distributions of confidence values for a poorly selected training data set. For a bad training data set, there is overlap between the confidence values for the positive and the negative training documents, and there is no gap 910 between the confidence values for the positive and negative training documents. In one embodiment, the quality of the training data set is checked—for example as described with reference to FIG. 4—before the determination of an appropriate suggested confidence level threshold. Thus, the histogram of the training data set scanned with the MLD profile will appear similar to the good training set histogram 902, with a gap 910 between the confidence values of the positive and negative training documents.
One embodiment for determining a suggested confidence level threshold is now described with reference to FIG. 10. FIG. 10 is a flow diagram illustrating one embodiment for a method 1000 of determining a suggested confidence level threshold. 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 the confidence level threshold module 385 of an ML manager such as ML manager 130 running on enforcement server 120 of FIG. 1 in collaboration with the user interface 335 of FIG. 3.
Referring to FIG. 10, at block 1005 a random document set is generated. In one embodiment, a large random publicly available non-sensitive document repository is accessed and a statistically significant number of such documents is selected. For example, in one embodiment, the random document set generation is performed by selecting approximately 1,000 Wikipedia (the free online encyclopedia) pages at random. Various randomizing and pseudo-randomness generation techniques can be used for selecting such pages at random.
In other embodiments, other non-sensitive large document repositories can be used. The documents of the random document set need not be in any format—such as files or web pages—and may be made up of unstructured data. Thus, the terms “random document set” and “random data set” can be used interchangeably. In one embodiment, the document repository being used for selecting the random document set should have at least 10,000 documents, but more or fewer number can be used as well.
According to another embodiment, the random document set has already been generated as described with reference to block 1005, and the random document set need only be selected for processing. For example, a pre-generated random document set may be stored in or accessible to the ML manager 305, such as the random document set 390 shown in FIG. 3. Thus, in one embodiment, the same random document set can be used for confidence level threshold suggestion processing for each MLD profile.
At block 1010, the random document set is scanned using the MLD profile. This generates a confidence value for each document in the random document set. At block 1015, the highest confidence value is selected as the suggested confidence level threshold. Since the random document set is made up, in one embodiment, of publicly available non-sensitive documents, then setting the confidence level threshold below the highest confidence value would generate at least one false positive. Setting the suggested confidence level threshold to the maximum confidence value of the random data set does not guarantee the elimination of false positives in operation, but on the selected random data set at least, no false positives would result.
At block 1020, the suggested confidence level threshold is tested. In one embodiment, this is done by scanning a test document set using the MLD profile and observing how many—if any—false positives or false negatives result. In other words, such a test can ensure that the suggested confidence level threshold is indeed somewhere in the gap 910 shown in FIG. 9. In one embodiment, no false positives or false negatives are acceptable when the training data is scanned. The test document set, in one embodiment, is made up documents to be protected to test for false negatives and documents that contain no sensitive data to test for false positives.
At block 1025, a determination is made as to whether the suggested confidence level threshold was acceptable during the testing performed at block 1020. If the suggested confidence level threshold is acceptable, the process terminates and the suggested confidence level threshold has been determined. If, however, the suggested confidence level threshold is determined to be unacceptable, then, at block 1030, the random document set can be modified or regenerated, or a different random document set can be selected. The processing then continues at block 1010, as described above, with the scanning of the newly selected, modified, or regenerated random document set.
After the suggested confidence level threshold is determined, it can be presented or provided to the user. In one embodiment, the processing described in FIG. 10 is performed automatically by the confidence level threshold module 385 and the ML manager 305 without any user interaction after the user selects the training data set. Thus, once the training data set is acceptable, a UI presentation element, such as the temporary workspace view in FIG. 8, can display the suggested confidence level threshold to the user. In other embodiments, the suggested confidence level threshold is automatically set as the confidence level threshold for the MLD profile, which the user can then edit or configure.
During deployment of the MLD profile in an active policy, the user can further adjust the confidence level threshold to respond to false positives and false negatives caused by the confidence level threshold. However, the suggested confidence value threshold provides the user of the ML manager 305 with a good starting point for further configuration.
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 using the confidence level threshold, as explained above. If the document was classified as sensitive, the method continues to block 1130, and an action specified ay 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 305 of FIG. 3, 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:
training, by a processing device, a machine learning-based detection (MLD) profile, wherein the MLD profile is used to classify new data as sensitive data or as non-sensitive data; and
providing a confidence level threshold for the MLD profile to a user, wherein the confidence level threshold is used as a boundary between the sensitive data and the non-sensitive data, wherein providing the confidence level threshold comprises setting a default value of a confidence level threshold user interface element to the confidence level threshold, and wherein the confidence level threshold user interface element allows the user to change the confidence level threshold,
wherein the MLD profile is used to classify new data by using the MLD profile to assign a confidence value to the new data and classifying the new data as the sensitive data in response to determining that the confidence value of the new data is above the confidence level threshold, wherein the confidence value is from a range of confidence values, and wherein a higher value in the range of confidence values indicates a higher likelihood of being the sensitive data than a lower value in the range of confidence values.

2. The method of claim 1, further comprising scanning a random data set using the MLD profile to determine the confidence level threshold.

3. The method of claim 2, wherein scanning the random data set using the MLD profile comprises determining a confidence value for each random datum in the random data set using the MLD profile.

4. The method of claim 3, wherein each random datum in the random data set comprises a document from a random document set.

5. The method of claim 3, wherein each random datum in the random data set comprises a web page selected randomly from a subset of public web pages.

6. The method of claim 3, wherein scanning the random data set using the MLD profile further comprises selecting the confidence value of the random datum with the highest confidence value as the confidence level threshold.

7. The method of claim 3, wherein the method further comprises testing the confidence level threshold using a test data set that is different from the random data set to determine whether the confidence level threshold has a threshold gap between a distribution of confidence values for ones of the test data set classified as the sensitive data and a distribution of confidence values for ones of the test data set classified as the non-sensitive data.

8. The method of claim 7, further comprising selecting a second random data set in response to the determination that the confidence level threshold does not have the threshold gap between the distribution of confidence values for the ones of the test data set classified as the sensitive data and the distribution of confidence values for the ones of the test data set classified as the non-sensitive data.

9. The method of claim 1, wherein training comprises extracting features using a user-selectable feature extraction algorithm.

10. 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:
training, by the processing device, a machine learning-based detection (MLD) profile, wherein the MLD profile is used to classify new data as sensitive data or as non-sensitive data;
scanning a random data set using the MLD profile to determine a confidence level threshold; and
providing the confidence level threshold for the MLD profile to a user, wherein the confidence level threshold is used as a boundary between the sensitive data and the non-sensitive data,
wherein the MLD profile is used to classify new data by using the MLD profile to assign a confidence value to the new data and classifying the new data as the sensitive data in response to determining that the confidence value of the new data is above the confidence level threshold, wherein the confidence value is from a range of confidence values, and wherein a higher value in the range of confidence values indicates a higher likelihood of being the sensitive data than a lower value in the range of confidence values.

11. The non-transitory computer-readable storage medium of claim 10, wherein providing the confidence level threshold comprises setting a default value of a confidence level threshold user interface element to the confidence level threshold, and wherein the confidence level threshold user interface element allows the user to change the confidence level threshold.

12. The non-transitory computer-readable storage medium of claim 10, wherein scanning the random data set using the MLD profile comprises determining a confidence value for each random datum in the random data set using the MLD profile.

13. The non-transitory computer-readable storage medium of claim 12, wherein each random datum in the random data set comprises a document from a random document set.

14. The non-transitory computer-readable storage medium of claim 12, wherein scanning the random data set using the MLD profile further comprises selecting the confidence value of the random datum with the highest confidence value as the confidence level threshold.

15. The non-transitory computer-readable storage medium of claim 10, wherein training comprises extracting features using a user-selectable feature extraction algorithm.

16. A system comprising:
a memory to store instructions for a machine learning manager; and
a processing device to execute the instructions to:
train a machine learning-based detection (MLD) profile, wherein the MLD profile is used to classify new data as sensitive data or as non-sensitive data;
scan a random data set using the MLD profile to determine a confidence level threshold; and
provide the confidence level threshold for the MLD profile to a user, wherein the confidence level threshold is used as a boundary between the sensitive data and the non-sensitive data,
wherein the MLD profile is used to classify new data by using the MLD profile to assign a confidence value to the new data and classifying the new data as the sensitive data in response to determining that the confidence value of the new data is above the confidence level threshold, wherein the confidence value is from a range of confidence values, and wherein a higher value in the range of confidence values indicates a higher likelihood of being the sensitive data than a lower value in the range of confidence values.

17. The system of claim 16, wherein to scan the random data set the processing device is to determine a confidence value for each random datum in the random data set using the MLD profile, and wherein to determine the confidence level threshold the processing device is to select the confidence value of the random datum with the highest confidence value as the confidence level threshold.

18. The system of claim 17, wherein each random datum in the random data set comprises a document from a random document set, and wherein the processing device is further to test the confidence level threshold using a test document set that is different from the random document set to determine whether the confidence level threshold has a threshold gap between a distribution of confidence values for sensitive ones of the test document set and a distribution of confidence values for non-sensitive ones of the test document set.

19. The system of claim 16, wherein to provide the confidence level threshold the processing device is to set a default value of a confidence level threshold user interface element to the confidence level threshold, and wherein the confidence level threshold user interface element allows the user to change the confidence level threshold.

20. The system of claim 16, wherein training comprises extracting features using a user-selectable feature extraction algorithm.