Conditional Gumbel-Softmax for constrained feature selection with application to node selection in wireless sensor networks

Thomas Strypsteen,Alexander Bertrand
2024-06-03
Abstract:In this paper, we introduce Conditional Gumbel-Softmax as a method to perform end-to-end learning of the optimal feature subset for a given task and deep neural network (DNN) model, while adhering to certain pairwise constraints between the features. We do this by conditioning the selection of each feature in the subset on another feature. We demonstrate how this approach can be used to select the task-optimal nodes composing a wireless sensor network (WSN) while ensuring that none of the nodes that require communication between one another have too large of a distance between them, limiting the required power spent on this communication. We validate this approach on an emulated Wireless Electroencephalography (EEG) Sensor Network (WESN) solving a motor execution task. We analyze how the performance of the WESN varies as the constraints are made more stringent and how well the Conditional Gumbel-Softmax performs in comparison with a heuristic, greedy selection method. While the application focus of this paper is on wearable brain-computer interfaces, the proposed methodology is generic and can readily be applied to node deployment in wireless sensor networks and constrained feature selection in other applications as well.
Machine Learning
What problem does this paper attempt to address?
This paper proposes a solution to the problem of node selection in Wireless Sensor Networks (WSNs), specifically in Wireless Electroencephalogram (EEG) sensor networks, where both task optimization and communication distance constraints need to be considered. Traditional feature selection methods typically do not take these constraints into account, while this paper introduces a new method called Conditional Gumbel-Softmax, which allows for end-to-end learning of optimal feature subsets while adhering to pairwise constraints between features. This method is implemented through conditional distributions, ensuring that the selection of each feature depends on another feature, thus ensuring a node configuration that meets the communication distance restrictions. The main application focus of the paper is selecting nodes in the Wireless EEG sensor network that maximize task accuracy while ensuring that the communication distance between nodes is not excessively large, in order to reduce the energy required for communication. The authors validate this method on simulated Wireless EEG sensor networks and compare it with a heuristic greedy selection method. The study analyzes the performance of the network as the constraints become stricter, as well as the performance of Conditional Gumbel-Softmax relative to other methods. Furthermore, the paper points out that although the application is centered around wearable brain-machine interfaces, this method is applicable in general to node deployment in wireless sensor networks and other constrained feature selection problems. Experimental results show that Conditional Gumbel-Softmax can identify high-performance node subsets in certain cases, but it may also suffer from getting trapped in suboptimal solutions. Future research needs to explore how to avoid this situation.