E FFICIENT W RAPPER F EATURE S ELECTION USING A UTOENCODER AND M ODEL B ASED E LIMINATION
Anonymous authors
Paper under double-blind review
A BSTRACT
We propose a computationally efﬁcient wrapper feature selection method - called
Autoencoder and Model Based Elimination of features using Relevance and Redundancy scores (AMBER) - that uses a single ranker model along with autoencoders to perform greedy backward elimination of features.
The ranker model is used to prioritize the removal of features that are not critical to the classiﬁcation task, while the autoencoders are used to prioritize the elimination of correlated features.
We demonstrate the superior feature selection ability of AMBER on 4 well known datasets corresponding to different domain applications via comparing the accuracies with other computationally efﬁcient state-of-the-art feature selection techniques.
Interestingly, we ﬁnd that the ranker model that is used for feature selection does not necessarily have to be the same as the ﬁnal classiﬁer that is trained on the selected features.
Finally, we hypothesize that overﬁtting the ranker model on the training set facilitates the selection of more salient features.
I NTRODUCTION
Feature selection is a preprocessing technique that ranks the signiﬁcance of features to eliminate features that are insigniﬁcant to the task at hand.
As examined by Yu and Liu (2003), it is a powerful tool to alleviate the curse of dimensionality, reduce training time and increase the accuracy of learning algorithms, as well as to improve data comprehensibility.
For classiﬁcation problems, Weston et al.
(2001) divide feature selection problems into two types: ( a ) given a ﬁxed k (cid:28) d , where d is the total number of features, ﬁnd the k features that lead to the least classiﬁcation error and ( b ) given a maximum expected classiﬁcation error, ﬁnd the smallest possible k .
In this paper, we will be focusing on problems of type ( a ) .
Weston et al.
(2001) formalize this type of feature selection problems as follows.
Given a set of functions y
=
f
( x, α ) , ﬁnd a mapping of data x (cid:55)→ ( x ∗ σ ) , σ ∈ { 0 , 1 } d , along with the parameters α for the function f that lead to the minimization of
[Equation]
subject to (cid:107) σ (cid:107)
= k , where the distribution P ( x, y ) - that determines how samples are generated - is unknown, and can be inferred only from the training set, x ∗ σ = ( x 1 σ 1 , . . .
, x d σ d ) is an elementwise product, V ( · , · ) is a loss function and (cid:107) · (cid:107) 0 is the L 0 -norm.
Feature selection algorithms are of 3 types: Filter, Wrapper, and Embedded methods.
Filters rely on intrinsic characteristics of data to measure feature importance while wrappers iteratively measure the learning performance of a classiﬁer to rank feature importance.
Li et al.
(2017) assert that although ﬁlters are more computationally efﬁcient than wrappers, due to the absence of a learning algorithm that supervises the selection of features, the features selected by ﬁlters are not as good as those selected by wrappers.
Embedded methods use the structure of learning algorithms to embed feature selection into the underlying model to reconcile the efﬁciency advantage of ﬁlters with the learning algorithm interaction advantage of wrappers.
As examined by Saeys et al.
(2007), embedded methods are model dependent because they perform feature selection during the training of the learning algorithm.
This serves as a motivation for the use of wrapper methods that are not model dependent.
Weston et al.
(2001) deﬁne wrapper methods as an exploration of the feature space, where the saliency of subsets of features are ranked using the estimated accuracy of a learning algorithm.
Hence, τ ( σ, α ) in (1) can be approximated by minimizing
[Equation]
subject to σ ∈ { 0 , 1 } d , where τ alg is a classiﬁer having estimates of α .
Wrappers methods can further be divided into three types:
Exhaustive Search Wrappers, Random Search Wrappers, and
Heuristic Search Wrappers.
We will focus on Heuristic Search Wrappers that iteratively select or eliminate one feature at each iteration because unlike Exhaustive Search Wrappers, they are more computationally efﬁcient and unlike Random Search Wrappers, they have deterministic guarantees on the set of selected salient features, as illustrated in Hira and Gillies (2015).
1. M OTIVATION 1.1. R ELEVANCE AND R EDUNDANCY
We hypothesize that the saliency of features is determined by two factors: Relevance and Redundancy.
Irrelevant features are insigniﬁcant because their direct removal does not result in a drop in classiﬁcation accuracy, while redundant features are insigniﬁcant because they are linearly or nonlinearly dependent on other features and can be inferred - or approximated - from them as long as these other features are not removed.
As shown in Fig.
1 and detailed by Guyon et al.
(2008), one does not necessarily imply the other.
Fig.
1 (a) shows 2 highly redundant features (represented by x and y values) that are both relevant, as removal of any of the features will lead to an inability to classify the data and (b) shows 2 features, where the removal of any one feature does not deter the classiﬁcation ability, and thus any of them is irrelevant given the other.
However, they are not highly redundant as the value of any of them cannot be well approximated using the other.
[Figure]
Filter methods are better at identifying redundant features while wrapper methods are better at identifying irrelevant features, and this highlights the power of embedded methods as they utilize aspects of both in feature selection as mentioned in Bol ´ on-Canedo et al.
Since most wrapper methods do not take advantage of ﬁlter method based identiﬁcation of redundant features, there is a need to incorporate a ﬁlter based technique to identify redundant features into wrapper methods, which we address using autoencoders.
1.1. T RAINING THE C LASSIFIER ONLY ONCE
Although wrapper methods often deliver higher classiﬁcation accuracies compared to ﬁlter methods, their computational complexity is often signiﬁcantly higher because the classiﬁer needs to be trained for every considered feature set at every iteration.
For greedy backward elimination wrappers, the removal of one out of d features requires removing each feature separately and training the classiﬁer with the remaining d − 1 features and testing its performance on the cross-validation set.
The feature whose removal results in the highest classiﬁcation accuracy is removed because its removal caused the least impact on performance.
This is the procedure followed by most backward feature selection algorithms such as the Recursive Feature Elimination (RFE) method proposed by Guyon et al.
For iterative greedy elimination of k features from a set of d features, the classiﬁer has to be trained for (cid:80) ki
=1 ( d − i + 1) times, which poses a practical limitation when the number of features is large.
Also, the saliency of the features selected is governed by how good the classiﬁer that ranks the features is, and as such, we need to use state-of-the-art classiﬁers for ranking the features (CNNs for image data, etc.).
These models are often complex and thus, consume a lot of training time which implies a trade-off between speed and the saliency of selected features.
We address this issue by training the the feature ranker model only once.
S TATE OF THE ART
In this section, we describe top-notch fast/efﬁcient feature selection methods that we will be comparing our proposed method to.
With the exception of FQI, the implementations of these methods can be found in the scikit-feature package created by Li et al.
2.0. F ISHER S CORE
The Fisher Score encourages selection of features where feature values within the same class are similar and feature values belonging to different classes are distinct.
Duda et al.
(2012) deﬁne the
Fisher Score for feature f i as
[Equation]
where c is the number of classes, n j represents the number of training examples in class j , µ i represents the mean value of feature f i , µ ij represents the mean value of feature
f i for training examples in class j , and σ 2 ij represents the variance of feature
f i for training examples in class j .
2.0. CMIM
Conditional Mutual Information Maximization (CMIM) is a fast feature selection method proposed in Vidal-Naquet and Ullman (2003) and Fleuret (2004) that iteratively selects features while maximizing the Shannon mutual information function between the feature being selected and class labels, given already selected features.
Li et al.
(2017) deﬁne the CMIM score for feature f i as
[Equation]
where S is the set of currently selected features, X i is the random variable representing the value of feature f i , and I ( X ; Y | Z ) is the conditional mutual information between discrete random variables
X and Y given a random variable Z .
Also, we use empirical distributions to compute mutual information functions based on the training set.
2.0. RFS
Efﬁcient and Robust Feature Selection (RFS) is an efﬁcient feature selection method proposed by
Nie et al.
(2010) that exploits the noise robustness property of the joint (cid:96) 2 , 1 -norm loss function, by applying the (cid:96) 2 , 1 -norm minimization on both the loss function and its associated regularization function.
Li et al.
(2017) deﬁne RFS’s objective function as
[Equation]
where X is the data matrix, Y is the one-hot label indicator matrix, W is a matrix indicating feature contributions to classes, and α is the regularization parameter.
Features are then ranked by the (cid:96)norm values of the corresponding row in the optimal matrix W .
The value of α for our experiments was chosen by performing RFS on a wide range of values and picking the value that led to the highest accuracy on the cross-validation set.
2.0. FQI
Feature Quality Index (FQI) is a feature selection method proposed by De Rajat et al.
(1997) that utilizes the output sensitivity of a learning model to changes in the input, to rank features.
FQI serves as the main inspiration for our proposed method and as elaborated in Verikas and Bacauskiene (2002), the FQI of feature f
i is computed as
[Equation]
where n is the total number of training examples, o j is the output of the model when the j th training example is the input, and o ij is the output of the neural network when the j th training example, with the value of the i th feature set to 0 , is the input.
AMBER 3. S ENSITIVITIES OF WEIGHTS TO FEATURES
During backpropagation, higher losses in the output layer tend to manifest as a result of larger changes, from the optimal, in the values of the weights in the neural network.
Generally, the magnitudes of weights connected to the neurons in the input layer that correspond to more salient features tend to have larger magnitudes and this has been extensively documented by Bauer Jr et al.
Belue and Bauer Jr (1995), and Priddy et al.
Similar to FQI, we measure the relevance of each feature by setting the input to the neuron corresponding to that feature to 0 .
This essentially means that the input neuron is dead because all the weights/synapses from that neuron to the next layer will not have an impact on the output of the neural network which implies that the model that has been trained with all features will experience a degradation in its ability to classify the input data.
Since more salient features possess weights of higher magnitude, these weights inﬂuence the output to a greater extent and setting the values of more salient features to 0 in the input will result in a greater degradation in the ability of the neural network to classify the input compared to when the same is done for less salient features.
This can be measured by the loss given in the output layer where a greater loss corresponds to a greater degradation.
This is the basis of the Weight Based
Analysis feature selection methods outlined by Lal et al.
We further note that we normalize the training set before training by setting the mean of each feature to 0 and the variance to 1 , so that our simulation of feature removal is effectively setting the feature to its mean value for all training examples.
To summarize, the pre-trained model prioritizes the removal of features that are nonrelevant to the classiﬁcation task by simulating the removal of a feature and computing the resulting loss of the model.
Features whose removal results in a lower loss are less relevant and we will refer to the loss value of this model as a feature’s Relevance Score .
3. A UTOENCODERS R EVEAL N ON -L INEAR C ORRELATIONS
It is possible that the weights connected to less salient features possess high magnitudes.
This can take place as these features are redundant in presence of other salient features as described in Sec.
Hence, we use a ﬁlter based technique that is independent of a learning algorithm to detect these redundant features.
We experimented with methods like PCA as detailed by Witten et al.
(2009) and correlation coefﬁcients as detailed by Mitra et al.
but these methods revealed only linear correlations in data, which is why we introduced autoencoders into the proposed method because they reveal non-linear correlations as examined by Sakurada and Yairi (2014).
To eliminate one feature from a set of k features, we train the autoencoder with one hidden layer consisting of k − 1 hidden neurons as illustrated in Fig.
2 using the normalized training set.
This hidden layer can either be dense, LSTM, or of other types depending on the data we are dealing with.
To evaluate a feature, we set its corresponding values in the training set to 0 and pass the set into the autoencoder.
We then take the Mean Squared Error (MSE) between the output and the original input before the values corresponding to the evaluated feature were set to 0 , and perform this for each of the k features separately.
The feature with the lowest MSE is the least salient feature because the other features in the latent space consisting of k − 1 neurons were able to compensate - with least reconstruction error - for the loss of this feature.
We refer to this MSE as a feature’s
Redundancy Score .
3. U SING T RANSFER L EARNING TO PREVENT RETRAINING
To eliminate k out of d features, we pick a state-of-the-art model for the dataset and train it on the training set using part of it as the cross-validation set.
We call this model the Ranker Model (RM) as it allows us to rank the saliency of the features.
Next, we set the input for each of the d features in all the examples of the training set to 0 one at a time in a round-robin fashion to obtain a list of d Relevance Scores after evaluating the modiﬁed training sets on the RM.
Additionally, we train the autoencoder with one hidden layer consisting of d − 1 hidden neurons and pass the same modiﬁed training sets through the autoencoder to obtain d Redundancy Score for each of the d features.
We then divide the Relevance and Redundancy Scores by their corresponding ranges so that they both contribute equally to the ﬁnal decision and add the corresponding Relevance and Redundancy Scores to obtain the Saliency Score .
The feature with the lowest Saliency Score is then eliminated from the training set.
In the context of the RM, elimination means that that feature is permanently set to 0 for all the examples in the training set.
Thus, we can reuse the same RM on the next iteration of AMBER.
In the context of the autoencoder, elimination means that that feature is permanently removed from the training set for all the examples.
This entire process is done iteratively k times to eliminate k features.
AMBER uses the RM and autoencoders to examine both relevance and redundancy relationships among features in the training data that they are already ﬁt for, to iteratively eliminate features.
The pseudocode for AMBER is described in Algorirthm 1.
It is important to note that a feature selection algorithm can be implemented either to remove a speciﬁed number of features or to stop when the accuracy of the learning algorithm ceases to increase depending on the application.
We chose the former method to speciﬁcally take into account the scenarios when this accuracy decreases before it increases later on in the feature selection process.
Further, it is not difﬁcult to implement a variant of AMBER according to the latter objective.
Inputs function AMBER
Train state of the
Initialize for i = 1 to k do
Set dmTrainSet as trainSet where all features in featList are set to
Set autoTrainSet as trainSet where all features in featList are removed
Train autoencoder with one hidden layer containing d − i units using autoTrainSet for j = 1 to d − i + 1 do
Record loss of RM when dmTrainSet is evaluated after setting feature j to
Set cTrainSet as autoTrainset where feature j is set to
Record cTrainSet is evaluated
Normalize
MSEs corresponding
Once the ﬁnal set of k features to be eliminated are determined, they are completely removed from both the training and testing sets.
The ﬁnal architecture is then trained on the training set consisting of d − k features and tested on the testing set also consisting of d − k features.
R ESULTS 4. E XPERIMENTAL S ETUP
We used a GPU server equipped with 3 Nvidia Tesla P100 GPUs, each with 16 GB of memory and used Keras with a TensorFlow backend as the environment of choice.
With the exception of the
RadioML2016.10b dataset for which we used all 3 GPUs, we only used 1 GPU for training.
The experiments were performed 3 times and the average accuracies were plotted at each feature count in Fig.
The source code for AMBER, links to the datasets considered, and the error bars for the comparison plots are available at https://github.com/amber-iclr/AMBER .
4. D ATASETS
Each dataset corresponds to a different domain of data and encompasses both low and high dimensional data to demonstrate the versatility of AMBER.
The ﬁnal models that are trained on the set of selected features are common across all the feature selection methods that are compared and are trained until early stopping is achieved with a patience value of 5 to ensure that the comparisons are fair.
For all the datasets, the softmax activation function is applied to the output layer with the crossentropy loss function.
The test split used for the Reuters and the Wisconsin Breast Cancer dataset is 0 .
2 while the test split used for the RadioML2016.10b dataset is 0 .
The plots in Fig.
3 were jagged when feature counts in decrements of 1 were plotted and thus, in the interest of readability, we plotted them in larger feature count decrements as speciﬁed for each dataset.
Finally, to demonstrate that the ﬁnal model does not necessarily have to be the same as the RM used by AMBER, we used different models as the ﬁnal model and the RM for the MNIST and RadioML2016.10b datasets.
MNIST is an image dataset created by LeCun et al.
(1998) consisting of 60000 28 x 28 grayscale images with 10 classes, each belonging to one of the 10 digits, along with a test set that contains 10000 images of the same dimensions.
The total number of features here is 784 .
The Ranker Model here is a CNN consisting of 2 convolutional layers, a max pooling layer, and 2 dense layers, in that order.
The convolutional layers have 32 and 64 ﬁlters, in order of depth, with kernel sizes of ( 3 x 3 ) for both layers.
The max pooling layer has a of pool size of ( 2 x 2 ) and the dense layers haveand 10 (output layer) neurons, with Softmax activation for the output layer.
ReLU is applied to the remaining layers.
The ﬁnal model used is an MLP model consisting of 3 fully connected layers with 512 , 512 , and 10 (output layer) neurons.
ReLU is applied to each of the layers with 512 neurons and these layers are followed by dropout layers with a dropout rate of 0 . 2 .
Reuters is a text dataset from the Keras built-in datasets that consists of 11228 newswires from Reuters with 46 classes, each representing a different topic.
Each wire is encoded as a sequence of word indices, where the index corresponds to a word’s frequency in the dataset.
For our demonstration, the 1000 most frequent words will be used and thus, the total number of features is 1000 .
The
Ranker Model and the ﬁnal model are the same MLP model consisting of 2 fully connected layers with 512 and 46 (output layer) neurons with ReLU and Softmax activation functions, respectively.
Wisconsin Breast Cancer is a biological dataset created by Street et al.
(1993) that consists of features that represent characteristics of cell nuclei that have been measured from an image of Fine Needle Aspirates (FNAs) of breast mass.
The dataset consists of 569 examples that belong toclasses: malignant or benign.
The total number of features here is 30 .
The Ranker Model and the ﬁnal model are the same MLP model consisting of 4 fully connected layers with 16 , 8 , 6 , and(output layer) neurons, in order of depth.
Softmax is applied to the output layer while ReLU is applied to the remaining layers.
RadioML2016.10b is a datset of signal samples used by O’Shea et al.
(2016) that consists of 1200000 128 -sample complex time-domain vectors with 10 classes, representing different modulation types.
It consists of 20 Signal to Noise Ratio (SNR) values ranging from - 20 dB to 18 dB in increments of 2 dB; we only choose the results of the 18 dB data for better illustration.
Each of the 128 samples consists of a real part and a complex part and thus, the input dimensions are 2 x 128 , where the total number of features is 256 .
This dataset has the unique property that only pairs of features (belonging to the same sample) can be eliminated.
AMBER, like FQI, is powerful in such situations as the pairs of features can be set to 0 to evaluate their collective rank.
This is also useful in the case of GANs, where sets of pixels/features in a 2 -D pool need to be evaluated to craft adversarial attacks as elaborated by Papernot et al.
The other feature selection methods fail in this case because they account for feature interactions between the pairs of features as well, which is one reason for why AMBER outperforms them as it does not.
For the other methods, to eliminate pairs of features belonging to the same sample, we simply added the scores belonging to the two features to obtain a single score for each sample.
The Ranker Model used here is a CLDNN while the ﬁnal model used is a ResNet, both of which are described in Ramjee et al.
4. C LASSIFICATION A CCURACIES
The ﬁnal models’ classiﬁcation accuracy plots using the selected features can be observed in Fig.
We observe the impressive performance delivered by AMBER that generally outperforms that of all 4 considered methods, particularly when the number of selected features becomes very low (about 99% average accuracy with 4 out of 30 features for the Cancer dataset and about 95% average accuracy with 16 out of 128 samples for the RadioML dataset).
The comparisons of the accuracies of the ﬁnal models using the top 25% and 10% of features are given in Table 1.
The results in the last two rows refer to using a version of AMBER without the Autoencoder’s redundancy score, and another version of AMBER where the ranker model is retrained in every iteration, respectively.
Note from the depicted results (purple curve in the ﬁgures) how solely relying on the RM signiﬁcantly reduces
AMBER’s performance, which validates our intuition about the beneﬁt of using the Autoencoder to capture correlations to reduce the generalization error.
Further, we observe how negligible gains are achieved when retraining the RM in every iteration, which comes at a signiﬁcant computational cost, as demonstrated in Table 2, which validates our intuition about simulating the removal of features without retraining for computational efﬁciency while maintaining good performance.
[Table]
[Figure] [Table]
D ISCUSSION
In Sec. 3 and as evidenced by Wang et al.
, we elaborated on how more salient features possess higher magnitudes of weights in the input layer than features that are less salient, which is the property of neural networks that serves as the basis for AMBER.
The performance of AMBER heavily depends on the performance of the RM that ranks the features.
In some cases, however, even the state-of-the-art models do not have high classiﬁcation accuracies.
In such cases, we can obtain better feature selection results with AMBER by overﬁtting the RM on the training set.
[Figure]
We will demonstrate this using the toy example illustrated in Fig. 4 that portrays the architecture used for the RM along with the corresponding hyperparameters.
Feature 1 is x 1 and feature 2 is x 1 x 2 .
Feature 2 is more salient than feature 1 as it is able to form a decision boundary that allows for better classiﬁcation (see (b) and (c)).
Each of these features have 3 weights in the input layer and as expected, the weights connected to feature 2 manifest into weights of higher average magnitude than those belonging to feature 1 as shown in Fig.
As the number of training epochs increases, the difference in the average magnitudes of the weights increases.
This implies that the RM will be able to better rank the saliency of features as the difference between the Relevance Scores of more and less salient features increases.
Thus, we can overﬁt the RM on the training set by training it for a large number of epochs without regularization to enable better feature selection.
[Figure]
C ONCLUDING R EMARKS
AMBER presents a valuable balance in the trade-off between computational efﬁciency in feature selection, in which ﬁlter-based methods excel at, and performance (i.e. classiﬁcation accuracy), in which traditional wrapper methods excel at.
It is inspired by FQI with two major differences: 1 - Instead of making the ﬁnal selection of the desired feature set based on simulating the model’s performance with elimination of only a single feature, the ﬁnal model’s performance in AMBER is simulated with candidate combinations of selected features, 2 - The autoencoder is used to capture redundant features; a property that is missing in FQI as well as most wrapper feature selection methods.
However, we found AMBER to require slightly larger computational time than the considered 4 state-of-the-art methods, and we also found it to require far less time than state-of-the-art wrapper feature selection methods, as it does not require retraining the RM in each iteration.