This is the code for the Bachelor's project "From Complexity to Clarity?: Pruning Towards Explainable Multi-task Models".
Notes:
- Dataset and pre-trained model weights not included. Input the correct path to the dataset and model instances in the appropriate locations.
Multi-Task FAMO model: https://github.com/Cranial-XIX/FAMO.git Netslim Pruning: https://github.com/yeyun11/netslim.git GradCAM/SegGradCam: https://github.com/jacobgil/pytorch-grad-cam.git IROF metric: https://github.com/understandable-machine-intelligence-lab/Quantus.git
The repositories above were forked and refactored to suit the requirements of the present study. Conceptual details on the adjustments made can be found in the written paper for this study.