PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021

Last update: Nov 08, 2022

Overview

HyperSPN

This repository contains code for the paper:

HyperSPNs: Compact and Expressive Probabilistic Circuits

"HyperSPNs: Compact and Expressive Probabilistic Circuits"
Andy Shih, Dorsa Sadigh, Stefano Ermon
In Proceedings of the 35th Conference on Neural Information Processing Systems (NeurIPS), 2021

@inproceedings{ShihSEneurips21,
  author    = {Andy Shih and Dorsa Sadigh and Stefano Ermon},
  title     = {HyperSPNs: Compact and Expressive Probabilistic Circuits},
  booktitle = {Advances in Neural Information Processing Systems 34 (NeurIPS)},
  month     = {december},
  year      = {2021},
  keywords  = {conference}
}

Installation

conda env create -f environment.yml

Optionally, for EinsumNetworks:

cd EinsumNetworks
pip3 install -r requirements.txt

Datasets and Repos

The Twenty Datasets benchmark is from here.

The Amazon Baby Registries benchmark is from here. The dataset was converted from the set format into the binary format.

The Einsum Network repository is from here.

Commands

Experiments can be launched with the helper bash files

runid=0
bash bashfiles/run_hyperspn.bash ${runid} 5
bash bashfiles/run_hyperspn.bash ${runid} 10
bash bashfiles/run_hyperspn.bash ${runid} 20

bash bashfiles/run_spn.bash ${runid} 1e-3
bash bashfiles/run_spn.bash ${runid} 1e-4
bash bashfiles/run_spn.bash ${runid} 1e-5

cd EinsumNetworks/src/
python train_svhn_mixture.py --run=0
python train_svhn_mixture.py --nn --run=0

PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021

Related tags

Overview

HyperSPN

Installation

Datasets and Repos

Commands

Owner

Seg-Torch for Image Segmentation with Torch

ReConsider is a re-ranking model that re-ranks the top-K (passage, answer-span) predictions of an Open-Domain QA Model like DPR (Karpukhin et al., 2020).

Autoencoder - Reducing the Dimensionality of Data with Neural Network

8-week curriculum for AI Builders

OMLT: Optimization and Machine Learning Toolkit

NDE: Climate Modeling with Neural Diffusion Equation, ICDM'21

Efficiently Disentangle Causal Representations

Parameterized Explainer for Graph Neural Network

Experiments on Flood Segmentation on Sentinel-1 SAR Imagery with Cyclical Pseudo Labeling and Noisy Student Training

Quantized models with python

PyTorch implementation of adversarial patch

Source code for the GPT-2 story generation models in the EMNLP 2020 paper "STORIUM: A Dataset and Evaluation Platform for Human-in-the-Loop Story Generation"

Malware Env for OpenAI Gym

NeurIPS 2021, self-supervised 6D pose on category level

Colar: Effective and Efficient Online Action Detection by Consulting Exemplars, CVPR 2022.

This is the repository for our paper SimpleTrack: Understanding and Rethinking 3D Multi-object Tracking

App customer segmentation cohort rfm clustering

Next-Best-View Estimation based on Deep Reinforcement Learning for Active Object Classification

Extracts essential Mediapipe face landmarks and arranges them in a sequenced order.

Unified API to facilitate usage of pre-trained "perceptor" models, a la CLIP