├── .gitignore ├── README.md ├── conv ├── attack.py ├── bayes_convnet.py ├── ctx.py ├── models.py ├── svgd.py ├── tp │ ├── batch_norm.py │ └── training.py ├── utils.py └── variationals.py ├── dcb ├── README.md ├── bandits │ ├── algorithms │ │ ├── bb_alpha_divergence_model.py │ │ ├── bf_variational_neural_bandit_model.py │ │ ├── bootstrapped_bnn_sampling.py │ │ ├── fixed_policy_sampling.py │ │ ├── linear_full_posterior_sampling.py │ │ ├── multitask_gp.py │ │ ├── neural_bandit_model.py │ │ ├── neural_linear_sampling.py │ │ ├── nnet.py │ │ ├── parameter_noise_sampling.py │ │ ├── posterior_bnn_sampling.py │ │ ├── rf.py │ │ ├── svgd.py │ │ ├── svgd_model.py │ │ ├── uniform_sampling.py │ │ ├── utils.py │ │ └── variational_neural_bandit_model.py │ ├── core │ │ ├── bandit_algorithm.py │ │ ├── bayesian_nn.py │ │ ├── contextual_bandit.py │ │ └── contextual_dataset.py │ └── data │ │ ├── data_sampler.py │ │ └── synthetic_data_sampler.py ├── gather-result.py ├── gather-result.sh ├── main.py └── run.py ├── exps ├── bandit.sh ├── convnet.sh ├── mnist-adv.sh ├── mnist-clean.sh ├── synth-fig1-fig4.sh ├── synth-fig5.sh ├── synthetic_vis.py ├── uci-collect-results.sh ├── uci-exp-main.py ├── uci-exp-tbl6.py └── uci-exp-tbl7.py ├── ffn ├── attack.py ├── attacks_tf_orig.py ├── bnn_stein.py ├── bnn_stein_f.py ├── dataset.py ├── dsdgp_datasets.py ├── hmc.py ├── rf.py ├── svgd.py ├── utils.py └── utils_tf.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/README.md -------------------------------------------------------------------------------- /conv/attack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/attack.py -------------------------------------------------------------------------------- /conv/bayes_convnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/bayes_convnet.py -------------------------------------------------------------------------------- /conv/ctx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/ctx.py -------------------------------------------------------------------------------- /conv/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/models.py -------------------------------------------------------------------------------- /conv/svgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/svgd.py -------------------------------------------------------------------------------- /conv/tp/batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/tp/batch_norm.py -------------------------------------------------------------------------------- /conv/tp/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/tp/training.py -------------------------------------------------------------------------------- /conv/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/utils.py -------------------------------------------------------------------------------- /conv/variationals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/conv/variationals.py -------------------------------------------------------------------------------- /dcb/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/README.md -------------------------------------------------------------------------------- /dcb/bandits/algorithms/bb_alpha_divergence_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/bb_alpha_divergence_model.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/bf_variational_neural_bandit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/bf_variational_neural_bandit_model.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/bootstrapped_bnn_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/bootstrapped_bnn_sampling.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/fixed_policy_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/fixed_policy_sampling.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/linear_full_posterior_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/linear_full_posterior_sampling.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/multitask_gp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/multitask_gp.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/neural_bandit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/neural_bandit_model.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/neural_linear_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/neural_linear_sampling.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/nnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/nnet.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/parameter_noise_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/parameter_noise_sampling.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/posterior_bnn_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/posterior_bnn_sampling.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/rf.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/svgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/svgd.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/svgd_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/svgd_model.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/uniform_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/uniform_sampling.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/utils.py -------------------------------------------------------------------------------- /dcb/bandits/algorithms/variational_neural_bandit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/algorithms/variational_neural_bandit_model.py -------------------------------------------------------------------------------- /dcb/bandits/core/bandit_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/core/bandit_algorithm.py -------------------------------------------------------------------------------- /dcb/bandits/core/bayesian_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/core/bayesian_nn.py -------------------------------------------------------------------------------- /dcb/bandits/core/contextual_bandit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/core/contextual_bandit.py -------------------------------------------------------------------------------- /dcb/bandits/core/contextual_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/core/contextual_dataset.py -------------------------------------------------------------------------------- /dcb/bandits/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/data/data_sampler.py -------------------------------------------------------------------------------- /dcb/bandits/data/synthetic_data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/bandits/data/synthetic_data_sampler.py -------------------------------------------------------------------------------- /dcb/gather-result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/gather-result.py -------------------------------------------------------------------------------- /dcb/gather-result.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/gather-result.sh -------------------------------------------------------------------------------- /dcb/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/main.py -------------------------------------------------------------------------------- /dcb/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/dcb/run.py -------------------------------------------------------------------------------- /exps/bandit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/bandit.sh -------------------------------------------------------------------------------- /exps/convnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/convnet.sh -------------------------------------------------------------------------------- /exps/mnist-adv.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/mnist-adv.sh -------------------------------------------------------------------------------- /exps/mnist-clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/mnist-clean.sh -------------------------------------------------------------------------------- /exps/synth-fig1-fig4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/synth-fig1-fig4.sh -------------------------------------------------------------------------------- /exps/synth-fig5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/synth-fig5.sh -------------------------------------------------------------------------------- /exps/synthetic_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/synthetic_vis.py -------------------------------------------------------------------------------- /exps/uci-collect-results.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/uci-collect-results.sh -------------------------------------------------------------------------------- /exps/uci-exp-main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/uci-exp-main.py -------------------------------------------------------------------------------- /exps/uci-exp-tbl6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/uci-exp-tbl6.py -------------------------------------------------------------------------------- /exps/uci-exp-tbl7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/exps/uci-exp-tbl7.py -------------------------------------------------------------------------------- /ffn/attack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/attack.py -------------------------------------------------------------------------------- /ffn/attacks_tf_orig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/attacks_tf_orig.py -------------------------------------------------------------------------------- /ffn/bnn_stein.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/bnn_stein.py -------------------------------------------------------------------------------- /ffn/bnn_stein_f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/bnn_stein_f.py -------------------------------------------------------------------------------- /ffn/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/dataset.py -------------------------------------------------------------------------------- /ffn/dsdgp_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/dsdgp_datasets.py -------------------------------------------------------------------------------- /ffn/hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/hmc.py -------------------------------------------------------------------------------- /ffn/rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/rf.py -------------------------------------------------------------------------------- /ffn/svgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/svgd.py -------------------------------------------------------------------------------- /ffn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/utils.py -------------------------------------------------------------------------------- /ffn/utils_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/ffn/utils_tf.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-ml/fpovi/HEAD/requirements.txt --------------------------------------------------------------------------------