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

TL;DR

Optimized Relief-based algorithms (e.g., MultiSWRF*, μ-Relief) significantly enhance detection of 2nd-order interactions with 10-35x faster runtime.

cs.LG 🔴 Advanced 2026-08-29 73 views
Kia Kazemi-Nia Harsh Bandhey Philip J. Freda Ryan J. Urbanowicz
feature selection genetics epistasis algorithm optimization biomedical data

Key Findings

Methodology

By refactoring scikit-rebate, the study incorporated novel Relief variants—SWRF*, μ-Relief, and five new algorithms—focusing on neighbor selection and feature scoring strategies. These methods utilize adaptive neighborhood definitions, sigmoid-based weighting, and dynamic recalculations to improve interaction sensitivity. Extensive benchmarking on simulated genomic datasets with varying sample sizes, feature counts, heritability, and association types demonstrated superior detection of 2- and 3-way epistasis. The optimized code reduced runtime by 10-35 times, enabling large-scale analysis while maintaining detection accuracy.

Key Results

  • All RBAs except μ-Relief effectively detected 2-way interactions in noisy data, with far-scoring variants like MultiSWRFDB outperforming others. The new algorithms achieved high success rates (>95%) in identifying true interactions across diverse simulations. Refactoring led to significant runtime reductions, making these methods practical for large datasets. Notably, MultiSWRFDB excelled in detecting 3-way interactions, while maintaining balanced detection of main effects. These results validate the robustness and efficiency of the proposed methods, marking a substantial advancement over existing approaches.

Significance

This work addresses critical bottlenecks in high-dimensional feature selection, especially in genomics, where detecting complex gene-gene interactions is vital. By enhancing the sensitivity and scalability of Relief-based algorithms, it opens new avenues for understanding genetic architectures underlying diseases. The improved methods facilitate more accurate biomarker discovery, personalized medicine, and genetic research, bridging the gap between computational efficiency and biological complexity. The ability to detect subtle, high-order interactions in large datasets represents a significant leap forward for computational genomics and biomedical data mining.

Technical Contribution

The study introduces innovative variants combining sigmoid neighbor weighting with dynamic neighborhood recalculations, notably MultiSWRF* and MultiSWRFDB. These algorithms balance detection sensitivity for both main effects and high-order interactions while maintaining linear scalability. The refactored scikit-rebate package leverages vectorized operations, drastically reducing runtime. Theoretical guarantees include improved detection power for pure and impure epistasis, and the methods are compatible with high-dimensional data, supporting large-scale genomic analyses. These contributions push the boundary of Relief-based feature selection, enabling more nuanced interaction detection in complex datasets.

Novelty

This is the first systematic integration of sigmoid-based neighbor weighting with dynamic neighborhood recalculations in Relief algorithms, specifically targeting high-order epistasis detection. The development of MultiSWRF* and MultiSWRFDB variants represents a novel approach that combines multiple neighborhood strategies, overcoming the limitations of previous RBAs that struggled with high-order interactions and computational efficiency. These innovations significantly improve the detection of pure and impure interactions in large, noisy datasets, setting new standards in feature selection for biomedical applications.

Limitations

  • While the algorithms perform well on simulated data, their effectiveness on real-world datasets with complex noise and heterogeneity remains to be validated. High computational complexity in some variants may limit scalability in extremely large datasets. The methods primarily focus on genetic data; adaptation to other domains requires further validation. Future work should explore integration with deep learning models and real-world case studies to confirm robustness and generalizability.

Future Work

Future directions include combining these Relief variants with deep learning frameworks for multi-modal data integration, developing adaptive neighborhood strategies for diverse datasets, and validating performance on real-world biomedical datasets. Additionally, efforts will focus on automating parameter tuning, extending applicability to multi-class and continuous outcomes, and exploring parallel computing to further accelerate analysis, ultimately aiming to facilitate real-time, large-scale biomedical discovery.

AI Executive Summary

In the era of high-throughput genomics, identifying relevant features—especially complex gene-gene interactions—remains a formidable challenge. Traditional feature selection methods excel at detecting individual effects but falter when it comes to epistasis, which involves intricate, high-order interactions. Relief-based algorithms (RBAs) have emerged as promising tools due to their local neighborhood sensitivity and scalability. However, their performance in detecting high-order interactions and computational efficiency needed significant improvement.

This study advances the state-of-the-art by refactoring and optimizing the scikit-rebate package, a popular Python library implementing RBAs. The authors introduce five novel variants—MultiSWRF*, SWRF, MultiSWRF, MultiSWRFDB, and MultiSWRFDB*—each combining innovative neighbor selection and feature scoring strategies. These methods incorporate sigmoid-based neighbor weighting, dynamic neighborhood recalculations, and the exclusion of middle-distance instances, aiming to enhance the detection of pure and impure epistasis.

Extensive benchmarking on simulated genomic datasets demonstrates that these new algorithms outperform existing methods in detecting 2- and 3-way interactions, with detection success rates exceeding 95%. The refactoring also reduces runtime by 10-35 times, making large-scale analyses feasible. These improvements are crucial for practical applications in genetic research, enabling more accurate identification of interaction effects underlying complex diseases.

The broader impact of this work lies in its potential to transform feature selection in biomedical data mining, facilitating discoveries in personalized medicine, genetic architecture, and disease mechanisms. Nonetheless, validation on real-world data remains essential, and future work will focus on integrating these algorithms with deep learning and multi-omics data to further enhance their robustness and applicability.

Deep Analysis

Background

随着高通量测序和多组学技术的发展,生物医学数据呈现出高维、多样化的特征空间。传统的单变量统计方法(如χ2检验、ANOVA)在检测主效应方面表现良好,但在捕获基因间复杂交互(如基因-基因互作)方面能力有限。Relief及其变体因其对局部邻域的敏感性,成为检测非线性和高阶交互的有效工具。近年来,针对高阶交互的检测需求不断增长,出现了MultiSURF、SWRF*等算法,但在大规模模拟和真实数据中的性能仍有限。本文在此基础上,优化了scikit-rebate包,开发多种新变体,旨在提升高阶交互检测能力和算法效率,推动生物信息学中的特征筛选技术发展。

Core Problem

现有Relief类算法在检测高阶交互(如3阶及以上)方面存在效率低、敏感性不足的问题,尤其在噪声多或样本少的情况下表现不佳。传统方法计算复杂度高,难以在大规模基因组数据中快速应用。如何在保持检测能力的同时,大幅提升算法速度,成为亟待解决的难题。特别是在生物医学领域,数据规模不断扩大,现有方法难以满足实时分析和大规模筛选的需求,亟需新型高效、敏感的特征选择工具。

Innovation

本研究提出多种结合sigmoid邻域权重和动态邻域定义的RBAs变体,包括MultiSWRF*、SWRF、MultiSWRF、MultiSWRFDB等,创新点在于:

  • �� 采用sigmoid函数平滑调整邻域邻居的权重,增强对交互的敏感性;
  • �� 引入邻域的动态重计算,适应不同样本和特征空间的变化;
  • �� 结合“远”邻域和“中间”邻域,兼顾检测主效应和高阶交互;
  • �� 重构scikit-rebate包,显著提升运行效率,支持大规模模拟数据分析。这些创新突破了传统RBAs在高阶交互检测和大数据处理中的瓶颈。

Methodology

  • �� 设计多种邻域定义策略(如sigmoid邻域、动态邻域、远邻域)以增强交互检测能力;
  • �� 结合邻域权重和特征评分,开发新变体(如MultiSWRF*、MultiSWRFDB);
  • �� 利用NumPy向量化操作优化scikit-rebate的计算速度;
  • �� 在模拟基因组数据上,涵盖不同样本量、特征数、遗传力和关联类型(主效应、交互)进行系统评估;
  • �� 比较检测纯交互和主效应的能力,以及算法运行时间,验证优越性;
  • �� 利用ground truth评估检测的准确性和鲁棒性。

Experiments

采用GAMETES软件生成多种模拟基因组数据,涵盖不同阶数的交互(2阶至5阶)、不同样本量(200-1600)和特征数(20-100000)。比较7个核心RBAs、5个新变体、Mutual Information和随机排序。指标包括:预测特征排名成功率、检测纯交互能力、运行时间。多次重复确保统计显著性,参数敏感性分析验证鲁棒性。

Results

新算法在模拟数据中表现优异,特别在检测2阶和3阶纯交互方面,MultiSWRFDB在考虑3阶交互时检测成功率超过95%。Refactor后,scikit-rebate运行时间缩短10-35倍,极大提升实用性。新变体在噪声环境下依然保持较高检测能力,验证其鲁棒性。整体而言,新算法在保持主效应检测的同时,显著增强对复杂交互的敏感性,为生物信息学分析提供了强有力工具。

Applications

该算法适用于基因组关联分析、疾病机制研究和个性化医疗等场景。能帮助研究者筛选潜在的交互作用基因,为疾病机制提供线索。其高效性支持大规模数据快速筛查,推动精准医疗和药物开发。未来结合深度学习和多组学数据,将进一步提升算法的鲁棒性和适应性。

Limitations & Outlook

虽然新算法在模拟数据中表现优异,但在真实生物医学数据中的泛化能力仍待验证。高阶交互检测在极端噪声或样本极少情况下效果有限。算法复杂度较高,可能在超大数据集上存在计算瓶颈。未来需结合深度学习等技术,提升鲁棒性和扩展性。

Plain Language Accessible to non-experts

想象你在一家工厂里,工厂里有许多不同的机器(特征),每台机器都在生产某种产品(预测目标)。有些机器单独工作效果不错(主效应),但有些必须和其他机器合作(交互)才能发挥作用。传统的方法就像只看每台机器的单独表现,难以发现那些只有合作才能展现的效果。本文提出的方法就像给每台机器装上传感器,能感知它们之间的合作关系,特别是那些合作效果很强但单独不明显的组合。这样,工厂可以更好地找到关键的机器组合,从而提升整体生产效率。

ELI14 Explained like you're 14

想象你在玩一个超级复杂的拼图游戏,有很多拼图块(特征),每块可以单独拼出一些图案(主效应),但有些图案只有把几块拼在一起才能拼出来(交互作用)。以前的方法就像只看每块拼图的样子,难以发现那些只有组合在一起才有的特别图案。现在的研究就像给每块拼图装上了感应器,能检测到它们之间的合作关系,特别是那些只有在一起拼时才出现的图案。这样,你就能更快找到最重要的拼图组合,拼出更完整、更漂亮的图案。

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.

cs.LG