Skip to main content
Advertisement
Browse Subject Areas
?

Click through the PLOS taxonomy to find articles in your field.

For more information about PLOS Subject Areas, click here.

  • Loading metrics

Weighted nearest neighbours-based control group selection method for observational studies

  • Szabolcs Szekér ,

    Contributed equally to this work with: Szabolcs Szekér, Ágnes Vathy-Fogarassy

    Roles Formal analysis, Methodology, Software, Visualization, Writing – original draft, Writing – review & editing

    vathy@dcs.uni-pannon.hu

    Affiliations Department of Computer Science and Systems Technology, University of Pannonia, Veszprém, Hungary, Healthcare Business Analytics Research and Development Centre, University of Pannonia, Veszprém, Hungary

  • Ágnes Vathy-Fogarassy

    Contributed equally to this work with: Szabolcs Szekér, Ágnes Vathy-Fogarassy

    Roles Conceptualization, Methodology, Supervision, Validation, Visualization, Writing – original draft, Writing – review & editing

    Affiliations Department of Computer Science and Systems Technology, University of Pannonia, Veszprém, Hungary, Healthcare Business Analytics Research and Development Centre, University of Pannonia, Veszprém, Hungary

Abstract

Although in observational studies, propensity score matching is the most widely used balancing method, it has received much criticism. The main drawback of this method is that the individuals of the case and control groups are paired in the compressed one-dimensional space of propensity scores. In this paper, such a novel multivariate weighted k-nearest neighbours-based control group selection method is proposed which can eliminate this disadvantage of propensity score matching. The proposed method pairs the elements of the case and control groups in the original vector space of the covariates and the dissimilarities of the individuals are calculated as the weighted distances of the subjects. The weight factors are calculated from a logistic regression model fitted on the status of treatment assignment. The efficiency of the proposed method was evaluated by Monte Carlo simulations on different datasets. Experimental results show that the proposed Weighted Nearest Neighbours Control Group Selection with Error Minimization method is able to select a more balanced control group than the most widely applied greedy form of the propensity score matching method, especially for individuals characterized with few descriptive features.

Introduction

Case-control studies are widely applied risk analysis methods mainly used in healthcare and social sciences. In retrospective medical studies, the case group typically contains diseased patients, while the control group includes non-diseased individuals, and the study aims to evaluate those risk factors that potentially led to disease occurrence. In the case of prospective studies, typically the effect of a treatment, intervention or other factor is evaluated by comparing the outcome variables of the case (treated) and control (untreated) groups. The popularity of these studies arises from their relatively inexpensive nature, however, the degree of their evidence is lower than randomized trials. The reliability of these studies can be increased by (1) increasing the number of cases included in the study, (2) performing thorough data preparation and data cleaning activities and (3) selecting a proper control group for the case group. The latter highlights the fact that the primary aim of case-control studies is to compare individuals in the case and control groups. A prerequisite of carrying out appropriate analyses requires that the case and control groups are similar on covariates that predict group membership (e.g. treatment assignment) and affect the examined output. However, fulfilling these requirements is not a trivial task. Many articles have highlighted the importance of the proper implementation of a control group selection method and the effect of unbalanced control groups on the result of analyses [15].

Nowadays, propensity score matching (PSM) [6] is the most widely used control group selection method. It is widespread in healthcare analyses [79], and is gaining ground in social sciences [1012] and economics [1315]. PSM matches the individuals of the case and control groups based on their propensity score values, which is the probability of the group (e.g. treatment) assignment conditional on the observed baseline covariates. From a practical viewpoint, various PSM methods exist [1620] that may differ in terms of selection methodology, the ratio of the treated and untreated individuals or the nature of the selection process. On one hand, individuals can be selected into the control group with or without the replacement of the candidates. A general tendency is to apply PSM with replacement when the population from which the control group is selected is too small. Otherwise, matching without replacement can also be used. On the other hand, the ratio of the case and control groups can also be varied. 1:1 matching is common practice, but in the case of large datasets, other implementations e.g 1:M matching can also be used. Thirdly, the variety of the propensity score-based matching methods also increases by the fact that during the selection of the individuals, greedy or optimal matching can also be applied. In the first case, untreated subject whose propensity score is the closest to the score of a given treated subject is selected and matched. When optimal matching is used, the aim is to minimize the total within-pair difference of the propensity scores, and the pairing is optimized globally [21, 22]. Although, in the literature, different extensions (e.g. radius matching, kernel matching, inverse probability weighting, GPS-CDF method, etc. [2326]) of the basic PSM methods were also proposed, in everyday practice the greedy 1:1 matching without replacement is the most widely used form of the PSM method.

Despite the popularity of these methods, they have also received much criticism [2729]. Most of the critical comments point to the possible imbalance between the case group and the control group. It was shown in a recent article [30], that out of 1000 articles using PSM (published between 1983 and 2015), only 6% used any iterative balance checking procedure. In the remaining 94% of the articles, simple 1:1 greedy PSM was applied without any balance checking. King and Nielsen highlighted that PSM is blind to the often large imbalance that can be eliminated by approximating full blocking with other matching methods [30]. Moreover, they pointed out that propensity score matching may increase imbalance even relative to the original data. In a review article published by P. Austin [31] some articles were also identified containing imbalance on the baseline covariates between the case and control groups despite the suitable application of the PSM method. In [32] cardiovascular studies were examined, and authors showed that propensity score methods are not necessarily superior to conventional covariate adjustment.

The main problem with the most widely applied form of the PSM method presumably originates from the application of dimension reduction of the original feature space. Namely, pairing is performed in the 1-dimensional space of the propensity score values, which reduced space might hide the distributions of the original dimensions (features). Covariates that equally affect the probability of individuals belonging to the treated (or diseased) group also affect the value of the propensity score to the same extent. However, the distribution of these variables may be different, and this difference will no longer appear in the 1-dimensional probability values. As long as the matching is performed in the 1-dimensional space of the propensity scores, these differences can not be taken into account during the matching procedure. Publications [33] and [34] also highlighted that matched-pair analysis has to be performed only when matched individuals are highly correlated, but matching subjects having similar propensity scores do not necessarily result in matched subjects with similar covariate values. Based on these considerations, we regard matching in the original n-dimensional vector space or its subspace more suitable than in the 1-dimensional space of propensity scores. The mentioned subspace refers to the covariates which should be included in the propensity score model. Based on recommendations of Austin [35] and Brookhart [36], all variables that affect both the exposure of the group membership and the outcome of the study should always be taken into account.

In this paper, a novel weighted k-nearest neighbour matching method is proposed to select the most suitable individuals into the control group. The proposed method and the most widely used PSM are similar as both methods calculate the probabilities of group membership (e.g. treatment assignment) using a logistic regression (LR) fit. However, PSM pairs the individuals in the reduced 1-dimensional space of propensity scores, the proposed method performs the matching process in the original feature space of covariates utilizing the regression coefficients of the LR model as weight factors of the dimensions. In this way, on the one hand, the proposed method eliminates the main disadvantage of the PSM method, but at the same time retains its advantage by utilizing its regression coefficients as weight factors of the covariates. Monte Carlo simulations show, that the proposed method can overcome the aforementioned shortcomings of the PSM method and may result in a more balanced control group.

In the following, the propensity score matching method and the proposed Weighted Nearest Neighbours Control Group Selection with Error Minimization method (WNNEM) are introduced. Following this, a Monte Carlo study is presented to compare the most widely applied form of the PSM method with the WNNEM method and to highlight the main advantages of the proposed method. To demonstrate the relative utility of the proposed method, some head to head comparisons with other well-known methods (stratified matching, nearest neighbour matching and Mahalanobis metric matching) were also included in our study.

Methods

Matching-based control group selection methods aim to select and pair individuals from a set of potential candidates (XC) to individuals of the case group (XT). Individuals Xi ∈ {XCXT} are characterized by n descriptive features (e.g. age, gender, diagnoses) denoted as f1, f2, …, fn. Therefore, each subject is denoted as an vector of variables, where i = 1, 2, …, L and L = |XCXT|. The aim of control group selection methods is to select such an XUTXC control group, that is balanced to the case group, meaning that the distributions of the variables in both sets are similar. Naturally, XT and XUT must be disjoint sets, that is XTXUT = ∅. To ensure this requirement, XT and XC must also be disjoint (XTXC = ∅).

Propensity score matching

Propensity score matching refers to matching techniques that are based on propensity scores (PS). Propensity score is the conditional probability of treatment assignment based on the observed baseline covariates: (1) where pi denotes the propensity score for the i-th participant, and zi ∈ {0, 1} denotes the treatment variable in such a way that zi = 0 refers to the control (e.g untreated) group and zi = 1 refers to the case (e.g. treated) group. Subjects characterized by the same properties have the same propensity scores.

In retrospective observational studies, the true propensity score is unknown and has to be estimated from available data. Usually, it is estimated using a logistic regression model, but other methods have also been examined and used (e.g. recursive partitioning [37], random forests [38], bagging and boosting [39, 40] and neural networks [35, 41]). When the dependent variable is dichotomous, logistic regression is the most commonly used method to estimate the propensity scores. In this case, treatment status is regressed on the observed baseline covariates and propensity scores are estimated by the fitted model. The multiple linear regression function estimated by the logistic regression model can be defined as follows: (2) where (3) and p is the probability of being exposed. Furthermore, the values bk (k = 1, 2, …, n) are the regression coefficients that describe the relative effects of the covariates (fk) on the status of group (e.g. treatment) assignment. The propensity score estimated by logistic regression is calculated as: (4)

PSM methods essentially follow the nearest neighbour-based approach. That is, that individual is selected from the candidates pairing whose propensity score is the most similar to the propensity score of the individual to be paired in the case group. If multiple subjects from the candidates have equally close propensity scores to the propensity score of the sample subject, one of those is selected at random. As the greedy method does not contain any restrictions concerning the maximum acceptable difference between the propensity scores of the two matched subjects, practical implementations often take into account a threshold parameter for the selection [42]. Individuals within a certain distance of the propensity scores (caliper size) are matched together, and subjects that fall outside this caliper are neglected. Various suggestions have been made for optimal caliper size in the literature [4244], but usually 0.2 of the standard deviation of the logit of the propensity scores is recommended [42].

Weighted nearest neighbours control group selection with error minimization

As it was described in the Introduction, PSM methods are constantly in the midst of criticism due to the imbalance of the covariates observed in some studies. The disadvantage of the most widely used PSM and all other propensity score-based methods is that they perform the pairing of individuals in the 1-dimensional space of the propensity scores. Additionally, uncertainty is also increased by the fact that propensity scores are estimated and not known a priori. Although earlier some methods have been proposed to pair individuals in the original vector space of the features [4547], to the best of our knowledge none of them utilizes the result of fitting a logistic regression model during the matching performed in the original vector space. In this section, such a novel weighted nearest neighbours-based control group selection method is proposed in which the relevance of the covariates is estimated by fitting a logistic regression model, furthermore control group selection is performed in the original vector space of the covariates by calculating the weighted distances of the treated individuals and the possible candidates.

The suggested Weighted Nearest Neighbours Control Group Selection with Error Minimization method (WNNEM) considers each subject as an n-dimensional data point in an n-dimensional space, where each covariate (fk, k = 1, …, n) represents a unique dimension. This way, the problem of control group selection can be interpreted as a distance minimization problem. To select a proper control group, we have to identify such individuals from the candidates that lie close to the individuals of the case group. The concept of lying close can be defined in numerous ways. In the case of the proposed method, multivariate matching is performed in which the adjusted odds ratio (OR) values of the fitted multivariable logistic regression model are utilized as weighting factors of the covariates to compute the distances between the individuals.

Before the whole algorithm is presented, two aspects have to be clarified. Firstly, the term distance and secondly the suggested weighting method has to be specified.

Generally, as individuals may be characterized by different types of variables (binary, nominal, ordinal, numerical), the distance calculation method to be applied must be able to handle different data types. Furthermore, as the significance of the covariates may differ, distances have to be calculated separately for each dimension. The third requirement of distance calculation is that the dissimilarity measures with identical values have to express the same degree of dissimilarity.

To fulfil these requirements, the proposed algorithm calculates the differences for each dimension separately and converts all dissimilarity values into the range of [0, 1]. The distance calculation for different data types is performed as follows:

  • In the case of binary variables, the simple matching distance is calculated. That is, (5) where yields the distance of individuals XiXT and XjXC, xif is the value of individual Xi on binary variable f, and xjf of Xj, respectively.
  • In the case of nominal variables, either the simple matching distance presented before (Eq 5) can be calculated or these variables can be coded as a set of binary variables and the distance can be calculated as the normalized distance of binary features, where the normalization constant represents the number of possible values of the nominal variable.
  • In the case of numerical variables, the dissimilarity measure can be calculated as the difference of the original values. As the distance calculated in this way depends on the range of the original values, normalization is needed to achieve uniform significance for the same dissimilarities and to make them comparable to the dissimilarity measures calculated on other types of attributes. To fulfil this requirement, min-max normalization must be performed separately for each numerical dimension to map the original values into the range of [0, 1] as follows: (6) where xif denotes the original value of individual Xi in the f-th dimension without normalization, minf represent the minimum and maxf the maximum value measured in the f-th dimension taking into account all individuals from XTXC, and yields the normalized value of the individual Xi with regard to the f-th covariate. Subsequently, the distance of individuals XiXT and XjXC is calculated as the differences of their normalized feature values: (7)
  • In the case of ordinal variables, the ordered values have to be coded as ranks and the distance can be calculated as the aforementioned distance of numerical values.

After ensuring that meaning of the dissimilarity values is identical in each dimension, the next step is to weight them according to their relevance to treatment assignment. Previously mentioned, the adjusted odds ratio values of the multivariable logistic regression model fitted on the status of treatment assignment are utilized for this purpose. Generally, the odds ratio is the probability of a characteristic being present divided by the probability of the same characteristic being absent. In our case, the adjusted odds ratio for each independent variable can be obtained by applying the exponential function to the corresponding regression coefficient (bk) obtained from the multivariable logistic regression model (Eq 2). That is, adjusted odds ratios as the weights of the covariates are calculated as follows: (8) where wk denotes the weighting factor of the k-th covariate (k = 1, 2, …, n).

The proposed WNNEM method calculates the distances for individuals XiXT and XjXC the following way: (9) where represents the normalized dissimilarity value of Xi and Xj in the k-th dimension, and wk is the weighting factor of dimension k.

The presented weighted attribute distance is utilized to match the best pairs of candidates (XC) and individuals of the treated group (XT). Basically, the best pair for each XiXT is that XjXC for which dist(Xi, Xj) is minimal. This way, the matching procedure can be regarded an optimization problem, where ∑i,j dist(Xi, Xj) has to be minimized.

Our practical experiments show that for 1:1 matching, an adequate solution can be found even without the use of a complex optimization algorithm. The only problem that needs to be handled during optimization is management of the pairing process of those candidates which lie closest to more than one individual from the case group. These candidates are called candidates in conflict and formally are defined as follows: XjXC is a candidate in conflict if d(Xi, Xj) is minimal for more than one XiXT.

For handling these conflicts, the order of the neighbours has to be determined. Let NN1(Xi) denote the closest and NN2(Xi) the second closest neighbour to individual XiXT. By definition, NN1(Xi) and NN2(Xi) are calculated as follows: (10) (11)

The design of the conflict-handling method to solve the competition of two individuals was inspired by the Vogel-Korda method: instead of a greedy selection, the second neighbours of the treated individuals are also taken into account. That is, the candidate in conflict is matched to the individual for which the error function is greater. The error function is calculated as the distance of the first and second neighbours of the individuals as follows: (12)

In this way, the problem of two competing individuals, Xl and XmXT, is solved. In the case of multiple competing individuals, conflicts are handled by dynamic programming. First, the conflict with the largest error is resolved followed by the others in descending order. This principle is applied iteratively until all the conflicts are resolved.

In summary, the steps of the proposed Weighted Nearest Neighbours Control Group Selection with Error Minimization method (WNNEM) are presented by Algorithm 1. It should be noted that although the presented WNNEM algorithm performs 1:1 matching, it can be easily extended for 1:M matching as well. In this case, in Step 3 the set of unpaired elements (Xunpaired) has to be defined in the way, that each individual of the case group (XT) should be placed M times into the set Xunpaired. The other parts of the algorithm do not change.

Algorithm 1: Weighted Nearest Neighbours Control Group Selection with Error Minimization (WNNEM)

Input: XT case group, XC set of candidate individuals

Output: XUT control group

1 Perform a logistic regression to estimate wk weights for all covariates.

2 Normalize XT and XC collectively using feature scaling and calculate the D = dist(Xi, Xj) distance matrix for all pairs of individuals of XiXT and XjXC by Eq 9.

3 Set

Xunpaired = XT

XUT = ∅

4 Determine NN1(Xi) and NN2(Xi) based on the distance matrix D for all XiXunpaired.

5 Calculate E(Xi) for all XiXunpaired by Eq 12.

6 For all i ∈ {arg(Xunpaired)}

  Set

  If NN1(Xl)∉XUT:

   XUT = XUT ∪ {NN1(Xl)}

   Xunpaired = Xunpaired − {Xl}

   Set m = arg(NN1(Xl))

   Set D(i, m) = ∞ for all i ∈ {arg(XT)}

7 Repeat Steps 4 to 6, till Xunpaired ≠ ∅.

Comparison of the PSM and WNNEM methods—Monte Carlo study

Study design.

To present the effectiveness of the proposed method, a Monte Carlo simulation-based experimental study was performed. In this study, the quality of the control group arising from the proposed WNNEM method was compared to the quality of the control groups arising from the following matching methods: (i) two greedy PSM methods, (ii) nearest neighbour matching (NNM) [48], (iii) Mahalanobis metric matching (MMM) [47] and (iv) stratified matching (SM) [49]. In the cases using PSM methods, the most widely applied form, namely, the greedy 1:1 propensity score matching performed without replacement of individuals was applied. The only difference between the two forms of the PSM methods applied was the ‘caliper size’ setting. In the first case (PSM_02), the caliper size was set as 0.2 of the standard deviation of the logit of the propensity scores. In the second case (PSM_DYN), the caliper size in each simulation was determined dynamically and was set at the minimal value for which 1:1 matching could be performed.

As the greedy nature of the PSM, the nearest neighbour matching and the Mahalanobis metric matching makes them sensitive to the order of the candidates, these methods were run 10 times on each generated dataset in such a way, that the matching order was randomized in each experiment. In contrast, when the WNNEM method or stratified matching was applied, the control group selection was performed only once for each of the generated datasets, because of the deterministic nature of these algorithms.

Datasets.

For the comparisons, three scenarios of datasets with varying feature characteristics were designed. For each scenario, 100 datasets were generated randomly with the same distribution parameters predefined for the covariates. As a result, each scenario contained 100 different datasets with the same number of individuals. That is, in total, we evaluated the accuracy of the various control group selection methods on 300 datasets. Since the PSM methods, the nearest neighbour matching and the Mahalanobis metric matching ran 10 times on each dataset, the presented research is based on a total of 12600 outcome evaluations.

Scenario I In the first scenario, datasets contained 1000 individuals and correspond to the dataset widely applied in theoretical PSM studies [50]. Individuals were characterized by 10 binary variables, each from a Bernoulli distribution (xj ∼ B(0.5), j = 1, …, 10).

The calculation of the probability of treatment assignment was based on the following logistic regression model: (13) where weights b denote low (L), medium (M), high (H) or very high (VH) effect on treatment assignment.

For each subject, a treatment status indicator (Zi) was generated from a Bernoulli distribution with a subject-specified probability equal to pi,treat (Zi ∼ B(pi,treat)). The treated group consisted of subjects where Zi = 1, while subjects where Zi = 0 were assigned to the untreated group (from which the control group was selected). The weights (bL, bM, bH and bVH) were assigned in such a way that approximately 25% of the subjects were treated. The applied weight coefficients were as follows:

  • correction for binary: b0,treat = −1.344090
  • low: bL = log(1.1)
  • medium: bM = log(1.25)
  • high: bH = log(1.5)
  • very high: bVH = log(2.1)

Scenario II The second scenario models such studies in which fewer descriptive variables are available. In this scenario, each individual was characterized by 1 ordinal and 5 binary variables. The ordinal variable represents 5 age groups, while the binary variables may represent, for example, the gender of the subject or various diagnoses. In this scenario, 700 individuals were simulated in each dataset, and the ratio of the candidate subjects to the treated individuals in the 100 datasets was between 2.0 and 3.1.

The assignment of weights to the descriptive variables was as follows: the ordinal variable (x1) has very high effect, the binary variables have low (x2), medium (x3x5) and high (x6) effect on the status of treatment (Eq 14). (14)

The applied weight coefficients, in this case, were the same as in Scenario I.

Scenario III The third scenario is similar to the second one regarding the attributes of individuals and the total number of subjects in each dataset. However, it simulates a more difficult control group selection problem. Although each dataset still contained 700 individuals, the number of treated individuals in the case of the third scenario is higher than in the second one. While in scenario II, the size of the treated group varied between 24.5 and 33.0 percent of the dataset, in the case of the third scenario, it is between 31.7 and 40.7 percent. In other words, the third scenario simulates a more difficult case, where the ratio of the candidate individuals to the treated subjects is lower (XC/XT ≈ [1.5, 2.2]) than in the second scenario (XC/XT ≈ [2.0, 3.1]). Therefore, in this scenario, it is harder to find a proper pair for each treated individual.

To achieve a higher treatment rate, the weight coefficients were modified as follows:

  • correction for binary: b0,treat = −1.344090
  • low: bL = log(1.35)
  • medium: bM = log(1.6)
  • high: bH = log(2.1)
  • very high: bVH = log(3.1)

Methods to evaluate the similarity of the case and control groups.

To validate the result of control group selection, the similarity of the case group and the control group has to be evaluated. In our study, comparison was conducted the following ways: (1) comparing the distributions of the covariates and (2) measuring the similarity of the matched pairs.

In the first case, two different performance evaluations were done. On the one hand, the most common balance metric for comparing propensity score methods, namely the standardized mean difference (SMD) [51] was calculated. Additionally, the most commonly applied goodness of fit tests were also calculated. In the case of continuous covariates with normal distribution, Student two-sample t-test [52] was applied, and in the case of non-normal continuous covariates the Kolmogorov-Smirnov test [53, 54] was used. In the case of testing the similarity of categorical variables, Chi-square test [55] was applied.

The most common drawback of the previously mentioned methods is that they only examine the similarity of the values of a single covariate. As the Hansen and Bowers imbalance test [56] is applied for more complex evaluation in biomedical studies, this aggregated imbalance measure was also calculated in this study. The main virtue of this measure is that it is an aggregated imbalance test, which allows the imbalance of all covariates to be evaluated simultaneously. The application limit of the test, namely that it can be applied for 1:1 matching without replacement, made it ideal for our study. To extend the complex evaluation for all covariate distributions together, the Distribution Dissimilarity Index (DDI) presented in [57] was also utilized. This measure first compares the disparity between histograms for each covariate separately and calculates the differences of the frequencies of the histogram bins before the differences are totalled and normalized.

As the similarity of the distributions of the covariates does not mean that the paired individuals are also similar, two aggregated pairwise similarity measures, the Nearest Neighbour Index (NNI) and the Global Dissimilarity Index (GDI) [57], were also calculated for each comparison. The common characteristic of these measures is that the overall distance of the case and control groups is calculated by evaluating the pairwise distances of the matched subjects. The NNI examines for each treated subject whether the paired individual is the closest individual in the given population. The GDI evaluates the similarity of the paired elements more precisely. First, it calculates the normalized distances of the paired subjects, then these distances are totalled and normalized for all matched pairs. In this way, the Global Dissimilarity Index expresses the magnitude of the distances of the paired individuals as well.

Results

In this section, the results of the study are presented. For a fair comparison, not only two types of the PSM method were compared to the proposed WNNEM method, but other distance-based approaches (nearest neighbour matching, Mahalanobis metric matching) and the widely used stratified matching were also examined.

Scenario I

As mentioned before, in this scenario each dataset contains 1000 individuals, each characterized by 10 binary covariates.

Table 1 shows the mean value of the NNI, GDI and DDI distance measures with their standard deviations. All presented dissimilarity measures may fall within the range of [0, 1], and the value of zero expresses that the case and control groups are identical. Consequently, the greater the dissimilarity value, the higher the difference of the case and control groups is. The results show that the WNNEM method performed the best in terms of control group selection. All dissimilarity values, both for the paired evaluations (NNI and GDI) and the distribution-based evaluation (DDI), are lower in the case of control groups selected by the WNNEM method than by the other methods.

thumbnail
Table 1. Dissimilarity measures for Scenario I.

The Nearest Neighbour Index (NNI) and the Global Dissimilarity Index (GDI) present the results of the evaluation of the similarity of matched pairs. The Distribution Dissimilarity Index (DDI) evaluates the similarity of the histograms of covariates.

https://doi.org/10.1371/journal.pone.0236531.t001

To evaluate the results of the selected control groups, the similarities of the covariates were also evaluated separately. Firstly, the SMD values were calculated for all covariates and for all matching methods applied. The detailed results are presented in Table 2. It can be seen that all matching methods resulted in well-balanced control groups as all SMD values are less than 0.1. There are only small differences between the values in favour of the distance-based methods. On covariate x1 the nearest neighbour matching method, on covariates x2 and x3 the Mahalanobis metric matching and on covariates x4x10 the WNNEM method resulted in the most balanced control groups. Given that covariates x1x3 were associated only with low weights to the group membership (Eq 13), we can say that the WNNEM method gave the most accurate balancing on the major covariates.

thumbnail
Table 2. SMD values for Scenario I.

The table shows the average value of the standardized mean differences and their standard deviations for each covariate separately.

https://doi.org/10.1371/journal.pone.0236531.t002

In the next step, the similarity of the covariates for the case and control groups was tested by the Chi-square test. A higher p-value means a more balanced control group in terms of a given covariate. The detailed results are presented as box plots in Fig 1. As can be seen, the median of the p-values for covariates x4x10 is the highest in the case of WNNEM method, and for covariates x5x10 the interquartile ranges of the WNNEM method is the smallest. Furthermore, the first quartiles from x4 to x10 are also the highest in the case of the WNNEM method.

thumbnail
Fig 1. Distribution of covariates in Scenario I.

Distribution of the Chi-square p-values calculated for each covariate based on all simulations.

https://doi.org/10.1371/journal.pone.0236531.g001

To summarize the balances of the covariates, the widely used Hansen and Bowers test was also evaluated for all matching algorithms. In the Hansen and Bowers test, covariates are considered poorly balanced if the test value is significant (p < 0.05). The higher the p-value, the more similar the case and control groups are. The p-values for all simulations are presented as box plots in Fig 2.

thumbnail
Fig 2. Results of Hansen and Bowers tests in Scenario I.

Comparison of the p-values of the Hansen and Bowers test for the WNNEM method, the stratified matching, the nearest neighbour matching, the Mahalanobis metric matching and two variants of the PSM method.

https://doi.org/10.1371/journal.pone.0236531.g002

Fig 2 clearly shows that the WNNEM method resulted in the most balanced control groups in terms of the Hansen and Bowers test. Not only the interquartile range is the smallest in the case of the WNNEM method, but the tail of the box as well. Furthermore, only a few outlier simulations can be observed, and, except for one simulation, they all have high p-values. All these facts support the advantages of applying the proposed WNNEM method.

Scenario II

The second scenario simulates such studies in which the age of the patients and another 5 binary parameters are considered as covariates.

The overall statistics of the control group selections are presented in Table 3. It can be seen that the proposed WNNEM method performed better in terms of GDI and DDI measures than the other methods. However, in terms of the Nearest Neighbour Index, the lowest value can be seen at the nearest neighbour matching, but the difference is negligible (0.005). Furthermore, we can observe that distance-based algorithms generally give 1 order of magnitude better results than PSM methods, except the DDI value of the PSM_DYN method. Furthermore, by comparing these results to the results of Scenario I, it can be seen, that in this case, the selected control groups are more similar to the case group.

thumbnail
Table 3. Dissimilarity measures for Scenario II.

The Nearest Neighbour Index (NNI) and the Global Dissimilarity Index (GDI) present the results of the evaluation of the similarity of matched pairs. The Distribution Dissimilarity Index (DDI) evaluates the similarity of the histograms of covariates.

https://doi.org/10.1371/journal.pone.0236531.t003

For further evaluation, the similarity of the covariates was also calculated. Evaluating the SMD measures for the covariates separately, we found that for all covariates and for all methods the average values are below 0.1, so the matched control groups are well balanced on each covariate. The similarity of the case and control groups measured by the Chi-square test is presented in Fig 3. It is important to emphasize that in the case of completely missing boxes, the first, second and third quartiles of the p-values were all equal to 1. In the case of partially missing boxes, the median was equal to 1, therefore the third quartile and maximum value were equal. Consistent with the results observed for the NNI, GDI and DDI indices, Fig 3 also emphasizes the better performance of distance-based measures (WNNEM, nearest neighbour matching and Mahalanobis metric matching) for this data set.

thumbnail
Fig 3. Distribution of covariates in Scenario II.

Distribution of the Chi-square p-values calculated for each covariate based on all simulations.

https://doi.org/10.1371/journal.pone.0236531.g003

Evaluating the Hansen and Bowers test (Fig 4), we can establish that all methods have selected very similar control groups to the case groups. The differences between the boxes are marginal, but the PSM method with a dynamically determined caliper size parameter resulted in less similar control groups more times.

thumbnail
Fig 4. Results of Hansen and Bowers tests in Scenario II.

Comparison of the p-values of the Hansen and Bowers test for the WNNEM method, the stratified matching, the nearest neighbour matching, the Mahalanobis metric matching and two variants of the PSM method.

https://doi.org/10.1371/journal.pone.0236531.g004

In this example, we have seen that all methods were able to select perfectly balanced control groups. This is due to the lower variability of features of the individuals and the relatively large number of available candidates.

Scenario III

The third scenario is similar to the second one but it simulates a more difficult control group selection problem as the ratio of the candidate individuals to the treated ones is less than in Scenario II. In this scenario, on average, only 1.7 candidate individuals were available per treated person, while in Scenario II this value was 2.5.

In Table 4 the overall dissimilarity measures are presented. By comparing Tables 3 and 4, it can be seen that in the case of the third scenario, it was harder to select fully balanced control groups. However, Table 4 shows that the distance-based methods were still able to select more balanced control groups than the greedy PSM methods, and the stratified matching resulted in the worst dissimilarity measures. The nearest neighbour matching and the Mahalanobis metric matching seem to give better results in terms of NNI and GDI measures than the WNNEM method, but the deviation is less than 0.01.

thumbnail
Table 4. Dissimilarity measures for Scenario III.

The Nearest Neighbour Index (NNI) and the Global Dissimilarity Index (GDI) present the results of the evaluation of the similarity of matched pairs. The Distribution Dissimilarity Index (DDI) evaluates the similarity of the histograms of covariates.

https://doi.org/10.1371/journal.pone.0236531.t004

Fig 5 details the covariate imbalances separately. As can be seen, the WNNEM method was able to select better-balanced control groups than the PSM methods and the stratified matching on all covariates. Comparing the WNNEM method to other distance-based methods we can see that the WNNEM method was able to achieve more balanced results on covariates with high (x6) and very high (x1) effect on the group assignment (e.g. treatment assignment). The calculated average SMD values were still below 0.1 for all methods.

thumbnail
Fig 5. Distribution of covariates in Scenario III.

Distribution of the Chi-square p-values calculated for each covariate based on all simulations.

https://doi.org/10.1371/journal.pone.0236531.g005

Fig 6 presents the overall results of the Hansen and Bowers tests. By comparing Fig 4 and Fig 6, it can be seen that in the case of the third scenario, it was harder for each method to select a fully balanced control group. However, Fig 6 shows that the WNNEM method was the most suitable to select a balanced control group. In this way, Fig 6 confirms the ability of the WNNEM method to select better control groups in harder situations. However it should be noted, that the greedy PSM with dynamically determined caliper size also gave good results, but in a few cases the resulting control groups are not well balanced. In the case of using the WNNEM method, it happened only one time.

thumbnail
Fig 6. Results of Hansen and Bowers tests in Scenario III.

Comparison of the p-values of the Hansen and Bowers test for the WNNEM method, the stratified matching, the nearest neighbour matching, the Mahalanobis metric matching and two variants of the PSM method.

https://doi.org/10.1371/journal.pone.0236531.g006

Discussion

Although propensity score matching is the most widely used balancing method, many publications have drawn attention to the disadvantages of using it. Probably, the main problem arises from the fact that the selection of individuals happens in a compressed one-dimensional space of the propensity scores. In this paper, a multivariate control group selection method was proposed, which performs the control group selection in the original vector space of the covariates. The proposed Weighted Nearest Neighbours Control Group Selection with Error Minimization method takes into account the nearest neighbours of the treated individuals and tries to solve the problem of candidates in conflict. Candidates in conflict are those individuals who are the closest to more than one treated subject. Distances between the individuals of treated and untreated groups are calculated as the weighted sum of the distances of the covariates. As covariates may have different effects on treatment assignment, the proposed WNNEM method weights them according to their relevance. The weighting factors for the covariates are acquired from the logistic regression model fitted on the status of treatment assignment.

Monte Carlo simulations show that in the case of 1:1 matching, the proposed WNNEM method is able to select a more balanced control group than the most widely applied greedy forms of propensity score matching. In this article, we aimed to emphasize the fact that although the most applied forms of PSM can result in a well-balanced control group, a more similar control group can be reached with a simple nearest neighbour-based method proposed in this article. The proposed method is advantageous when individuals are characterized by fewer covariates and the search space is such small that there exist many individuals for selecting as control which are the most similar pairs of more than one treated subject. If more variables are available to characterize individuals, propensity scores describe the exposure of the individuals more precisely, so individuals in conflict may also be less prevalent. As a result, the selected control group is more balanced using the PSM method.

Furthermore, it must be emphasized that the presented form of the WNNEM method solves the problem locally and does not consider further candidates in the selection process. As a result, the proposed WNNEM method does not always select the best balanced control group. This shortcoming of the proposed method can be eliminated by applying a global optimization method, e.g. particle swarm optimization or simulated annealing. In the next step of our research, the advantages of extending the proposed method in this regard will be examined. However, it should be noted that the method proposed in this article is able to select a better control group for small datasets. Moreover, in the presented simple form, this can be achieved without specifying extra parameters. Because of its simplicity, applying the proposed method may be worthwhile, however, further investigations into this topic are necessary. The other limitation of the presented method, that it can only handle covariates with positively associated with the treatment assignment. In our next work, we plan to extend the method to negatively associated covariates as well.

References

  1. 1. Harris AD, Samore MH, Lipsitch M, Kaye KS, Perencevich E, Carmeli Y. Control-Group Selection Importance in Studies of Antimicrobial Resistance: Examples Applied to Pseudomonas aeruginosa, Enterococci, and Escherichia coli. Clinical Infectious Diseases. 2002;34(12): 1558–1563.
  2. 2. Pell GS, Briellmann RS, Chan CHP, Pardoe H, Abbott DF, Jackson GD. Selection of the control group for VBM analysis: influence of covariates, matching and sample size. Neuroimage. 2008;41(4): 1324–1335.
  3. 3. Behar P, Teixeira P, Fachel J, Kalil AC. The effect of control group selection in the analysis of risk factors for extended spectrum β-lactamase-producing Klebsiella pneumoniae infections. A prospective controlled study. Journal of Hospital Infection. 2008;68(2): 123–129.
  4. 4. Ripollone JE, Huybrechts KF, Rothman KJ, Ferguson RE, Franklin JM. Implications of the Propensity Score Matching Paradox in Pharmacoepidemiology. American journal of epidemiology. 2018;187(9): 1951–1961.
  5. 5. Moser P. Out of Control? Managing Baseline Variability in Experimental Studies with Control Groups. In: Bespalov A, Michel M, Steckler T, editors. Good Research Practice in Non-Clinical Pharmacology and Biomedicine. Handbook of Experimental Pharmacology. Vol 257. Springer, Cham; 2019. pp. 101–117.
  6. 6. Rosenbaum PR, Rubin DB. The central role of the propensity score in observational studies for causal effects. Biometrika. 1983;70(1): 41–55.
  7. 7. Shin SH, Kim SC, Song KB, Hwang DW, Lee JH, Lee D, et al. A comparative study of laparoscopic vs open distal pancreatectomy for left-sided ductal adenocarcinoma: a propensity score-matched analysis. Journal of the American College of Surgeons. 2015;220(2): 177–185. pmid:25529901
  8. 8. Tokuda M, Yamashita S, Matsuo S, Kato M, Sato H, Oseto H, et al. Clinical significance of early recurrence of atrial fibrillation after cryoballoon vs. radiofrequency ablation—A propensity score matched analysis. PLOS ONE. 2019;14(7): e0219269. pmid:31265482
  9. 9. Chuang YW, Shih-Ting H, Tung-Min Y, Chi-Yuan L, Mu-Chi C, Cheng-Li L, et al. Acute pancreatitis risk after kidney transplantation: Propensity score matching analysis of a national cohort. PLOS ONE. 2019;14(9): e0222169. pmid:31509567
  10. 10. Thoemmes FJ, Kim ES. A systematic review of propensity score methods in the social sciences. Multivariate behavioral research. 2011;46(1): 90–118.
  11. 11. Hwang SH, Cappella E. Rethinking Early Elementary Grade Retention: Examining Long-Term Academic and Psychosocial Outcomes. Journal of Research on Educational Effectiveness. 2018;11(4): 559–587.
  12. 12. Xu D, Solanki S, Harlow A. Examining the Relationship Between 2-year College Entry and Baccalaureate Aspirants’ Academic and Labor Market Outcomes: Impacts, Heterogeneity, and Mechanisms. Research in Higher Education. 2019;
  13. 13. Shipman JE, Swanquist QT, Whited RL. Propensity score matching in accounting research. The Accounting Review. 2016;92(1): 213–244.
  14. 14. Cushman DO, De Vita G. Exchange rate regimes and FDI in developing countries: A propensity score matching approach. Journal of International Money and Finance. 2017;77: 143–163.
  15. 15. Rosholm M, Mikkelsen MB, Svarer M. Bridging the gap from welfare to education: Propensity score matching evaluation of a bridging intervention. PLOS ONE. 2019;14(5): e0216200.
  16. 16. Dehejia RH, Wahba S. Propensity score-matching methods for nonexperimental causal studies. Review of Economics and statistics, 2002;84(1): 151–161.
  17. 17. Rubin DB. Matched sampling for causal effects. Cambridge University Press; 2006.
  18. 18. Lee BK, Lessler J, Stuart EA. Weight trimming and propensity score weighting. PLOS ONE. 2011;6(3): e18174 pmid:21483818
  19. 19. Lee W-S. Propensity score matching and variations on the balancing test. Empirical economics. 2013;44(1): 47–80.
  20. 20. Austin PC. A comparison of 12 algorithms for matching on the propensity score. Statistics in medicine. 2014;33(6): 1057–1069.
  21. 21. Rosenbaum PR. Overt bias in observational studies. In: Observational studies. Springer Series in Statistics. Springer; 2002. pp. 71–104.
  22. 22. Rosenbaum PR. Design of observational studies. Springer; 2010.
  23. 23. Baser O. Too much ado about propensity score models? Comparing methods of propensity score matching. Value in Health. 2006;9(6): 377–385.
  24. 24. Austin PC. An introduction to propensity score methods for reducing the effects of confounding in observational studies. Multivariate behavioral research. 2011;46(3): 399–424.
  25. 25. Caliendo M, Kopeinig S. Some Practical Guidance For The Implementation Of Propensity Score Matching. Journal of Economic Surveys. 2008;22(1): 31–72.
  26. 26. Brown DW, DeSantis SM, Greene TJ, Maroufy V, Yaseen A, Wu H, et al. A novel approach for propensity score matching and stratification for multiple treatments: Application to an electronic health record–derived study. Statistics in Medicine. 2020;
  27. 27. Biondi-Zoccai G, Romagnoli E, Agostoni P, Capodanno D, Castagno D, D’Ascenzo F, et al. Are propensity scores really superior to standard multivariable analysis? Contemporary clinical trials. 2011;32(5): 731–740. pmid:21616172
  28. 28. Pearl J. Remarks on the method of propensity score. Statistics in medicine. 2009;28(9): 1415–1416.
  29. 29. Mansournia MA, Jewell NP, Greenland S. Case–control matching: effects, misconceptions, and recommendations. European journal of epidemiology. 2018;33(1): 5–14.
  30. 30. King G, Nielsen R. Why propensity scores should not be used for matching. Political Analysis. 2019;27(4): 435–454.
  31. 31. Austin PC. A critical appraisal of propensity-score matching in the medical literature between 1996 and 2003. Statistics in medicine. 2008;27(12): 2037–2049.
  32. 32. Elze MC, Gregson J, Baber U, Williamson E, Sartori S, Mehran R, et al. Comparison of propensity score methods and covariate adjustment: evaluation in 4 cardiovascular studies. Journal of the American College of Cardiology. 2017;69(3): 345–357. pmid:28104076
  33. 33. Stuart EA. Matching methods for causal inference: A review and a look forward. Statistical science: a review journal of the Institute of Mathematical Statistics. 2010;25(1): 1–21.
  34. 34. Wan F. Matched or unmatched analyses with propensity-score–matched data? Statistics in medicine. 2019;38(2): 289–300.
  35. 35. Austin PC, Grootendorst P, Anderson GM. A comparison of the ability of different propensity score models to balance measured variables between treated and untreated subjects: a Monte Carlo study. Statistics in medicine. 2007;26(4): 734–753.
  36. 36. Brookhart MA, Schneeweiss S, Rothman KJ, Glynn RJ, Avorn J, Stürmer T. Variable selection for propensity score models. American journal of epidemiology. 2006;163(12): 1149–1156.
  37. 37. Setoguchi S, Schneeweiss S, Brookhart MA, Glynn RJ, Cook EF. Evaluating uses of data mining techniques in propensity score estimation: a simulation study. Pharmacoepidemiology and drug safety. 2008;17(6): 546–555.
  38. 38. Zhao P, Su X, Ge T, Fan J. Propensity score and proximity matching using random forest. Contemporary clinical trials. 2016;47: 85–92.
  39. 39. McCaffrey DF, Ridgeway G, Morral AR. Propensity score estimation with boosted regression for evaluating causal effects in observational studies. Psychological methods. 2004;9(4): 403–425.
  40. 40. Lee BK, Lessler J, Stuart EA. Improving propensity score weighting using machine learning. Statistics in medicine. 2010;29(3): 337–346.
  41. 41. Cavuto S, Bravi F, Grassi M, Apolone G. Propensity score for the analysis of observational data: an introduction and an illustrative example. Drug Development Research. 2006;67(3): 208–216.
  42. 42. Austin PC. Optimal caliper widths for propensity-score matching when estimating differences in means and differences in proportions in observational studies. Pharmaceutical statistics. 2011;10(2): 150–161.
  43. 43. Wang Y, Cai H, Li C, Jiang Z, Wang L, Song J, et al. Optimal Caliper Width for Propensity Score Matching of Three Treatment Groups: A Monte Carlo Study. PLOS ONE. 2013;8(12): e81045. pmid:24349029
  44. 44. Lee M. Matching, regression discontinuity, difference in differences, and beyond. Oxford University Press; 2016.
  45. 45. Cochran WG, Rubin DB. Controlling bias in observational studies: A review. Sankhyā: The Indian Journal of Statistics, Series A. 1973;35(4): 417–446.
  46. 46. Rubin DB. Multivariate matching methods that are equal percent bias reducing, I: Some examples. Biometrics. 1976;32(1): 109–120.
  47. 47. Rubin DB. Bias reduction using Mahalanobis-metric matching. Biometrics. 1980;36(2): 293–298.
  48. 48. Rubin DB. Matching to remove bias in observational studies. Biometrics. 1973;29: 159–184.
  49. 49. Anderson DW, Kish L, Cornell RG. On Stratification, Grouping and Matching. Scandinavian Journal of Statistics. 1980;7(2): 61–66.
  50. 50. Austin PC. Comparing paired vs non-paired statistical methods of analyses when making inferences about absolute risk reductions in propensity-score matched samples. Statistics in medicine. 2011;30(11): 1292–1301.
  51. 51. Austin PC. Balance diagnostics for comparing the distribution of baseline covariates between treatment groups in propensity-score matched samples. Statistics in medicine. 2009;28(25): 3083–3107.
  52. 52. Student B. The probable error of a mean. Biometrika. 1908;6(1): 1–25.
  53. 53. Kolmogorov A. Sulla determinazione empirica di una legge di distribuzione. Giornale dell’Istituto Italiano degli Attuari, 1933;4: 83–91.
  54. 54. Smirnov N. Table for estimating the goodness of fit of empirical distributions. The annals of mathematical statistics. 1948;19(2): 279–281.
  55. 55. Pearson K. X. On the criterion that a given system of deviations from the probable in the case of a correlated system of variables is such that it can be reasonably supposed to have arisen from random sampling. The London, Edinburgh, and Dublin Philosophical Magazine and Journal of Science. 1900;50(302): 157–175.
  56. 56. Hansen BB, Bowers J. Covariate balance in simple, stratified and clustered comparative studies. Statistical Science. 2008;23(2): 219–236.
  57. 57. Szekér S, Vathy-Fogarassy Á. How Can the Similarity of the Case and Control Groups be Measured in Case-Control Studies? In Proceedings of IEEE International Work Conference on Bioinspired Intelligence 2019; 33–40.