Statistical-Neural Interaction Networks for Interpretable Mixed-Type Data Imputation
About
Real-world tabular databases routinely combine continuous measurements and categorical records, yet missing entries are pervasive and can distort downstream analysis. We propose Statistical-Neural Interaction (SNI), an interpretable mixed-type imputation framework that couples correlation-derived statistical priors with neural feature attention through a Controllable-Prior Feature Attention (CPFA) module. CPFA learns head-wise prior-strength coefficients $\{\lambda_h\}$ that softly regularize attention toward the prior while allowing data-driven deviations when nonlinear patterns appear to be present in the data. Beyond imputation, SNI aggregates attention maps into a directed feature-dependency matrix that summarizes which variables the imputer relied on, without requiring post-hoc explainers. We evaluate SNI against six baselines (Mean/Mode, MICE, KNN, MissForest, GAIN, MIWAE) on six datasets spanning ICU monitoring, population surveys, socio-economic statistics, and engineering applications. Under MCAR/strict-MAR at 30\% missingness, SNI is generally competitive on continuous metrics but is often outperformed by accuracy-first baselines (MissForest, MIWAE) on categorical variables; in return, it provides intrinsic dependency diagnostics and explicit statistical-neural trade-off parameters. We additionally report MNAR stress tests (with a mask-aware variant) and discuss computational cost, limitations -- particularly for severely imbalanced categorical targets -- and deployment scenarios where interpretability may justify the trade-off.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Data Imputation | eICU MNAR 10% missingness | NRMSE0.136 | 8 | |
| Imputation | Six datasets aggregated (test) | NRMSE4.22 | 8 | |
| Imputation | ComCri MNAR 10% missingness (test) | NRMSE0.1 | 8 | |
| Imputation | ComCri MNAR 30% missingness (test) | NRMSE0.126 | 8 | |
| Imputation | ComCri 50% missingness MNAR (test) | NRMSE0.177 | 8 | |
| Imputation | NHANES MNAR (10% missingness) | NRMSE0.082 | 8 | |
| Imputation | NHANES MNAR (30% missingness) | NRMSE0.12 | 8 | |
| Imputation | AutoMPG MNAR 10% missingness | NRMSE0.077 | 8 | |
| Imputation | AutoMPG MNAR (30% missingness) | NRMSE0.109 | 8 | |
| Imputation | AutoMPG MNAR (50% missingness) | NRMSE0.15 | 8 |