Source code for menelaus.data_drift.cdbd

from menelaus.data_drift.histogram_density_method import HistogramDensityMethod


[docs]class CDBD(HistogramDensityMethod): """ The Confidence Distribution Batch Detection (CDBD) algorithm is a statistical test that seeks to detect concept drift in classifiers, without the use of labeled data. It is intended to monitor a classifier's confidence scores but could be substituted with any univariate performance related statistic obtained from a learner., e.g. posterior probabilities. This method relies upon three statistics: * KL divergence: the Kullback-Leibler Divergence (KLD) measure * Epsilon: the differences in divergence metrics between sets of reference and test batches. * Beta: the adaptive threshold adapted at each time stamp. It is based on the mean of Epsilon plus the scaled standard deviation of Epsilon. The scale applied to the standard deviation is determined by the ``statistic`` parameter. It is either the number of standard deviations deemed significant (``"stdev"``) or the t-statistic (``"tstat"``). CDBD operates by: #. Estimating density functions of reference and test data using histograms. The number of bins in each histogram equals the square root of the length of reference window. Bins are aligned by computing the minimum and maximum value for each feature from both test and reference window. #. Computing the distance between reference and test distributions. The KL divergence metric is used to calculate the distance between univariate histograms. #. Computing Epsilon. #. Computing the adaptive threshold Beta. #. Comparing current Epsilon to Beta. If Epsilon > Beta, drift is detected. The new reference batch is now the test batch on which drift was detected. All statistics are reset. If Epsilon <= Beta, drift is not detected. The reference batch is updated to include this most recent test batch. All statistics are maintained. Ref. :cite:t:`lindstrom2013drift` Attributes: Epsilon (list): stores Epsilon values since the last drift detection. reference_n (int): number of samples in reference batch. total_epsilon (int): stores running sum of Epsilon values until drift is detected, initialized to 0. bins (int): number of bins in histograms, equivalent to square root of number of samples in reference batch . num_feat (int): number of features in reference batch. lambda (int): batch number on which last drift was detected. distances (dict): For each batch seen (key), stores the distance between test and reference batch (value). Useful for visualizing drift detection statistics. epsilon_values (dict): For each batch seen (key), stores the Epsilon value between the current and previous test and reference batches (value). Useful for visualizing drift detection statistics. Does not store the bootstrapped estimate of Epsilon, if used. thresholds (dict): For each batch seen (key), stores the Beta thresholds between test and reference batch (value). Useful for visualizing drift detection statistics. """ input_type = "batch"
[docs] def __init__( self, divergence="KL", detect_batch=1, statistic="tstat", significance=0.05, subsets=5, ): """ Args: divergence (str): divergence measure used to compute distance between histograms. Default is "H". * "H" - Hellinger distance, original use is for HDDDM * "KL" - Kullback-Leibler Divergence, original use is for CDBD * User can pass in custom divergence function. Input is two two-dimensional arrays containing univariate histogram estimates of density, one for reference, one for test. It must return the distance value between histograms. To be a valid distance metric, it must satisfy the following properties: non-negativity, identity, symmetry, and triangle inequality, e.g. that in examples/cbdb_example.py. detect_batch (int): the test batch on which drift will be detected. See class docstrings for more information on this modification. Defaults to 1. * if ``detect_batch == 1`` - CDBD can detect drift on the first test batch passed to the update method. Total samples and samples since reset will be number of batches passed to HDM plus 1, due to splitting of reference batch * if ``detect_batch == 2`` - CDBD can detect drift on the second test batch passed to the update method. * if ``detect_batch == 3`` - CDBD can detect drift on the third test batch passed to the update method. statistic (str): statistical method used to compute adaptive threshold. Defaults to ``"tstat"`` * ``"tstat"`` - t-statistic with desired significance level and degrees of freedom = 2 for hypothesis testing on two populations. * ``"stdev"`` - uses number of standard deviations deemed significant to compute threshold. significance (float): statistical significance used to identify adaptive threshold. Defaults to 0.05. * if statistic = ``"tstat"`` - statistical significance of t-statistic, e.g. .05 for 95% significance level. * if statistic = ``"stdev"`` - number of standard deviations of change around the mean accepted. subsets (int): the number of subsets of reference data to take to compute initial estimate of Epsilon. * if too small - initial Epsilon value will be too small. Increases risk of missing drift. * if too high - intial Epsilon value will be too large. Increases risk of false alarms. """ super().__init__( divergence=divergence, detect_batch=detect_batch, statistic=statistic, significance=significance, subsets=subsets, )
[docs] def set_reference(self, X, y_true=None, y_pred=None): """ Initialize detector with a reference batch. After drift, reference batch is automatically set to most recent test batch. Option for user to specify alternative reference batch using this method. Args: X (pandas.DataFrame): initial baseline dataset y_true (numpy.array): true labels for dataset - not used by CDBD y_pred (numpy.array): predicted labels for dataset - not used by CDBD """ # Ensure only being used with 1 variable in reference if len(X.shape) > 1 and X.shape[1] != 1: raise ValueError("CDBD should only be used to monitor 1 variable.") super().set_reference(X, None, None)
[docs] def update(self, X, y_true=None, y_pred=None): """ Update the detector with a new test batch. If drift is detected, new reference batch becomes most recent test batch. If drift is not detected, reference batch is updated to include most recent test batch. Args: X (DataFrame): next batch of data to detect drift on. y_true (numpy.ndarray): true labels of next batch - not used in CDBD y_pred (numpy.ndarray): predicted labels of next batch - not used in CDBD """ # Ensure only being used with 1 variable in test if len(X.shape) > 1 and X.shape[1] != 1: raise ValueError("CDBD should only be used to monitor 1 variable.") super().update(X, None, None)