├── LICENSE ├── README.md ├── __init__.py ├── architectures_and_layers ├── __init__.py ├── custom_batch_norm.py ├── resnet18_arch.py ├── resnet2056_arch.py └── vgg_arch.py ├── config.py ├── experiments.py ├── figures_and_tables ├── Figure_2.py ├── Figure_3.py ├── Figure_4.py ├── Figure_5.py ├── Figure_6.py ├── Figure_B1.py ├── Figure_B2.py ├── Figure_C3.py ├── Figure_D2.py ├── Figure_Ei.py ├── PDFs │ └── README.md ├── Table_C1.py ├── Table_F1.py ├── __init__.py └── utils.py ├── hessian_eigenthings ├── __init__.py ├── hvp_operator.py ├── lanczos.py ├── power_iter.py └── readme.md ├── logs └── README.md ├── models.py ├── scripts_for_experiments ├── Figure_2.sh ├── Figure_3.sh ├── Figure_4.sh ├── Figure_5.sh ├── Figure_B2.sh ├── Figure_C3.sh ├── Figure_D2.sh ├── Table_C1.sh └── Table_F1.sh └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /architectures_and_layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /architectures_and_layers/custom_batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/architectures_and_layers/custom_batch_norm.py -------------------------------------------------------------------------------- /architectures_and_layers/resnet18_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/architectures_and_layers/resnet18_arch.py -------------------------------------------------------------------------------- /architectures_and_layers/resnet2056_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/architectures_and_layers/resnet2056_arch.py -------------------------------------------------------------------------------- /architectures_and_layers/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/architectures_and_layers/vgg_arch.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/config.py -------------------------------------------------------------------------------- /experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/experiments.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_2.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_3.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_4.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_5.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_6.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_B1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_B1.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_B2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_B2.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_C3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_C3.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_D2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_D2.py -------------------------------------------------------------------------------- /figures_and_tables/Figure_Ei.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Figure_Ei.py -------------------------------------------------------------------------------- /figures_and_tables/PDFs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/PDFs/README.md -------------------------------------------------------------------------------- /figures_and_tables/Table_C1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Table_C1.py -------------------------------------------------------------------------------- /figures_and_tables/Table_F1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/Table_F1.py -------------------------------------------------------------------------------- /figures_and_tables/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /figures_and_tables/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/figures_and_tables/utils.py -------------------------------------------------------------------------------- /hessian_eigenthings/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/hessian_eigenthings/__init__.py -------------------------------------------------------------------------------- /hessian_eigenthings/hvp_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/hessian_eigenthings/hvp_operator.py -------------------------------------------------------------------------------- /hessian_eigenthings/lanczos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/hessian_eigenthings/lanczos.py -------------------------------------------------------------------------------- /hessian_eigenthings/power_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/hessian_eigenthings/power_iter.py -------------------------------------------------------------------------------- /hessian_eigenthings/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/hessian_eigenthings/readme.md -------------------------------------------------------------------------------- /logs/README.md: -------------------------------------------------------------------------------- 1 | ## This folder holds results of runs 2 | -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/models.py -------------------------------------------------------------------------------- /scripts_for_experiments/Figure_2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Figure_2.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Figure_3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Figure_3.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Figure_4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Figure_4.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Figure_5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Figure_5.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Figure_B2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Figure_B2.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Figure_C3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Figure_C3.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Figure_D2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Figure_D2.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Table_C1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Table_C1.sh -------------------------------------------------------------------------------- /scripts_for_experiments/Table_F1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/scripts_for_experiments/Table_F1.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbartoldson/GeneralizationStabilityTradeoff/HEAD/utils.py --------------------------------------------------------------------------------