Explanations as a New Metric for Feature Selection: A Systematic Approach

With the extensive use of Machine Learning (ML) in the biomedical field, there was an increasing need for Explainable Artificial Intelligence (XAI) to improve transparency and reveal complex hidden relationships between variables for medical practitioners, while meeting regulatory requirements. Feature Selection (FS) is widely used as a part of a biomedical ML pipeline to significantly reduce the number of variables while preserving as much information as possible. However, the choice of FS methods affects the entire pipeline including the final prediction explanations, whereas very few works investigate the relationship between FS and model explanations. Through a systematic workflow performed on 145 datasets and an illustration on medical data, the present work demonstrated the promising complementarity of two metrics based on explanations (using ranking and influence changes) in addition to accuracy and retention rate to select the most appropriate FS/ML models. Measuring how much explanations differ with/without FS are particularly promising for FS methods recommendation. While reliefF generally performs the best on average, the optimal choice may vary for each dataset. Positioning FS methods in a tridimensional space, integrating explanations-based metrics, accuracy and retention rate, would allow the user to choose the priorities to be given on each of the dimensions. In biomedical applications, where each medical condition may have its own preferences, this framework will make it possible to offer the healthcare professional the appropriate FS technique, to select the variables that have an important explainable impact, even if this comes at the expense of a limited drop of accuracy.

Automated summary

With the increasing use of Machine Learning in the biomedical field, there is a growing need for Explainable Artificial Intelligence (XAI) to improve transparency and reveal complex relationships between variables for medical practitioners. Feature Selection (FS) is commonly used to reduce the number of variables while preserving information, but there is limited research on the relationship between FS and model explanations. This study demonstrates the complementary nature of explanation-based metrics, accuracy, and retention rate in selecting the most appropriate FS/ML models, providing a framework for offering healthcare professionals the appropriate FS technique based on their preferences.