An implementation of "Adaptive Neuron-wise Discriminant Criterion and Adaptive Center Loss at Hidden Layer for Deep Convolutional Neural Network" in Pytorch.

The paper is here https://arxiv.org/abs/2004.08074

• OS : Ubuntu 20.04 LTS
• Python : v 3.7.3
• Pytorch : v 1.12.0

pip install requirements

And you install pytorch from official page https://pytorch.org/

Modify the settings for training until 4th cell on adacenter_adadiscriminant.ipynb and then run it on the jupyter.

We provide program for related works criterions such as discriminatn criterion and center loss. You can train them

• adacenter.ipynb : training with adaptie center loss
• adadiscriminant_criterion.ipynb : training with adaptive discirminant criterion
• center.ipynb : training with center loss (related work)
• discriminant_criterion.ipynb : training with discriminant criterion (related work)