Advancing Interaction-Sensitive Feature Selection: Novel Relief-Based Algorithms, Expanded Comparisons, and Recommendations for Biomedical Data Mining

Abstract:As a precursor to high-dimensional biomedical data modeling, reliable feature selection can reduce computational expense, improve modeling performance, and yield simpler, more interpretable models. However, most filter-based feature selection methods struggle to detect feature interactions, while wrapper or embedded feature selection methods are computationally expensive. Relief-based algorithms (RBAs) are filter methods that are sensitive to feature interactions while mitigating these other limitations. This study (1) refactors, optimizes, and expands the scikit-rebate Python package with existing and newly proposed RBA variants and (2) conducts rigorous RBA benchmark comparisons across diverse genomic simulations. We expand scikit-rebate to include SWRF*, mu-Relief, and 5 novel RBA variants implementing alternative strategies for neighbor selection and feature scoring. All RBAs were evaluated to compare predictive feature ranking and runtime across simulated genomic datasets varying in sample size, number of features, heritability, and underlying association type (e.g. main effects and interactions). All RBAs, except mu-Relief, were proficient in detecting 2-way interactions in noisy data. RBAs utilizing 'far' scoring were best at detecting 2-way interactions - with MultiSWRFDB* top-performing - but were far less sensitive to main effects. SWRF, MultiSWRF, MultiSURF, and MultiSWRFDB yielded top performance across main effect and 2-way interaction datasets with MultiSWRFDB performing best when also considering 3-way interactions. Refactoring of scikit-rebate resulted in 10 to 35-fold reductions in RBA runtimes. The newly introduced RBAs were among the strongest performing, and by robustly retaining both main effects and 2-way epistatic interactions, these algorithms preserve predictive signals for downstream modeling.

Machine Learning (cs.LG)

arXiv:2608.28552 [cs.LG]

(or arXiv:2608.28552v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2608.28552 Focus to learn more arXiv-issued DOI via DataCite (pending registration)

S10_mainEff_100Feat_sigtests.csv

S11_mainEff_1kFeat_sigtests.csv

S12_mainEff_10kFeat_sigtests.csv

S13_mainEff_100kFeat_sigtests.csv

S1_pureEpi_2way_sigtests.csv

S2_2wayEpiHet_sigtests.csv

S3_3wayXOR_sigtests.csv

S4_3wayPureEpi_sigtests.csv

S5_mainEff1Feature_sigtests.csv

S6_additiveEff2Feature_sigtests.csv

S7_additiveEff4Feature_sigtests.csv

S8_2wayEpi_10kFeat_sigtests.csv

S9_2wayEpi_100kFeat_sigtests.csv

Supplementary_Materials.pdf

(9 additional files not shown) You must enabled JavaScript to view entire file list.

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.