├── LICENSE ├── README.md ├── __init__.py ├── alignment ├── align_along_curve.py ├── align_googlenet.py ├── align_models.py └── align_models_w2.py ├── definitions.py ├── evaluation ├── eval_adversarial_curve.py ├── eval_curve.py ├── eval_plane_with_alignment.py └── eval_plane_with_curve.py ├── images ├── img01.png └── neurips_img.png ├── models ├── __init__.py ├── curves.py ├── curves_pam.py ├── inception.py ├── resnet.py └── tiny_ten.py ├── requirements.txt ├── training ├── train_curve.py ├── train_curve_adversarial.py ├── train_curve_pam.py ├── train_model.py └── train_model_adversarial.py ├── utils ├── __init__.py ├── alignment.py ├── attack.py ├── attack_curve.py ├── birkhoff.py ├── crosscorrelation.py ├── data.py ├── log_write.py ├── perm_utils.py ├── tinyimagenet200_data.py └── utils.py └── visualization ├── layerwise_mean_correlation ├── viz_align_along_curves.py └── viz_align_models.py ├── viz_adversarial_curve.py ├── viz_curve.py ├── viz_curve_on_plane.py └── viz_plane.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /alignment/align_along_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/alignment/align_along_curve.py -------------------------------------------------------------------------------- /alignment/align_googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/alignment/align_googlenet.py -------------------------------------------------------------------------------- /alignment/align_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/alignment/align_models.py -------------------------------------------------------------------------------- /alignment/align_models_w2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/alignment/align_models_w2.py -------------------------------------------------------------------------------- /definitions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/definitions.py -------------------------------------------------------------------------------- /evaluation/eval_adversarial_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/evaluation/eval_adversarial_curve.py -------------------------------------------------------------------------------- /evaluation/eval_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/evaluation/eval_curve.py -------------------------------------------------------------------------------- /evaluation/eval_plane_with_alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/evaluation/eval_plane_with_alignment.py -------------------------------------------------------------------------------- /evaluation/eval_plane_with_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/evaluation/eval_plane_with_curve.py -------------------------------------------------------------------------------- /images/img01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/images/img01.png -------------------------------------------------------------------------------- /images/neurips_img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/images/neurips_img.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/curves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/models/curves.py -------------------------------------------------------------------------------- /models/curves_pam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/models/curves_pam.py -------------------------------------------------------------------------------- /models/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/models/inception.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/tiny_ten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/models/tiny_ten.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/requirements.txt -------------------------------------------------------------------------------- /training/train_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/training/train_curve.py -------------------------------------------------------------------------------- /training/train_curve_adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/training/train_curve_adversarial.py -------------------------------------------------------------------------------- /training/train_curve_pam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/training/train_curve_pam.py -------------------------------------------------------------------------------- /training/train_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/training/train_model.py -------------------------------------------------------------------------------- /training/train_model_adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/training/train_model_adversarial.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import * 2 | -------------------------------------------------------------------------------- /utils/alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/alignment.py -------------------------------------------------------------------------------- /utils/attack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/attack.py -------------------------------------------------------------------------------- /utils/attack_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/attack_curve.py -------------------------------------------------------------------------------- /utils/birkhoff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/birkhoff.py -------------------------------------------------------------------------------- /utils/crosscorrelation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/crosscorrelation.py -------------------------------------------------------------------------------- /utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/data.py -------------------------------------------------------------------------------- /utils/log_write.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/log_write.py -------------------------------------------------------------------------------- /utils/perm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/perm_utils.py -------------------------------------------------------------------------------- /utils/tinyimagenet200_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/tinyimagenet200_data.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/utils/utils.py -------------------------------------------------------------------------------- /visualization/layerwise_mean_correlation/viz_align_along_curves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/visualization/layerwise_mean_correlation/viz_align_along_curves.py -------------------------------------------------------------------------------- /visualization/layerwise_mean_correlation/viz_align_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/visualization/layerwise_mean_correlation/viz_align_models.py -------------------------------------------------------------------------------- /visualization/viz_adversarial_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/visualization/viz_adversarial_curve.py -------------------------------------------------------------------------------- /visualization/viz_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/visualization/viz_curve.py -------------------------------------------------------------------------------- /visualization/viz_curve_on_plane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/visualization/viz_curve_on_plane.py -------------------------------------------------------------------------------- /visualization/viz_plane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/NeuronAlignment/HEAD/visualization/viz_plane.py --------------------------------------------------------------------------------