English

TT-FSI: Scalable Faithful Shapley Interactions via Tensor-Train

Machine Learning 2026-01-06 v1

Abstract

The Faithful Shapley Interaction (FSI) index uniquely satisfies the faithfulness axiom among Shapley interaction indices, but computing FSI requires O(d2d)O(d^\ell \cdot 2^d) time and existing implementations use O(4d)O(4^d) memory. We present TT-FSI, which exploits FSI's algebraic structure via Matrix Product Operators (MPO). Our main theoretical contribution is proving that the linear operator vFSI(v)v \mapsto \text{FSI}(v) admits an MPO representation with TT-rank O(d)O(\ell d), enabling an efficient sweep algorithm with O(2d32d)O(\ell^2 d^3 \cdot 2^d) time and O(d2)O(\ell d^2) core storage an exponential improvement over existing methods. Experiments on six datasets (d=8d=8 to d=20d=20) demonstrate up to 280×\times speedup over baseline, 85×\times over SHAP-IQ, and 290×\times memory reduction. TT-FSI scales to d=20d=20 (1M coalitions) where all competing methods fail.

Keywords

Cite

@article{arxiv.2601.01903,
  title  = {TT-FSI: Scalable Faithful Shapley Interactions via Tensor-Train},
  author = {Ungsik Kim and Suwon Lee},
  journal= {arXiv preprint arXiv:2601.01903},
  year   = {2026}
}