├── .gitignore ├── LICENSE ├── README.md ├── README.rst ├── experiments ├── ensembling │ ├── bma_rank.ipynb │ ├── bma_samples_and_scaling.ipynb │ ├── bn_updates_fraction.py │ ├── ensembling.py │ └── swa_ecdf_ensembles.py ├── grad_cov │ ├── grad_cov_utils.py │ └── run_grad_cov.py ├── hessian_eigs │ ├── hess_vec_prod.py │ └── run_hess_eigs.py ├── imagenet │ ├── data.py │ ├── eval_net_imagenet.py │ ├── eval_swag_imagenet.py │ ├── run_swag_imagenet.py │ └── test_swag_gpu.py ├── segmentation │ ├── README.md │ ├── eval_ensemble.py │ ├── evaluate.py │ ├── model_results.ipynb │ ├── secondary_calibration_checks.ipynb │ ├── train.py │ └── utils │ │ ├── __init__.py │ │ ├── imgs.py │ │ └── training.py ├── train │ ├── README.md │ ├── run_swag.py │ └── train.py ├── uncertainty │ ├── README.md │ ├── out_of_dist_plots.ipynb │ ├── save_calibration_curves.py │ ├── temp_scaling.py │ └── uncertainty.py └── width │ ├── pca_2d_plane.py │ ├── pca_orth_width.py │ ├── pca_width.py │ ├── swag_2d_plane.py │ └── swag_width.py ├── neurips2019_poster.pdf ├── plots ├── c100_resnet110.jpg ├── c100_resnet110_swag_2d_01.jpg └── stl_wrn.jpg ├── requirements.txt ├── setup.py ├── short_slides.pdf ├── slides.pdf ├── swag ├── __init__.py ├── camvid.py ├── data.py ├── losses.py ├── models │ ├── __init__.py │ ├── joint_transforms.py │ ├── layers.py │ ├── lenet5.py │ ├── preresnet.py │ ├── preresnet_dropout.py │ ├── tiramisu.py │ ├── vgg.py │ ├── vgg_dropout.py │ ├── wide_resnet.py │ └── wide_resnet_dropout.py ├── posteriors │ ├── __init__.py │ ├── diag_laplace.py │ ├── laplace.py │ └── swag.py └── utils.py └── tests ├── test_lowrank_p_diag.py └── test_swag_sampling.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/README.md -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/ensembling/bma_rank.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/ensembling/bma_rank.ipynb -------------------------------------------------------------------------------- /experiments/ensembling/bma_samples_and_scaling.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/ensembling/bma_samples_and_scaling.ipynb -------------------------------------------------------------------------------- /experiments/ensembling/bn_updates_fraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/ensembling/bn_updates_fraction.py -------------------------------------------------------------------------------- /experiments/ensembling/ensembling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/ensembling/ensembling.py -------------------------------------------------------------------------------- /experiments/ensembling/swa_ecdf_ensembles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/ensembling/swa_ecdf_ensembles.py -------------------------------------------------------------------------------- /experiments/grad_cov/grad_cov_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/grad_cov/grad_cov_utils.py -------------------------------------------------------------------------------- /experiments/grad_cov/run_grad_cov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/grad_cov/run_grad_cov.py -------------------------------------------------------------------------------- /experiments/hessian_eigs/hess_vec_prod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/hessian_eigs/hess_vec_prod.py -------------------------------------------------------------------------------- /experiments/hessian_eigs/run_hess_eigs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/hessian_eigs/run_hess_eigs.py -------------------------------------------------------------------------------- /experiments/imagenet/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/imagenet/data.py -------------------------------------------------------------------------------- /experiments/imagenet/eval_net_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/imagenet/eval_net_imagenet.py -------------------------------------------------------------------------------- /experiments/imagenet/eval_swag_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/imagenet/eval_swag_imagenet.py -------------------------------------------------------------------------------- /experiments/imagenet/run_swag_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/imagenet/run_swag_imagenet.py -------------------------------------------------------------------------------- /experiments/imagenet/test_swag_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/imagenet/test_swag_gpu.py -------------------------------------------------------------------------------- /experiments/segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/README.md -------------------------------------------------------------------------------- /experiments/segmentation/eval_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/eval_ensemble.py -------------------------------------------------------------------------------- /experiments/segmentation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/evaluate.py -------------------------------------------------------------------------------- /experiments/segmentation/model_results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/model_results.ipynb -------------------------------------------------------------------------------- /experiments/segmentation/secondary_calibration_checks.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/secondary_calibration_checks.ipynb -------------------------------------------------------------------------------- /experiments/segmentation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/train.py -------------------------------------------------------------------------------- /experiments/segmentation/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/segmentation/utils/imgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/utils/imgs.py -------------------------------------------------------------------------------- /experiments/segmentation/utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/segmentation/utils/training.py -------------------------------------------------------------------------------- /experiments/train/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/train/README.md -------------------------------------------------------------------------------- /experiments/train/run_swag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/train/run_swag.py -------------------------------------------------------------------------------- /experiments/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/train/train.py -------------------------------------------------------------------------------- /experiments/uncertainty/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/uncertainty/README.md -------------------------------------------------------------------------------- /experiments/uncertainty/out_of_dist_plots.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/uncertainty/out_of_dist_plots.ipynb -------------------------------------------------------------------------------- /experiments/uncertainty/save_calibration_curves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/uncertainty/save_calibration_curves.py -------------------------------------------------------------------------------- /experiments/uncertainty/temp_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/uncertainty/temp_scaling.py -------------------------------------------------------------------------------- /experiments/uncertainty/uncertainty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/uncertainty/uncertainty.py -------------------------------------------------------------------------------- /experiments/width/pca_2d_plane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/width/pca_2d_plane.py -------------------------------------------------------------------------------- /experiments/width/pca_orth_width.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/width/pca_orth_width.py -------------------------------------------------------------------------------- /experiments/width/pca_width.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/width/pca_width.py -------------------------------------------------------------------------------- /experiments/width/swag_2d_plane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/width/swag_2d_plane.py -------------------------------------------------------------------------------- /experiments/width/swag_width.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/experiments/width/swag_width.py -------------------------------------------------------------------------------- /neurips2019_poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/neurips2019_poster.pdf -------------------------------------------------------------------------------- /plots/c100_resnet110.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/plots/c100_resnet110.jpg -------------------------------------------------------------------------------- /plots/c100_resnet110_swag_2d_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/plots/c100_resnet110_swag_2d_01.jpg -------------------------------------------------------------------------------- /plots/stl_wrn.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/plots/stl_wrn.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/setup.py -------------------------------------------------------------------------------- /short_slides.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/short_slides.pdf -------------------------------------------------------------------------------- /slides.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/slides.pdf -------------------------------------------------------------------------------- /swag/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/__init__.py -------------------------------------------------------------------------------- /swag/camvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/camvid.py -------------------------------------------------------------------------------- /swag/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/data.py -------------------------------------------------------------------------------- /swag/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/losses.py -------------------------------------------------------------------------------- /swag/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/__init__.py -------------------------------------------------------------------------------- /swag/models/joint_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/joint_transforms.py -------------------------------------------------------------------------------- /swag/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/layers.py -------------------------------------------------------------------------------- /swag/models/lenet5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/lenet5.py -------------------------------------------------------------------------------- /swag/models/preresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/preresnet.py -------------------------------------------------------------------------------- /swag/models/preresnet_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/preresnet_dropout.py -------------------------------------------------------------------------------- /swag/models/tiramisu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/tiramisu.py -------------------------------------------------------------------------------- /swag/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/vgg.py -------------------------------------------------------------------------------- /swag/models/vgg_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/vgg_dropout.py -------------------------------------------------------------------------------- /swag/models/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/wide_resnet.py -------------------------------------------------------------------------------- /swag/models/wide_resnet_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/models/wide_resnet_dropout.py -------------------------------------------------------------------------------- /swag/posteriors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/posteriors/__init__.py -------------------------------------------------------------------------------- /swag/posteriors/diag_laplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/posteriors/diag_laplace.py -------------------------------------------------------------------------------- /swag/posteriors/laplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/posteriors/laplace.py -------------------------------------------------------------------------------- /swag/posteriors/swag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/posteriors/swag.py -------------------------------------------------------------------------------- /swag/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/swag/utils.py -------------------------------------------------------------------------------- /tests/test_lowrank_p_diag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/tests/test_lowrank_p_diag.py -------------------------------------------------------------------------------- /tests/test_swag_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wjmaddox/swa_gaussian/HEAD/tests/test_swag_sampling.py --------------------------------------------------------------------------------