Tree-like Pairwise Interaction Networks

Explainable & Ethical AI
Published: arXiv: 2508.15678v1
Authors

Ronald Richman Salvatore Scognamiglio Mario V. Wüthrich

Abstract

Modeling feature interactions in tabular data remains a key challenge in predictive modeling, for example, as used for insurance pricing. This paper proposes the Tree-like Pairwise Interaction Network (PIN), a novel neural network architecture that explicitly captures pairwise feature interactions through a shared feed-forward neural network architecture that mimics the structure of decision trees. PIN enables intrinsic interpretability by design, allowing for direct inspection of interaction effects. Moreover, it allows for efficient SHapley's Additive exPlanation (SHAP) computations because it only involves pairwise interactions. We highlight connections between PIN and established models such as GA2Ms, gradient boosting machines, and graph neural networks. Empirical results on the popular French motor insurance dataset show that PIN outperforms both traditional and modern neural networks benchmarks in predictive accuracy, while also providing insight into how features interact with each another and how they contribute to the predictions.

Paper Summary

Problem
Predictive modeling in tabular data often struggles to capture the complex interactions between multiple input features. This is a significant challenge in fields like insurance pricing, where factors like driver age, location, and driving behavior interact in non-obvious ways to affect risk assessment and premium calculation. If these interactions are overlooked or misspecified, it can lead to suboptimal models, price distortions, and biased interpretations.
Key Innovation
The Tree-like Pairwise Interaction Network (PIN) is a novel neural network architecture that explicitly captures pairwise feature interactions in tabular data. This is achieved through a shared feed-forward neural network that mimics the structure of decision trees, enabling intrinsic interpretability and efficient SHapley's Additive Explanation (SHAP) computations.
Practical Impact
The PIN architecture has the potential to revolutionize predictive modeling in fields like insurance pricing. By accurately capturing pairwise feature interactions, PIN can provide valuable insights into how different factors contribute to the response variable, leading to more informed decision-making and improved model performance. This, in turn, can result in more accurate risk assessments, fairer premium calculations, and better customer outcomes.
Analogy / Intuitive Explanation
Imagine you're trying to predict the likelihood of a person getting a disease based on various factors like age, lifestyle, and medical history. Traditional models might look at each factor in isolation, but the PIN architecture would consider how each pair of factors interacts to affect the disease likelihood. For example, it might reveal that a person's age and lifestyle are highly correlated in their effect on disease likelihood, allowing for more accurate predictions and better treatment recommendations.
Paper Information
Categories:
stat.ML cs.LG stat.AP
Published Date:

arXiv ID:

2508.15678v1

Quick Actions