Less Is More Important: A Probability Density Function Guided Attention Module for Convolutional Neural Networks [PyTorch]
This repository is for PdfAM introduced in the following paper:
Jingfen Xie and Jian Zhang. Less Is More Important: A Probability Density Function Guided Attention Module for Convolutional Neural Networks. AAAI 2023.
- CUDA: 10.2
- Python: 3.6.9
- Toolkit: PyTorch 1.4.0
- GPU: NVIDIA Tesla V100
For the two experiments on classification task and MRI reconstruction task, please refer to the corresponding directory.