RE-SIZING DATA PARTITIONS FOR ENSEMBLE MODELS IN A MAPREDUCE FRAMEWORK

• US 20150356149A1
• Filed: 02/24/2015
• Published: 12/10/2015
• Est. Priority Date: 06/05/2014
• Status: Active Grant

• Current Assignee: International Business Machines Corporation
• Original Assignee: International Business Machines Corporation
• Inventors: Dagli, C. Ates, McCarroll, Niall Fraser, Vasilenko, Dmitry
• Granted Patent: US 10,459,934 B2
• Time in Patent Office: Days
• US Class Current: 1/1
• CPC Class Codes: G06F 16/212 with details for data modelling support
• CPC Class Codes: G06F 16/2465 Query processing support for facilitating data mining operations in structured databases
• CPC Class Codes: G06F 16/289 Object oriented databases
• CPC Class Codes: G06F 2216/03 Data mining

Abstract

Techniques are described for revising data partition size for use in generating predictive models. In one example, a method includes determining an initial number of base model partitions of data from a plurality of data sources; determining an initial base model partition size based at least in part on the initial number of base model partitions; and evaluating the initial base model partition size at least in part with reference to at least one base model partition size reference. The method further includes determining a finalized number of base model partitions based at least in part on the initial base model partition size; determining a revised base model partition size; and generating revised base models based at least in part on the revised base model partition size, including using a predictive modeling framework to randomly assign input data records from the plurality of data sources into the base model partitions.

Claims

1. A method comprising:

determining an initial number of base model partitions of data from a plurality of data sources;

determining an initial base model partition size based at least in part on the initial number of base model partitions;

evaluating the initial base model partition size at least in part with reference to at least one base model partition size reference;

determining a finalized number of base model partitions based at least in part on the initial base model partition size;

determining a revised base model partition size; and

generating revised base models based at least in part on the revised base model partition size, wherein generating the revised base models comprises using a predictive modeling framework to randomly assign input data records from the plurality of data sources into the finalized number of base model partitions.

2. The method of claim 1, wherein the predictive modeling framework comprises a MapReduce framework.

3. The method of claim 1, wherein using the predictive modeling framework to randomly assign the input data records from the plurality of data sources into the finalized number of base model partitions comprises using the MapReduce framework to randomly assign the input data records from the plurality of data sources into the finalized number of base model partitions.

4. The method of claim 1, further comprising determining the initial number of base model partitions of data based at least in part on a target size of an ensemble model.

5. The method of claim 4, further comprising:

determining a candidate adjustment factor for the number of base model partitions based at least in part on the target size of the ensemble model; and

determining the initial number of base model partitions as a sum of the target size of the ensemble model and the lower of either the candidate adjustment factor or a constant.

6. The method of claim 1, further comprising determining the initial base model partition size by determining a floor of a total number of records in a dataset, minus a size of a holdout sample and a size of a validation sample V, then divided by a number of base partitions.

7. The method of claim 1, wherein the initial number of base model partitions comprises an estimated target number of base model partitions, the method further comprising determining the initial base model partition size based also at least in part on a minimum number of base model partitions.

8. The method of claim 1, further comprising generating an output based at least in part on the revised base model partition size.

9. The method of claim 1, further comprising:

generating, using reduce operations, the plurality of base model partitions based on the plurality of training samples, prior to determining an initial number of base model partitions of data from a plurality of data sources.

10. The method of claim 1, further comprising generating an ensemble model based on the plurality of revised base models.

11. The method of claim 8, further comprising:

generating, using map operations for each of the data sources, a validation sample and a holdout sample from the data sources;

merging the holdout samples into a holdout dataset; and

generating, using a reduce operation, a reference model based on the validation samples.

12. The method of claim 11, further comprising evaluating the ensemble model and the reference model with reference to the holdout dataset.

13. The method of claim 12, further comprising generating a predictive model based at least in part on the evaluating of the ensemble model and the reference model with reference to the holdout dataset.