FeatureEnVi: Visual Analytics for Feature Engineering Using Stepwise Selection and Semi-Automatic Extraction Approaches

Abstract

The machine learning (ML) life cycle involves a series of iterative steps, from the effective gathering and preparation of the data, including complex feature engineering processes, to the presentation and improvement of results, with various algorithms to choose from in every step. Feature engineering in particular can be very beneficial for ML, leading to numerous improvements such as boosting the predictive results, decreasing computational times, reducing excessive noise, and increasing the transparency behind the decisions taken during the training. Despite that, while several visual analytics tools exist to monitor and control the different stages of the ML life cycle (especially those related to data and algorithms), feature engineering support remains inadequate. In this paper, we present FeatureEnVi, a visual analytics system specifically designed to assist with the feature engineering process. Our proposed system helps users to choose the most important feature, to transform the original features into powerful alternatives, and to experiment with different feature generation combinations. Additionally, data space slicing allows users to explore the impact of features on both local and global scales. FeatureEnVi utilizes multiple automatic feature selection techniques; furthermore, it visually guides users with statistical evidence about the influence of each feature (or subsets of features). The final outcome is the extraction of heavily engineered features, evaluated by multiple validation metrics. The usefulness and applicability of FeatureEnVi are demonstrated with two use cases and a case study. We also report feedback from interviews with two ML experts and a visualization researcher who assessed the effectiveness of our system.

This manuscript is accepted for publication in the IEEE Transactions on Visualization and Computer Graphics Journal (IEEE TVCG)

References in corpus (14)

• Scikit-learn: Machine Learning in Python
• XGBoost: A Scalable Tree Boosting System
• Human Factors in Model Interpretability: Industry Practices, Challenges, and Needs
• An Empirical Analysis of Feature Engineering for Predictive Modeling
• The State of the Art in Enhancing Trust in Machine Learning Models with the Use of Visualizations
• t-viSNE: Interactive Assessment and Interpretation of t-SNE Projections
• StackGenVis: Alignment of Data, Algorithms, and Models for Stacking Ensemble Learning Using Performance Metrics
• E.T.-RNN: Applying Deep Learning to Credit Loan Applications
• Explaining Vulnerabilities to Adversarial Machine Learning through Visual Analytics
• FeatureExplorer: Interactive Feature Selection and Exploration of Regression Models for Hyperspectral Images
• Feature Engineering for Predictive Modeling using Reinforcement Learning
• Progressive Data Science: Potential and Challenges
• VisEvol: Visual Analytics to Support Hyperparameter Search through Evolutionary Optimization
• FDive: Learning Relevance Models using Pattern-based Similarity Measures

Cited by in corpus (4)

• HardVis: Visual Analytics to Handle Instance Hardness Using Undersampling and Oversampling Techniques
• Visual Analytics for Explainable and Trustworthy Artificial Intelligence
• DeforestVis: Behavior Analysis of Machine Learning Models with Surrogate Decision Stumps
• VFLens: Co-design the Modeling Process for Efficient Vertical Federated Learning via Visualization