Causal Structure Learning for Sparse Matrix Fill-in Reduction
Ziwei Li, Shuzi Niu, Tao Yuan, Huiyuan Li
Abstract
The performance of sparse direct solvers is fundamentally governed by fill-in, i.e. new nonzero entries arising from the LU factorization of a sparse matrix, as they dictate memory footprint and subsequent computation time. For decades, a variety of graph-theoretic algorithms have aimed to minimize fill-in, a problem known to be both NP-hard and critically important. While recent deep learning methods, optimizing surrogate fill-in objectives, show empirical promise and can outperform classical algorithms on certain matrices, they offer limited interpretability into the underlying mechanism of fill-in generation. To address this, we propose a novel reordering approach, Causal Triplet Structure Learning (CTS), which is grounded in the Fill-Path Theorem and reduces arbitrary-length fill-paths to length-two candidate triplets, identifies the causal structures that trigger fill-in, and intervenes to block their formation. Empirically, we design a multigrid-style GAT with KAN activations to learn vertex embeddings and introduce a causal triplet loss that discourages such structures during training. Experiments on the SuiteSparse Matrix Collection demonstrate that our method reduces fill-in by 6, leading to 12 speedup in factorization time compared to state-of-the-art methods on Chemical Process Simulation and Computational Fluid Dynamics matrices.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on1
Related papers
- What's Wrong with Deep Learning in Tree Search for Combinatorial OptimizationMaximilian Böther, Otto Kißig, Martin Taraz, Sarel Cohen et al.ICLR 2022 · 56 citations
- CUTS: Neural Causal Discovery from Irregular Time-Series DataYuxiao Cheng, Runzhao Yang, Tingxiong Xiao, Zongren Li et al.ICLR 2023 · 9 citations
- Simultaneous Missing Value Imputation and Structure Learning with GroupsPablo Morales-Alvarez, Wenbo Gong, Angus Lamb, Simon Woodhead et al.NeurIPS 2022 · 24 citations
- CausalSim: A Causal Framework for Unbiased Trace-Driven SimulationAbdullah Omar Alomar, Pouya Hamadanian, Arash Nasr-Esfahany, Anish Agarwal et al.NSDI 2023 · 46 citations
- Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural NetworksArnaud Deza, Elias B. Khalil, Zhenan Fan, Zirui Zhou et al.AAAI 2025
