ICLRMar, 2024

加速 LiNGAM:以 GPU 速度学习因果 DAG

TL;DR通过将现有的因果发现方法进行高效并行化,可以使它们适用于数千个维度的数据集,从而解决了基于组合优化或搜索的现有方法在大规模数据集上运行速度慢的问题。具体地,我们并行化了 LiNGAM 方法,并加速其中的因果排序子过程,实现了与现有顺序实现相比高达 32 倍的加速,从而使其在大规模基因表达数据和美国股票数据的因果推断和因果发现中表现出与专门的连续优化方法相媲美的结果。