├── LICENSE ├── README.md ├── environment.yml └── src ├── ablation.py ├── add.py ├── args.py ├── correlation.py ├── cosine_similarity.py ├── cross_product.py ├── datasets ├── cars.py ├── cifar10.py ├── cifar100.py ├── common.py ├── dtd.py ├── eurosat.py ├── gtsrb.py ├── imagenet.py ├── mnist.py ├── registry.py ├── resisc45.py ├── stl10.py ├── sun397.py ├── svhn.py └── templates.py ├── distributed.py ├── dot.py ├── eval.py ├── eval_optimized.py ├── eval_single_task.py ├── eval_task_addition.py ├── eval_task_negation.py ├── figure.py ├── finetune.py ├── heads.py ├── linearize.py ├── linearize_0.py ├── linearize_t.py ├── main_task_wise_adamerging.py ├── modeling.py ├── optimize.py ├── orthogonal.py ├── task_vectors.py ├── task_vectors_edited.py ├── utils.py ├── vector_relation.py └── visualization.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/environment.yml -------------------------------------------------------------------------------- /src/ablation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/ablation.py -------------------------------------------------------------------------------- /src/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/add.py -------------------------------------------------------------------------------- /src/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/args.py -------------------------------------------------------------------------------- /src/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/correlation.py -------------------------------------------------------------------------------- /src/cosine_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/cosine_similarity.py -------------------------------------------------------------------------------- /src/cross_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/cross_product.py -------------------------------------------------------------------------------- /src/datasets/cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/cars.py -------------------------------------------------------------------------------- /src/datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/cifar10.py -------------------------------------------------------------------------------- /src/datasets/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/cifar100.py -------------------------------------------------------------------------------- /src/datasets/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/common.py -------------------------------------------------------------------------------- /src/datasets/dtd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/dtd.py -------------------------------------------------------------------------------- /src/datasets/eurosat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/eurosat.py -------------------------------------------------------------------------------- /src/datasets/gtsrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/gtsrb.py -------------------------------------------------------------------------------- /src/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/imagenet.py -------------------------------------------------------------------------------- /src/datasets/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/mnist.py -------------------------------------------------------------------------------- /src/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/registry.py -------------------------------------------------------------------------------- /src/datasets/resisc45.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/resisc45.py -------------------------------------------------------------------------------- /src/datasets/stl10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/stl10.py -------------------------------------------------------------------------------- /src/datasets/sun397.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/sun397.py -------------------------------------------------------------------------------- /src/datasets/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/svhn.py -------------------------------------------------------------------------------- /src/datasets/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/datasets/templates.py -------------------------------------------------------------------------------- /src/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/distributed.py -------------------------------------------------------------------------------- /src/dot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/dot.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/eval_optimized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/eval_optimized.py -------------------------------------------------------------------------------- /src/eval_single_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/eval_single_task.py -------------------------------------------------------------------------------- /src/eval_task_addition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/eval_task_addition.py -------------------------------------------------------------------------------- /src/eval_task_negation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/eval_task_negation.py -------------------------------------------------------------------------------- /src/figure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/figure.py -------------------------------------------------------------------------------- /src/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/finetune.py -------------------------------------------------------------------------------- /src/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/heads.py -------------------------------------------------------------------------------- /src/linearize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/linearize.py -------------------------------------------------------------------------------- /src/linearize_0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/linearize_0.py -------------------------------------------------------------------------------- /src/linearize_t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/linearize_t.py -------------------------------------------------------------------------------- /src/main_task_wise_adamerging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/main_task_wise_adamerging.py -------------------------------------------------------------------------------- /src/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/modeling.py -------------------------------------------------------------------------------- /src/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/optimize.py -------------------------------------------------------------------------------- /src/orthogonal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/orthogonal.py -------------------------------------------------------------------------------- /src/task_vectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/task_vectors.py -------------------------------------------------------------------------------- /src/task_vectors_edited.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/task_vectors_edited.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/utils.py -------------------------------------------------------------------------------- /src/vector_relation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/vector_relation.py -------------------------------------------------------------------------------- /src/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyrie-23/linear_task_arithmetic/HEAD/src/visualization.py --------------------------------------------------------------------------------