Vine Copulas as Differentiable Computational Graphs
By: Tuoyuan Cheng , Thibault Vatter , Thomas Nagler and more
Potential Business Impact:
Makes AI better at predicting and understanding data.
Vine copulas are sophisticated models for multivariate distributions and are increasingly used in machine learning. To facilitate their integration into modern ML pipelines, we introduce the vine computational graph, a DAG that abstracts the multilevel vine structure and associated computations. On this foundation, we devise new algorithms for conditional sampling, efficient sampling-order scheduling, and constructing vine structures for customized conditioning variables. We implement these ideas in torchvinecopulib, a GPU-accelerated Python library built upon PyTorch, delivering improved scalability for fitting, sampling, and density evaluation. Our experiments illustrate how gradient flowing through the vine can improve Vine Copula Autoencoders and that incorporating vines for uncertainty quantification in deep learning can outperform MC-dropout, deep ensembles, and Bayesian Neural Networks in sharpness, calibration, and runtime. By recasting vine copula models as computational graphs, our work connects classical dependence modeling with modern deep-learning toolchains and facilitates the integration of state-of-the-art copula methods in modern machine learning pipelines.
Similar Papers
Probabilistic patient risk profiling with pair-copula constructions
Methodology
Predicts surgery risks to help doctors decide care.
Trunc-Opt vine building algorithms
Methodology
Makes complex math models work better and faster.
Throwing Vines at the Wall: Structure Learning via Random Search
Methodology
Finds better patterns in data for computers.