Deep neural networks have shown remarkable performance across diverse machine learning tasks. However, the balance between predictive accuracy and model interpretability remains a persistent challenge: high-performing models often exhibit complex structures defying human understanding, while interpretable (concise) models may sacrifice performance. In this paper, we show that feature interaction can be a crucial perspective when pursuing such balance, and propose flexible-order feature-interaction (FOFI), a new approach to exploit grouped feature interactions as the key to building accurate yet interpretable models. FOFI encourages local feature interactions that are organized into groups, which allows model capacity (parameters) to be distributed in a nuanced manner: at the lower granularity, dense interactions are restricted locally within each group to account for the complexity (performance); at the higher granularity, a flat predictive function is defined at group-level that guarantees the overall interpretability. Furthermore, FOFI is versatile in accommodating feature interactions of arbitrary order among mixed continuous and categorical variables. Extensive experiments on both simulated and real-world datasets showcase the encouraging performance and interpretability of FOFI.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Flexible-Order Feature-Interaction for Mixed Continuous and Discrete Variables with Group-Level Interpretability

  • Zijie Zhai,
  • Junchen Shen,
  • Ping Li,
  • Jie Zhang,
  • Kai Zhang

摘要

Deep neural networks have shown remarkable performance across diverse machine learning tasks. However, the balance between predictive accuracy and model interpretability remains a persistent challenge: high-performing models often exhibit complex structures defying human understanding, while interpretable (concise) models may sacrifice performance. In this paper, we show that feature interaction can be a crucial perspective when pursuing such balance, and propose flexible-order feature-interaction (FOFI), a new approach to exploit grouped feature interactions as the key to building accurate yet interpretable models. FOFI encourages local feature interactions that are organized into groups, which allows model capacity (parameters) to be distributed in a nuanced manner: at the lower granularity, dense interactions are restricted locally within each group to account for the complexity (performance); at the higher granularity, a flat predictive function is defined at group-level that guarantees the overall interpretability. Furthermore, FOFI is versatile in accommodating feature interactions of arbitrary order among mixed continuous and categorical variables. Extensive experiments on both simulated and real-world datasets showcase the encouraging performance and interpretability of FOFI.