International Business Machines Corporation
IDENTIFYING A FINDING IN A DATASET USING A MACHINE LEARNING MODEL ENSEMBLE

Abstract:

There is provided a computer-implemented method of identifying a finding in an input sub-dataset of an input dataset using an ensemble of machine learning (ML) models, comprising: obtaining outcomes of an ensemble of ML models generated in response to feeding each input sub-dataset of the input dataset into each ML model of the ensemble trained to generate an outcome indicative of likelihood of a finding depicted in a respective input sub-dataset, creating a two dimensional (2D) outcome dataset storing outcomes of the ensemble, wherein a first dimension denotes the respective ML model and a second dimension denotes the respective input sub-dataset, computing a filtered 2D dataset by applying a filter function to the 2D outcome dataset, and analyzing the filtered 2D dataset to identify specific input sub-dataset(s) likely depicting the finding.